Update app.py

app.py

@@ -7,32 +7,39 @@ def chat_interface_fn(message, history, session_id):
     'session_id' is used to store conversation across turns.
     Deduplicates consecutive repeated Q&A pairs to avoid repetition.
     """
-    # 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 = [
-            msg for h in history 
-            for msg in [
-                {"role": "user", "content": h[0]},
-                {"role": "assistant", "content": h[1]}
+    # Initialize history if None
+    if history is None:
+        history = []
+    
+    # Ensure we're working with the correct history format
+    # Gradio 3.x sends history as a list of tuples (user, assistant)
+    if isinstance(history, list):
+        if len(history) > 0 and isinstance(history[0], tuple):
+            history = [
+                {"role": "user" if i % 2 == 0 else "assistant", "content": msg}
+                for tup in history
+                for i, msg in enumerate([tup[0], tup[1]])
             ]
-        ]
     
-    # 1) Get answer from the session-based memory pipeline
-    answer = run_with_session_memory(message, session_id)
+    # Get answer from the session-based memory pipeline
+    try:
+        answer = run_with_session_memory(message, session_id)
+    except Exception as e:
+        print(f"Error in run_with_session_memory: {str(e)}")
+        answer = "I apologize, but I encountered an error processing your request."
     
-    # 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})
+    # Format for Gradio ChatInterface
+    # Gradio expects a tuple of (new_chat_history, internal_history)
+    new_history = history + [
+        {"role": "user", "content": message},
+        {"role": "assistant", "content": answer}
+    ]
     
-    # 3) Convert history to message dictionaries for display
-    message_dicts = []
-    for msg in history:
-        message_dicts.append(msg)
+    # Convert history to format expected by Gradio
+    chat_history = [(msg["content"], hist["content"]) 
+                    for msg, hist in zip(new_history[::2], new_history[1::2])]
     
-    # Return the message dicts and updated history
-    return message_dicts, history
+    return chat_history, new_history
 
 # Custom CSS for chat interface
 my_chat_css = """
@@ -58,5 +65,5 @@ with gr.Blocks(css=my_chat_css) as demo:
         description="Ask your questions. The session_id determines your stored memory."
     )
 
-# Launch the Gradio interface
-demo.launch()
+# Launch the Gradio interface with sharing enabled
+demo.launch(share=True)