update

app.py

@@ -46,9 +46,9 @@ def generate_text(messages):
         return "Unexpected output format."
 
 
-from gradio import ChatMessage
+
 def call_generate_text(message, history):
-    history.append(ChatMessage(role="user", content=message))
+    history.append({"role": "assistant", "content": message})
     print(message)
     print(history)
    
@@ -56,7 +56,7 @@ def call_generate_text(message, history):
    # messages = [{"role":"user","content":message}]
     try:
         text = generate_text(history)
-        history.append(ChatMessage(role="assistant", content=text))
+        history.append({"role": "assistant", "content": text})
         return text
     except RuntimeError  as e:
         print(f"An unexpected error occurred: {e}")