Update app.py

app.py

@@ -175,10 +175,15 @@ def respond_with_enhanced_streaming(message, history):
     try:
         api_messages = [{"role": "system", "content": SYSTEM_MESSAGE}]
         if history:
-            # Prepare history for API call
-            for exchange in history[-5:]:
-                api_messages.append({"role": "user", "content": exchange[0]})
-                api_messages.append({"role": "assistant", "content": exchange[1]})
+            # Handle the new messages format from Gradio
+            for exchange in history[-5:]:  # Only use last 5 exchanges for context
+                if isinstance(exchange, dict):
+                    # New format: {"role": "user", "content": "..."}
+                    api_messages.append(exchange)
+                else:
+                    # Old format: [user_message, assistant_message]
+                    api_messages.append({"role": "user", "content": exchange[0]})
+                    api_messages.append({"role": "assistant", "content": exchange[1]})
         
         api_messages.append({"role": "user", "content": message})
         
@@ -191,7 +196,7 @@ def respond_with_enhanced_streaming(message, history):
             temperature=0.7,
             top_p=0.9,
             stream=True,
-            tools=tools, # Pass the tool definitions here
+            tools=tools,
         )
         
         # Buffers to handle multi-chunk tool calls
@@ -236,7 +241,6 @@ def respond_with_enhanced_streaming(message, history):
 
                         except json.JSONDecodeError:
                             logger.error("JSON parsing failed for tool arguments.")
-                            # Yield an error or handle it silently
                             full_response += f"<p style='color:red;'>Error parsing graph data.</p>"
                             yield full_response
                         except Exception as e: