Update app.py

app.py

@@ -617,6 +617,17 @@ AVATAR_HTML = """
 </div>
 """
 
+def _init_state(memory, lang, history):
+    """Ensure state values are never None."""
+    if memory is None:
+        memory = ConversationMemory()
+    if lang is None:
+        lang = ""
+    if history is None:
+        history = []
+    return memory, lang, history
+
+
 def build_app():
     # Pre-load welcome audio once
     print("⏳ Generating welcome audio…")
@@ -626,9 +637,10 @@ def build_app():
     with gr.Blocks(css=CSS, title="Reem — Virtual Call Centre") as demo:
 
         # ── State ──────────────────────────────────────────────────────
-        memory_state   = gr.State(ConversationMemory())
-        lang_state     = gr.State("")          # "" = not selected yet
-        chat_history   = gr.State([])
+        # Use None defaults — Gradio cannot hash complex objects
+        memory_state   = gr.State(None)
+        lang_state     = gr.State(None)
+        chat_history   = gr.State(None)
 
         # ── Hero ───────────────────────────────────────────────────────
         gr.HTML("""
@@ -761,21 +773,22 @@ def build_app():
 
         # ── Language selection helpers ─────────────────────────────────
         def _set_language(code, memory):
+            memory, _, _ = _init_state(memory, None, None)
             confirm_text = LANGUAGE_CONFIRMATION[code]
             audio_path = tts_to_file(confirm_text, code)
             lang_names = {"en": "English", "ar": "العربية", "ur": "اردو"}
             badge = f"🗣️ Language: **{lang_names.get(code, code)}**"
-            # Return: lang_state, hide lang_group, show chat_group, badge, audio, memory
             return (
                 code,
-                gr.update(visible=False),   # lang_group
-                gr.update(visible=True),    # chat_group
+                gr.update(visible=False),
+                gr.update(visible=True),
                 badge,
                 audio_path,
                 memory,
             )
 
         def handle_lang_mic(audio, memory):
+            memory, _, _ = _init_state(memory, None, None)
             if audio is None:
                 return ("", gr.update(), gr.update(), "", None, memory)
             sr, data = audio
@@ -816,6 +829,7 @@ def build_app():
 
         # ── Voice input ────────────────────────────────────────────────
         def handle_voice(audio, lang_code, memory, history):
+            memory, lang_code, history = _init_state(memory, lang_code, history)
             if audio is None:
                 return history, None, memory, ""
             sr, data = audio
@@ -829,7 +843,7 @@ def build_app():
             if not transcript or len(transcript.strip()) < 3:
                 return history, None, memory, "⚠️ No clear speech detected."
             _, new_history, audio_path = process_query(
-                transcript, memory, lang_code, history)
+                transcript, memory, lang_code or "en", history)
             mem_info_txt = _mem_info(memory)
             return new_history, audio_path, memory, mem_info_txt
 
@@ -841,10 +855,11 @@ def build_app():
 
         # ── Text input ────────────────────────────────────────────────
         def handle_text(text, lang_code, memory, history):
+            memory, lang_code, history = _init_state(memory, lang_code, history)
             if not text or not text.strip():
                 return history, None, memory, "", ""
             _, new_history, audio_path = process_query(
-                text, memory, lang_code, history)
+                text, memory, lang_code or "en", history)
             mem_info_txt = _mem_info(memory)
             return new_history, audio_path, memory, mem_info_txt, ""