Update app.py

app.py

@@ -305,8 +305,11 @@ When citing search results, mention the source, and ensure your answer reflects
                                 if chunk_count <= 5:
                                     print(f"Chunk {chunk_count}: {repr(chunk_content[:50])}")
                                 
-                                # Yield updated history
-                                yield new_history, ""
+                                # Yield less frequently to avoid overwhelming Gradio
+                                # Only yield every 10 chunks or if it's a significant chunk
+                                if chunk_count % 10 == 0 or len(chunk_content) > 50:
+                                    yield new_history, ""
+                                    time.sleep(0.01)  # Small delay to prevent connection errors
                         
                     except json.JSONDecodeError as e:
                         print(f"JSON decode error: {e}")
@@ -320,6 +323,7 @@ When citing search results, mention the source, and ensure your answer reflects
                 new_history[-1] = (message, full_response)
             
             print(f"Final response length: {len(full_response)}")
+            # Always yield the final state
             yield new_history, ""
             
         except requests.exceptions.RequestException as e:
@@ -338,7 +342,12 @@ When citing search results, mention the source, and ensure your answer reflects
             new_history[-1] = (message, search_info + error_msg)
             yield new_history, error_msg
     
-    # Create Gradio interface
+    # Test function without streaming
+    def test_simple_response(message, history):
+        """Simple test function to verify UI is working"""
+        print(f"Test function called with message: {message}")
+        new_history = history + [(message, f"Echo: {message}")]
+        return new_history, ""
     with gr.Blocks(theme="soft", fill_height=True) as demo:
         # Header section
         gr.Markdown(
@@ -395,6 +404,7 @@ When citing search results, mention the source, and ensure your answer reflects
                 # Clear conversation button
                 with gr.Row():
                     clear = gr.ClearButton([msg, chatbot], value="🧹 Clear Conversation")
+                    test_btn = gr.Button("🧪 Test Echo", variant="secondary")
                     
                 # Example queries
                 gr.Examples(
@@ -410,6 +420,13 @@ When citing search results, mention the source, and ensure your answer reflects
                 error_box = gr.Markdown("")
         
         # Connect buttons to functions
+        # Test button for debugging
+        test_btn.click(
+            test_simple_response,
+            inputs=[msg, chatbot],
+            outputs=[chatbot, error_box]
+        )
+        
         submit.click(
             query_deepseek_streaming,
             inputs=[msg, chatbot, use_deep_research],
@@ -438,5 +455,12 @@ When citing search results, mention the source, and ensure your answer reflects
 # Run interface
 if __name__ == "__main__":
     demo = create_deepseek_interface()
-    demo.queue()  # Enable queue explicitly
-    demo.launch(debug=True, share=False, server_name="0.0.0.0", server_port=7860, ssr_mode=False)
+    demo.queue(max_size=10, concurrency_count=1)  # Configure queue with limits
+    demo.launch(
+        debug=True, 
+        share=False, 
+        server_name="0.0.0.0", 
+        server_port=7860, 
+        ssr_mode=False,
+        show_error=True  # Show detailed errors
+    )