Fix Gradio State schema error - simplify consultation state to avoid JSON schema processing bug

app.py

@@ -269,38 +269,40 @@ def build_ui():
                     interactive=False,
                 )
         
-        # Consultation state
-        consultation_state = gr.State(value={"consultant": None, "is_complete": False})
+        # Consultation state - use None as default to avoid Gradio schema issues
+        consultation_state = gr.State(value=None)
         
         def start_consultation():
             """Initialize a new consultation"""
-            return {
-                "consultant": None,
-                "is_complete": False
-            }, [], ""
+            return None, [], ""
         
         def send_consultation_message(user_message, state):
             """Send a message in the consultation"""
             if not user_message.strip():
-                # Get current chat history from state
-                consultant = state.get("consultant")
-                if consultant and hasattr(consultant, 'conversation_history'):
-                    # Convert conversation history to chatbot format
-                    chatbot_messages = []
-                    for msg in consultant.conversation_history:
-                        if msg.role == "user" and not msg.content.startswith("Initial request:"):
-                            chatbot_messages.append((msg.content, None))
-                        elif msg.role == "assistant" and chatbot_messages:
-                            # Pair with previous user message
-                            if chatbot_messages[-1][1] is None:
-                                chatbot_messages[-1] = (chatbot_messages[-1][0], msg.content)
-                            else:
-                                chatbot_messages.append((None, msg.content))
-                    return state, chatbot_messages, ""
+                # Return empty if no message
+                if state and isinstance(state, dict):
+                    consultant = state.get("consultant")
+                    if consultant and hasattr(consultant, 'conversation_history'):
+                        # Convert conversation history to chatbot format
+                        chatbot_messages = []
+                        for msg in consultant.conversation_history:
+                            if msg.role == "user" and not msg.content.startswith("Initial request:"):
+                                chatbot_messages.append((msg.content, None))
+                            elif msg.role == "assistant" and chatbot_messages:
+                                # Pair with previous user message
+                                if chatbot_messages[-1][1] is None:
+                                    chatbot_messages[-1] = (chatbot_messages[-1][0], msg.content)
+                                else:
+                                    chatbot_messages.append((None, msg.content))
+                        return state, chatbot_messages, ""
                 return state, [], ""
             
             from agent.consultation import NetworkConsultant
             
+            # Initialize state if needed
+            if not state:
+                state = {}
+            
             # Get or create consultant
             consultant = state.get("consultant")
             first_message = consultant is None
@@ -342,6 +344,9 @@ def build_ui():
         
         def use_consultation_summary(state):
             """Extract the consultation summary for use in the pipeline"""
+            if not state or not isinstance(state, dict):
+                return "No consultation history yet. Start a conversation first!"
+            
             consultant = state.get("consultant")
             if not consultant or not consultant.conversation_history:
                 return "No consultation history yet. Start a conversation first!"