Update app.py

app.py

@@ -3,138 +3,35 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from typing import Dict, Any, List
-from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, Tool
-
+from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, WikipediaSearchTool
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Wikipedia Tool ---
-from smolagents import Tool
-
-class WikipediaTool(Tool):
-    name = "wikipedia_search"
-    description = "Searches Wikipedia and returns concise extracts from top results."
-    output_type = "string"  # REQUIRED by smolagents
-
-    inputs = {
-        "query": {
-            "type": "string",
-            "description": "The search query to look up on Wikipedia."
-        }
-    }
-
-    def __init__(self, top_k: int = 3, language: str = "en"):
-        super().__init__()
-        self.top_k = max(1, min(top_k, 5))
-        self.api = f"https://{language}.wikipedia.org/w/api.php"
-
-    def forward(self, query: str) -> str:
-        import requests
-        try:
-            # search
-            r = requests.get(self.api, params={
-                "action": "query",
-                "list": "search",
-                "srsearch": query,
-                "srlimit": self.top_k,
-                "utf8": "1",
-                "format": "json",
-            }, timeout=10)
-            r.raise_for_status()
-            hits = r.json().get("query", {}).get("search", [])
-            pageids = [h.get("pageid") for h in hits if isinstance(h.get("pageid"), int)]
-            if not pageids:
-                return "No relevant Wikipedia snippets found."
-
-            # extracts
-            r2 = requests.get(self.api, params={
-                "action": "query",
-                "prop": "extracts",
-                "explaintext": "1",
-                "exintro": "1",
-                "pageids": "|".join(str(pid) for pid in pageids),
-                "format": "json",
-                "utf8": "1",
-            }, timeout=10)
-            r2.raise_for_status()
-            pages = r2.json().get("query", {}).get("pages", {})
-
-            snippets = []
-            for h in hits:
-                pid = h.get("pageid")
-                meta = pages.get(str(pid)) or pages.get(pid)
-                text = (meta or {}).get("extract", "") if pid is not None else ""
-                title = h.get("title", "")
-                if text and title:
-                    sentences = text.split(". ")
-                    preview = ". ".join(sentences[:2]).strip()
-                    if preview:
-                        snippets.append(f"{title}: {preview}")
-            return "\n".join(snippets) if snippets else "No relevant Wikipedia snippets found."
-        except Exception as e:
-            return f"Wikipedia error: {e}"
 
 # --- Basic Agent Definition ---
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
+        # Initialize the model
+        model = OpenAIServerModel(model_id="gpt-4o-mini")
 
-        # Provider selection (force away from OpenAI if quota-exceeded)
-        force = (os.getenv("FORCE_PROVIDER") or "").strip().lower()
-        openrouter_key = os.getenv("OPENROUTER_API_KEY")
-        together_key = os.getenv("TOGETHER_API_KEY")
-
-        provider = None
-        model_id = None
-        if force == "openrouter" or (openrouter_key and force != "together"):
-            os.environ["OPENAI_API_KEY"] = openrouter_key
-            os.environ["OPENAI_API_BASE"] = "https://openrouter.ai/api/v1"
-            os.environ["OPENAI_BASE_URL"] = "https://openrouter.ai/api/v1"
-            provider = "openrouter"
-            model_id = os.getenv("OPENROUTER_MODEL", "meta-llama/Meta-Llama-3-8B-Instruct")
-        elif force == "together" or together_key:
-            os.environ["OPENAI_API_KEY"] = together_key
-            os.environ["OPENAI_API_BASE"] = "https://api.together.xyz/v1"
-            os.environ["OPENAI_BASE_URL"] = "https://api.together.xyz/v1"
-            provider = "together"
-            model_id = os.getenv("TOGETHER_MODEL", "meta-llama/Meta-Llama-3-8B-Instruct")
-        else:
-            # No non-OpenAI key available → use Wikipedia-only fallback
-            self.wikipedia_only = True
-            wiki_top_k = int(os.getenv("WIKI_TOP_K", "3"))
-            wiki_lang = os.getenv("WIKI_LANG", "en")
-            self.wikipedia_tool = WikipediaTool(top_k=wiki_top_k, language=wiki_lang)
-            print("No non-OpenAI key set; running in Wikipedia-only mode.")
-            return
-
-        print(f"Using provider={provider}, model_id={model_id}")
-        model = OpenAIServerModel(model_id=model_id)
-
-        search_tool = DuckDuckGoSearchTool()
-        wiki_top_k = int(os.getenv("WIKI_TOP_K", "3"))
-        wiki_lang = os.getenv("WIKI_LANG", "en")
-        wikipedia_tool = WikipediaTool(top_k=wiki_top_k, language=wiki_lang)
+        # Initialize search tools
+        duckduckgo = DuckDuckGoSearchTool()
+        wikipedia = WikipediaSearchTool()
 
-        self.agent = CodeAgent(model=model, tools=[search_tool, wikipedia_tool])
-        self.wikipedia_only = False
+        # Initialize Agent with both tools
+        self.agent = CodeAgent(
+            model=model,
+            tools=[duckduckgo, wikipedia]
+        )
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        if self.wikipedia_only:
-            # Direct Wikipedia fallback
-            return self.wikipedia_tool.forward(question)
+        fixed_answer = self.agent.run(question)
+        print(f"Agent returning fixed answer: {fixed_answer}")
+        return fixed_answer
 
-        try:
-            return self.agent.run(question)
-        except Exception as e:
-            # If provider throws (e.g., 429/insufficient_quota), fall back to Wikipedia
-            print(f"LLM error, falling back to Wikipedia: {e}")
-            try:
-                return self.agent.tools[1].forward(question)  # WikipediaTool as second tool
-            except Exception:
-                return "I cannot determine the answer."
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -279,6 +176,7 @@ with gr.Blocks() as demo:
     run_button = gr.Button("Run Evaluation & Submit All Answers")
 
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
+    # Removed max_rows=10 from DataFrame constructor
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
 
     run_button.click(