Update app.py

app.py

@@ -10,34 +10,40 @@ def chat_interface_fn(message, history, session_id):
     # Ensure history is a list of dictionaries
     if history and isinstance(history[0], tuple):
         print("DEBUG: Converting history from tuple format to dictionary format.")
-        history = [{"role": "user", "content": h[0]}, {"role": "assistant", "content": h[1]} for h in history]
-
+        history = [
+            msg for h in history 
+            for msg in [
+                {"role": "user", "content": h[0]},
+                {"role": "assistant", "content": h[1]}
+            ]
+        ]
+    
     # 1) Get answer from the session-based memory pipeline
     answer = run_with_session_memory(message, session_id)
-
+    
     # 2) Deduplicate consecutive identical exchanges
     if not history or history[-1]["content"] != answer:
         history.append({"role": "user", "content": message})
         history.append({"role": "assistant", "content": answer})
-
+    
     # 3) Convert history to message dictionaries for display
     message_dicts = []
     for msg in history:
         message_dicts.append(msg)
-
+    
     # Return the message dicts and updated history
     return message_dicts, history
 
 # Custom CSS for chat interface
 my_chat_css = """
 .gradio-container {
-  margin: auto;
+    margin: auto;
 }
 .user .wrap {
-  text-align: right !important;
+    text-align: right !important;
 }
 .assistant .wrap {
-  text-align: left !important;
+    text-align: left !important;
 }
 """
 
@@ -53,4 +59,4 @@ with gr.Blocks(css=my_chat_css) as demo:
     )
 
 # Launch the Gradio interface
-demo.launch()
+demo.launch()