Update app.py

app.py

@@ -27,7 +27,13 @@ def build_prompt(history, new_message):
     return prompt
 
 def chat(history, new_message):
-    # history: list[str], new_message: str
+    # Nếu history là chuỗi (do nhập tay trên UI), thì parse sang list
+    if isinstance(history, str):
+        try:
+            import ast
+            history = ast.literal_eval(history)
+        except:
+            history = [history]
     prompt = build_prompt(history, new_message)
     input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(device)
     with torch.no_grad():
@@ -46,14 +52,13 @@ def chat(history, new_message):
         response = output_text.strip()
     return response
 
-# CHỈ DÙNG Gradio Interface, input là [history, new_message]
 iface = gr.Interface(
     fn=chat,
     inputs=[
-        gr.inputs.Textbox(lines=8, label="History (JSON list, ví dụ: [\"User: Xin chào\"] )"),
-        gr.inputs.Textbox(label="New message")
+        gr.Textbox(lines=8, label="History (JSON list, ví dụ: [\"User: Xin chào\"] )"),
+        gr.Textbox(label="New message")
     ],
-    outputs="text",
+    outputs=gr.Textbox(label="AI Response"),
     title="MindVR Therapy Chatbot",
     allow_flagging="never"
 )