Update app.py

app.py

@@ -223,7 +223,7 @@ def generate_speech_with_gpu(text: str, voice_name: str = 'af', tts_model=TTS_MO
         print(f"Error generating speech: {str(e)}")
         return None
 
-def process_query(query: str, history: List[List[str]], selected_voice: str = 'af') -> Dict[str, Any]:
+def process_query(query: str, history: List[List[str]], selected_voice: str = 'af'):
     """Process user query with streaming effect"""
     try:
         if history is None:
@@ -235,13 +235,14 @@ def process_query(query: str, history: List[List[str]], selected_voice: str = 'a
 
         current_history = history + [[query, "*Searching...*"]]
 
-        yield {
-            'answer_output': gr.Markdown("*Searching & Thinking...*"),
-            'sources_output': gr.HTML(sources_html),
-            'search_btn': gr.Button("Searching...", interactive=False),
-            'chat_history_display': current_history,
-            'audio_output': None
-        }
+        # Yield initial searching state
+        yield (
+            "*Searching & Thinking...*",  # answer_output (Markdown)
+            sources_html,                  # sources_output (HTML)
+            "Searching...",               # search_btn (Button)
+            current_history,              # chat_history_display (Chatbot)
+            None                          # audio_output (Audio)
+        )
 
         # Generate answer
         prompt = format_prompt(query, web_results)
@@ -253,13 +254,13 @@ def process_query(query: str, history: List[List[str]], selected_voice: str = 'a
 
         # Generate speech from the answer (only if enabled)
         if TTS_ENABLED:
-            yield {
-                'answer_output': gr.Markdown(final_answer),
-                'sources_output': gr.HTML(sources_html),
-                'search_btn': gr.Button("Generating audio...", interactive=False),
-                'chat_history_display': updated_history,
-                'audio_output': None
-            }
+            yield (
+                final_answer,              # answer_output
+                sources_html,              # sources_output
+                "Generating audio...",     # search_btn
+                updated_history,           # chat_history_display
+                None                       # audio_output
+            )
             try:
                 audio = generate_speech_with_gpu(final_answer, selected_voice)
                 if audio is None:
@@ -271,25 +272,26 @@ def process_query(query: str, history: List[List[str]], selected_voice: str = 'a
             final_answer += "\n\n*TTS is disabled. Audio not available.*"
             audio = None
 
-        yield {
-            'answer_output': gr.Markdown(final_answer),
-            'sources_output': gr.HTML(sources_html),
-            'search_btn': gr.Button("Search", interactive=True),
-            'chat_history_display': updated_history,
-            'audio_output': audio if audio is not None else gr.Audio(value=None)
-        }
+        # Yield final result
+        yield (
+            final_answer,              # answer_output
+            sources_html,              # sources_output
+            "Search",                  # search_btn
+            updated_history,           # chat_history_display
+            audio if audio is not None else None  # audio_output
+        )
 
     except Exception as e:
         error_message = str(e)
         if "GPU quota" in error_message:
             error_message = "⚠️ GPU quota exceeded. Please try again later when the daily quota resets."
-        yield {
-            'answer_output': gr.Markdown(f"Error: {error_message}"),
-            'sources_output': gr.HTML(sources_html),
-            'search_btn': gr.Button("Search", interactive=True),
-            'chat_history_display': history + [[query, f"*Error: {error_message}*"]],
-            'audio_output': None
-        }
+        yield (
+            f"Error: {error_message}",  # answer_output
+            sources_html,               # sources_output
+            "Search",                   # search_btn
+            history + [[query, f"*Error: {error_message}*"]],  # chat_history_display
+            None                        # audio_output
+        )
 
 # Update the CSS for better contrast and readability
 css = """