Update app.py

app.py

@@ -77,7 +77,7 @@ def respond_to_chat(message, history, state, selected_model):
     state["raw_documents"] = raw_documents
     state["outputs"] = outputs
 
-    return response
+    return {"role": "assistant", "content": response}
 
 
 def show_specs(state, history=None):
@@ -90,7 +90,11 @@ def show_specs(state, history=None):
     if not specs_map:
         df = pd.DataFrame({}, columns=columns)
         markdown_table = df.to_markdown(index=False)
-        return f"📄 Thông số kỹ thuật\n{markdown_table}"
+        history.append({
+            "role": "assistant",
+            "content": f"📄 Thông số kỹ thuật\n{markdown_table}"
+        })
+        return history
 
     # print(specs_map)
     for prod_id, data in specs_map.items():
@@ -184,7 +188,11 @@ def show_specs(state, history=None):
             [["Không có thông số kỹ thuật", "", ""]], columns=columns)
 
     markdown_table = df.to_markdown(index=False)
-    return f"📄 Thông số kỹ thuật\n{markdown_table}"
+    history.append({
+        "role": "assistant",
+        "content": f"📄 Thông số kỹ thuật\n{markdown_table}"
+    })
+    return history
 
 
 def show_advantages(state, history=None):
@@ -222,8 +230,12 @@ def show_advantages(state, history=None):
     else:
         df = pd.DataFrame([["Không có ưu điểm", ""]], columns=columns)
 
-    markdown_table = df.to_markdown(index=False)
-    return f"💡 Ưu điểm nổi trội\n{markdown_table}"
+    mmarkdown_table = df.to_markdown(index=False)
+    history.append({
+        "role": "assistant",
+        "content": f"💡 Ưu điểm nổi trội\n{markdown_table}"
+    })
+    return history
 
 
 def show_solution_packages(state, history=None):
@@ -256,7 +268,11 @@ def show_solution_packages(state, history=None):
 
         markdown_content += spec_content + "\n\n"
 
-    return markdown_content
+    history.append({
+            "role": "assistant",
+            "content": markdown_content
+        })
+    return history
 
 
 css = """
@@ -371,6 +387,7 @@ with gr.Blocks(fill_height=True, css=css) as demo:
             "https://cdn-icons-png.flaticon.com/512/219/219983.png",
             "assets/agent.png"
         ],
+        type="messages",
         height=800
     )