Update Gradio version and refactor chat message handling to use session IDs

requirements.txt

@@ -1,5 +1,6 @@
 # Core Framework
-gradio==5.12.0
+gradio==4.44.0
+pydantic>=2.11.0,<2.12.0  # Pin pydantic to compatible version
 fastmcp>=0.3.0
 
 # AI/ML

ui/app.py

@@ -18,6 +18,10 @@ import json
 from chat.chat_engine import ChatEngine
 from chat.conversation import ConversationManager
 from chat.geocoding import GeocodingService
+import uuid
+
+# Global session storage (simple in-memory store)
+SESSIONS = {}
 
 # ============================================
 # DATABASE FUNCTIONS
@@ -220,49 +224,71 @@ def get_api_status():
 """
 
 
-def handle_chat_message(message, conversation_state):
+def handle_chat_message(message, session_id):
     """
     Handle chat message from user
 
     Args:
         message: User's message
-        conversation_state: ConversationManager instance
+        session_id: Session identifier string
 
     Returns:
-        Updated chatbot history, tool display, conversation state
+        Updated chatbot history, tool display, session_id
     """
+    # Get or create conversation for this session
+    if session_id not in SESSIONS:
+        SESSIONS[session_id] = ConversationManager()
+        welcome = chat_engine.get_welcome_message()
+        SESSIONS[session_id].add_message("assistant", welcome)
+
+    conversation = SESSIONS[session_id]
+
     if not message.strip():
-        return conversation_state.get_formatted_history(), conversation_state.get_tool_calls(), conversation_state
+        return conversation.get_formatted_history(), conversation.get_tool_calls(), session_id
 
     # Process message through chat engine
-    response, tool_calls = chat_engine.process_message(message, conversation_state)
+    response, tool_calls = chat_engine.process_message(message, conversation)
 
     # Return updated UI
-    return conversation_state.get_formatted_history(), conversation_state.get_tool_calls(), conversation_state
+    return conversation.get_formatted_history(), conversation.get_tool_calls(), session_id
 
 
-def reset_conversation():
+def reset_conversation(session_id):
     """Reset conversation to start fresh"""
+    # Create new session ID for fresh conversation
+    new_session_id = str(uuid.uuid4())
+
+    # Create new conversation for this session
     new_conversation = ConversationManager()
 
     # Add welcome message
     welcome = chat_engine.get_welcome_message()
     new_conversation.add_message("assistant", welcome)
 
+    # Store in sessions
+    SESSIONS[new_session_id] = new_conversation
+
     return (
         new_conversation.get_formatted_history(),
         [],  # Clear tool calls
-        new_conversation
+        new_session_id
     )
 
 
 def get_initial_chat():
     """Get initial chat state with welcome message"""
+    # Create new session ID
+    session_id = str(uuid.uuid4())
+
+    # Create conversation for this session
     conversation = ConversationManager()
     welcome = chat_engine.get_welcome_message()
     conversation.add_message("assistant", welcome)
 
-    return conversation.get_formatted_history(), [], conversation
+    # Store in sessions
+    SESSIONS[session_id] = conversation
+
+    return conversation.get_formatted_history(), [], session_id
 
 
 # ============================================
@@ -433,33 +459,34 @@ def create_interface():
                     gr.Markdown("See what tools the AI is using behind the scenes:")
                     tool_display = gr.JSON(label="Tools Called")
 
-                # Conversation state
-                conversation_state = gr.State(value=None)
+                # Conversation state - use session ID instead of complex object
+                session_id_state = gr.State(value=None)
 
                 # Initialize with welcome message
-                chatbot.value, tool_display.value, conversation_state.value = get_initial_chat()
+                chatbot.value, tool_display.value, session_id_state.value = get_initial_chat()
 
                 # Event handlers
-                def send_message(message, conv_state):
+                def send_message(message, sess_id):
                     """Handle send button click"""
-                    chat_history, tools, new_state = handle_chat_message(message, conv_state)
-                    return chat_history, tools, new_state, ""  # Clear input
+                    chat_history, tools, new_sess_id = handle_chat_message(message, sess_id)
+                    return chat_history, tools, new_sess_id, ""  # Clear input
 
                 send_btn.click(
                     fn=send_message,
-                    inputs=[msg_input, conversation_state],
-                    outputs=[chatbot, tool_display, conversation_state, msg_input]
+                    inputs=[msg_input, session_id_state],
+                    outputs=[chatbot, tool_display, session_id_state, msg_input]
                 )
 
                 msg_input.submit(
                     fn=send_message,
-                    inputs=[msg_input, conversation_state],
-                    outputs=[chatbot, tool_display, conversation_state, msg_input]
+                    inputs=[msg_input, session_id_state],
+                    outputs=[chatbot, tool_display, session_id_state, msg_input]
                 )
 
                 clear_btn.click(
                     fn=reset_conversation,
-                    outputs=[chatbot, tool_display, conversation_state]
+                    inputs=[session_id_state],
+                    outputs=[chatbot, tool_display, session_id_state]
                 )
 
             # ==========================================