Update app.py

app.py

@@ -3,48 +3,58 @@ import gradio as gr
 import os
 import logging
 import json
-#import uuid
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 openai.api_key = os.environ['key']
-
-#session_id = str(uuid.uuid4())
-
-#messages = [{"role": "system", "content": "Your name is Harry and you are an AI assistant who can answer any question except to clarify that you are an AI"},]
+initMsg = os.environ['init']
+ansFirst = os.environ['ansfirst']
+
+mod = 3;
+
+def trimMessages(messages):
+    # Assuming messages is an array of objects with "role" and "content" properties
+    content = {}
+    for message in messages:        
+        if message["role"] == "assistant":
+            if "Current Status:" in message["content"]:
+                content = message["content"]
+                start_index = message["content"].index("Current Status:") + len("Current Status:")
+                end_index = message["content"].index("Wielding:")
+                new_content = message["content"][:start_index] + message["content"][end_index:]
+                message["content"] = new_content
+    if content != {}:
+        messages.append({"role": "assistant", "content": content})
+    return messages
 
 def chatbot(input, messages):
-    messages = messages or [{"role": "system", "content": "Your name is Harry and you are an AI assistant who can answer any question except to clarify that you are an AI!"}]
+    messages = messages or [{"role": "system", "content": initMsg}]
+    if len(messages) % mod == 0 or len(messages) == 1:
+        memory = trimMessages(messages)
+        logging.info("put memory:"+printMessages(memory))
+        chat = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo", messages=memory,
+            max_tokens=1048,n=1,temperature=0.5,
+        )
+        logging.info("put memory return:"+chat.choices[0].message.content)
+
     if input:
         messages.append({"role": "user", "content": input})
         chat = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo", messages=messages,
+            model="gpt-3.5-turbo", messages=messages[-2:],
             max_tokens=2048,n=1,temperature=0.5,
         )
         reply = chat.choices[0].message.content
         messages.append({"role": "assistant", "content": reply})
-        return reply, printMessages(messages), messages
+        return reply, messages
 
 def printMessages(messages):
     delimiter = '\n'
-    msg_string = delimiter.join([f"{obj['role']}:{obj['content']}" for obj in messages[1:]])
+    msg_string = delimiter.join([f"{obj['role']}:{obj['content']}" for obj in messages])
     logging.info("messages:"+msg_string)
     return msg_string
 
-#def clearMessage():
-#    global messages
-#    messages = [{"role": "system", "content": "Your name is Harry and you are an AI assistant who can answer any question except to clarify that you are an AI"},]
-#    logging.info("clear messages")
-
-#with gr.Blocks() as app:
-    #newChatBtn = gr.Button("New Chat")
-    #newChatBtn.click(clearMessage)
-    #tbReply = gr.Textbox(label="Reply")
-    #tbHistroy = gr.Textbox(label="History Chat")
-    #inputs = [gr.Textbox(lines=7, label="Chat with Harry"), newChatBtn] #Hidden(label="Session ID", value=session_id)]
-    #outputs = [gr.Textbox(label="History Chat"), gr.Textbox(label="Reply")]
-
-app = gr.Interface(fn=chatbot, inputs=[gr.Textbox(lines=7, label="Chat with Harry"), "state"], 
-                   outputs=[gr.Textbox(label="Reply"), gr.Textbox(label="History"), "state"], title="Chat with Harry",
-                   description="Ask anything you want",theme="compact")
+app = gr.Interface(fn=chatbot, inputs=[gr.Textbox(lines=7, label="You ask and answer questions below"), "state"], 
+                   outputs=[gr.Textbox(label="DND Game Reply", placeholder=ansFirst), "state"], title="DND Game",#, gr.Textbox(label="History"), 
+                   description="DND Game",theme="compact")
 app.launch(share=False)