Update app.py

app.py

@@ -4,10 +4,14 @@ import requests
 import pandas as pd
 import math
 
-from smolagents import CodeAgent, HfApiModel, tool
+from smolagents import CodeAgent, tool
+from smolagents.models import LiteLLMModel
 from duckduckgo_search import DDGS
 
-# --- Constants (DO NOT CHANGE) ---
+# --------------------------------------------------
+# CONSTANTS (DO NOT CHANGE)
+# --------------------------------------------------
+
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --------------------------------------------------
@@ -16,7 +20,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 @tool
 def web_search(query: str) -> str:
-    """Search the web and return short factual text."""
+    """Search the web and return factual text."""
     with DDGS() as ddgs:
         results = list(ddgs.text(query, max_results=3))
     if not results:
@@ -37,13 +41,16 @@ def calculator(expression: str) -> str:
 
 class BasicAgent:
     def __init__(self):
-        model = HfApiModel("meta-llama/Meta-Llama-3-8B-Instruct")
+        model = LiteLLMModel(
+            model_id="huggingface/meta-llama/Meta-Llama-3-8B-Instruct"
+        )
+
         self.agent = CodeAgent(
             tools=[web_search, calculator],
             model=model,
             system_prompt=(
                 "You are a precise AI agent.\n"
-                "Solve the task using tools if needed.\n"
+                "Solve the question using tools if needed.\n"
                 "Return ONLY the final answer.\n"
                 "No explanation. No markdown. No extra words."
             ),
@@ -58,21 +65,21 @@ class BasicAgent:
             return "I don't know"
 
 # --------------------------------------------------
-# MAIN EVALUATION + SUBMISSION LOGIC (DO NOT CHANGE)
+# MAIN EVALUATION + SUBMISSION LOGIC
+# (KEEP AS IS)
 # --------------------------------------------------
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     space_id = os.getenv("SPACE_ID")
 
-    if profile:
-        username = profile.username
-    else:
+    if not profile:
         return "Please login to Hugging Face.", None
 
-    api_url = DEFAULT_API_URL
-    questions_url = f"{api_url}/questions"
-    submit_url = f"{api_url}/submit"
+    username = profile.username
+
+    questions_url = f"{DEFAULT_API_URL}/questions"
+    submit_url = f"{DEFAULT_API_URL}/submit"
 
     # Instantiate agent
     try:
@@ -141,22 +148,23 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         return f"Submission failed: {e}", pd.DataFrame(results_log)
 
 # --------------------------------------------------
-# GRADIO UI (DO NOT CHANGE)
+# GRADIO UI
 # --------------------------------------------------
 
 with gr.Blocks() as demo:
-    gr.Markdown("# GAIA Level-1 Agent Evaluation")
+    gr.Markdown("# GAIA Level-1 Agent – Final Assignment")
 
     gr.Markdown(
         """
         **Instructions**
-        1. Login with Hugging Face
-        2. Click the button to run evaluation
-        3. Wait for submission and score
+        1. Login with your Hugging Face account
+        2. Click the button below
+        3. Wait for the agent to answer all questions and submit
         """
     )
 
     gr.LoginButton()
+
     run_button = gr.Button("Run Evaluation & Submit All Answers")
 
     status_output = gr.Textbox(label="Submission Result", lines=5)