Update app.py

app.py

@@ -146,7 +146,12 @@ def reveal_coupon(query, grag_response):
     coupon = get_coupon(query, grag_response)
     return coupon
 
+# from gradio import ChatMessage
 
+# def chat_function(message, history):
+#     history.append(ChatMessage(role="user", content=message))
+#     history.append(ChatMessage(role="assistant", content="Hello, how can I help you?"))
+#     return history
 
 with gr.Blocks() as demo:
     gr.Markdown("# Comfy Virtual Assistant")
@@ -165,7 +170,9 @@ with gr.Blocks() as demo:
 
     def respond(message, chat_history):
             bot_message = query_tqa(message, 2)
-            chat_history.append((message, bot_message))
+            # chat_history.append((message, bot_message))
+            chat_history.append(ChatMessage(role="user", content=message))
+            chat_history.append(ChatMessage(role="assistant", content=bot_message))
             time.sleep(1)
             return "", chat_history