Update app.py

app.py

@@ -7,56 +7,54 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class BasicAgent:
     def __init__(self):
-        self.api_url = "https://router.huggingface.co/v1/chat/completions"
-
         hf_token = os.getenv("HF_TOKEN")
 
         if not hf_token:
             raise ValueError("HF_TOKEN is missing")
 
         self.headers = {
-            "Authorization": f"Bearer {hf_token}",
-            "Content-Type": "application/json"
+            "Authorization": f"Bearer {hf_token}"
         }
 
-        self.model = "meta-llama/Llama-3.1-8B-Instruct"
+        self.model_url = (
+            "https://api-inference.huggingface.co/models/"
+            "meta-llama/Llama-3.1-8B-Instruct"
+        )
 
         print("BasicAgent initialized.")
 
     def __call__(self, question: str) -> str:
         try:
             payload = {
-                "model": self.model,
-                "messages": [
-                    {
-                        "role": "system",
-                        "content": (
-                            "You are a GAIA benchmark solving agent. "
-                            "Solve carefully and return only the exact final answer. "
-                            "Do not explain anything. "
-                            "Do not add extra text."
-                        )
-                    },
-                    {
-                        "role": "user",
-                        "content": question
-                    }
-                ],
-                "temperature": 0
+                "inputs": (
+                    "You are a GAIA solving agent.\n"
+                    "Return only the exact final answer.\n"
+                    "Do not explain anything.\n\n"
+                    f"Question: {question}"
+                ),
+                "parameters": {
+                    "max_new_tokens": 128,
+                    "temperature": 0
+                }
             }
 
             response = requests.post(
-                self.api_url,
+                self.model_url,
                 headers=self.headers,
                 json=payload,
-                timeout=60
+                timeout=120
             )
 
             response.raise_for_status()
 
             result = response.json()
 
-            answer = result["choices"][0]["message"]["content"].strip()
+            if isinstance(result, list):
+                answer = result[0]["generated_text"]
+            else:
+                answer = str(result)
+
+            answer = answer.strip()
 
             print(f"Answer: {answer}")
 
@@ -71,13 +69,11 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     if profile:
         username = profile.username
-        print(f"User logged in: {username}")
     else:
-        return "Please Login to Hugging Face with the button.", None
+        return "Please Login to Hugging Face.", None
 
-    api_url = DEFAULT_API_URL
-    questions_url = f"{api_url}/questions"
-    submit_url = f"{api_url}/submit"
+    questions_url = f"{DEFAULT_API_URL}/questions"
+    submit_url = f"{DEFAULT_API_URL}/submit"
 
     try:
         agent = BasicAgent()
@@ -106,35 +102,22 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         if not task_id or not question_text:
             continue
 
-        try:
-            submitted_answer = agent(question_text)
-
-            answers_payload.append(
-                {
-                    "task_id": task_id,
-                    "submitted_answer": submitted_answer
-                }
-            )
-
-            results_log.append(
-                {
-                    "Task ID": task_id,
-                    "Question": question_text,
-                    "Submitted Answer": submitted_answer
-                }
-            )
+        submitted_answer = agent(question_text)
 
-        except Exception as e:
-            results_log.append(
-                {
-                    "Task ID": task_id,
-                    "Question": question_text,
-                    "Submitted Answer": f"ERROR: {e}"
-                }
-            )
+        answers_payload.append(
+            {
+                "task_id": task_id,
+                "submitted_answer": submitted_answer
+            }
+        )
 
-    if not answers_payload:
-        return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
+        results_log.append(
+            {
+                "Task ID": task_id,
+                "Question": question_text,
+                "Submitted Answer": submitted_answer
+            }
+        )
 
     submission_data = {
         "username": username.strip(),
@@ -175,7 +158,7 @@ with gr.Blocks() as demo:
 
     gr.LoginButton()
 
-    run_button = gr.Button("Run Evaluation & Submit All Answers")
+    run_button = gr.Button("Run Evaluation & Submit")
 
     status_output = gr.Textbox(
         label="Submission Result",
@@ -184,7 +167,7 @@ with gr.Blocks() as demo:
     )
 
     results_table = gr.DataFrame(
-        label="Questions and Agent Answers",
+        label="Results",
         wrap=True
     )
 
@@ -194,5 +177,5 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True)