Update app.py

app.py

@@ -20,10 +20,12 @@ logger = logging.getLogger("GradioApp")
 COMPANY_NAME = "OSO Consulting"
 
 def process_input(audio_file, text_input, model_choice, tone, state):
+    if state is None or not isinstance(state, dict):
+        state = {"conversation": [], "lead_stage": "initial"}
+
     audio_filename = f"input_{uuid.uuid4().hex}.wav"
     user_input = text_input.strip() if text_input else ""
 
-    # Check audio file only if it exists and has data
     if audio_file is not None and hasattr(audio_file, "read"):
         audio_bytes = audio_file.read()
         if len(audio_bytes) > 0:
@@ -37,13 +39,11 @@ def process_input(audio_file, text_input, model_choice, tone, state):
 
     if not user_input:
         error_message = "⚠️ No valid input detected. Please upload audio or type a message."
-        # Return None for audio and video to avoid Gradio errors
-        return error_message, None, None, state.get("lead_stage", "initial"), state
+        return error_message, None, None, "initial", state
 
     user_id = state.get("user_id", f"user_{uuid.uuid4().hex[:8]}")
-    if "conversation" not in state:
-        state["conversation"] = []
-        state["lead_stage"] = "initial"
+    state.setdefault("conversation", [])
+    state.setdefault("lead_stage", "initial")
 
     try:
         emotion = emotion_detector.get_primary_emotion(user_input)