Update app.py

app.py

@@ -1,57 +1,72 @@
-import os
-import gradio as gr
-import requests
-import inspect
-import pandas as pd
-
-# (Keep Constants as is)
-# --- Constants ---
-DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
+from openai import OpenAI
+from ddgs import DDGS
 import os
 
 class BasicAgent:
     def __init__(self):
-        print("Final Agent initialized.")
-
-        hf_token = os.getenv("HF_TOKEN")
-        print("TOKEN EXISTS:", hf_token is not None)
-
-        self.model = InferenceClientModel(
-            model_id="HuggingFaceH4/zephyr-7b-beta",
-            provider="hf-inference",
-            token=hf_token
-        )
+        print("OpenAI Agent initialized.")
+        self.client = OpenAI(api_key=os.getenv("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"))
 
-        self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool()],
-            model=self.model,
-            add_base_tools=True,
-            max_steps=6
-        )
+    def web_search(self, query, max_results=5):
+        try:
+            with DDGS() as ddgs:
+                results = list(ddgs.text(query, max_results=max_results))
+            return "\n".join([
+                f"{r.get('title')}: {r.get('body')}"
+                for r in results
+            ])
+        except Exception as e:
+            print("SEARCH ERROR:", e)
+            return ""
 
     def clean(self, ans):
         ans = str(ans).strip()
         for x in ["FINAL ANSWER:", "Final Answer:", "Answer:"]:
             ans = ans.replace(x, "")
+        ans = ans.strip()
         return ans.split("\n")[0].strip()
 
     def __call__(self, question: str) -> str:
+        print("Question:", question[:100])
+
+        context = self.web_search(question)
+
         prompt = f"""
-Return ONLY the final answer.
-No explanation. No extra words.
+Return ONLY the exact final answer.
+No explanation.
+No extra words.
+Do not write FINAL ANSWER.
 
 Question:
 {question}
+
+Search context:
+{context}
 """
+
         try:
-            ans = self.agent.run(prompt)
-            return self.clean(ans)
+            response = self.client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=[
+                    {
+                        "role": "system",
+                        "content": "You solve exact-match benchmark questions. Return only the final answer."
+                    },
+                    {
+                        "role": "user",
+                        "content": prompt
+                    }
+                ],
+                temperature=0
+            )
+
+            answer = response.choices[0].message.content
+            final = self.clean(answer)
+            print("Submitted:", final)
+            return final
+
         except Exception as e:
-            print("ERROR:", e)
+            print("AGENT ERROR:", e)
             return "unknown"
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):