Update app.py

app.py

@@ -420,27 +420,6 @@ def clear_cache():
     processing_status = {"is_processing": False, "progress": 0, "total": 0}
     return "Cache cleared successfully.", None
 
-def test_single_question(question: str, model_choice: str):
-    """Test the agent with a single question."""
-    if not question.strip():
-        return "Please enter a question to test."
-    
-    # Map model choice to actual model name
-    model_map = {
-        "Llama 3.1 8B": "meta-llama/Llama-3.1-8B-Instruct",
-        "Llama 3.1 70B": "meta-llama/Llama-3.1-70B-Instruct",
-        "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
-        "CodeLlama 7B": "codellama/CodeLlama-7b-Instruct-hf"
-    }
-    
-    selected_model = model_map.get(model_choice, "meta-llama/Llama-3.1-8B-Instruct")
-    
-    try:
-        agent = IntelligentAgent(debug=True, model_name=selected_model)
-        answer = agent(question)
-        return f"**Answer using {model_choice}:**\n\n{answer}"
-    except Exception as e:
-        return f"Error testing question: {e}"
 
 # --- Enhanced Gradio Interface ---
 with gr.Blocks(title="Intelligent Agent with Conditional Search") as demo:
@@ -451,29 +430,6 @@ with gr.Blocks(title="Intelligent Agent with Conditional Search") as demo:
         gr.LoginButton()
         clear_btn = gr.Button("Clear Cache", variant="secondary")
 
-    with gr.Tab("Test Single Question"):
-        gr.Markdown("### Test the Agent with a Single Question")
-        with gr.Row():
-            test_question = gr.Textbox(
-                label="Enter your question",
-                placeholder="e.g., What is the current weather in Paris? or What is machine learning?",
-                lines=2
-            )
-            test_model = gr.Dropdown(
-                choices=["Llama 3.1 8B", "Llama 3.1 70B", "Mistral 7B", "CodeLlama 7B"],
-                value="Llama 3.1 8B",
-                label="Select Model"
-            )
-        
-        test_btn = gr.Button("Test Question", variant="primary")
-        test_result = gr.Textbox(label="Agent Response", lines=10, interactive=False)
-        
-        test_btn.click(
-            fn=test_single_question,
-            inputs=[test_question, test_model],
-            outputs=[test_result]
-        )
-
     with gr.Tab("Step 1: Fetch Questions"):
         gr.Markdown("### Fetch Questions from API")
         fetch_btn = gr.Button("Fetch Questions", variant="primary")