Update app.py

app.py

@@ -25,62 +25,59 @@ def summarize_query(query: str) -> str:
 
 search_tool = DuckDuckGoSearchTool()
 
-# --- System Message for ReACT + Scratchpad + Auto-Retry ---
+# --- ReACT + Scratchpad + Retry Prompt ---
 
-system_message = """
+system_prompt = """
 You are a ReACT agent with scratchpad memory and a retry mechanism.
 
-For every question:
-1. Thought: Think what is needed.
-2. Action: (Optional) Use a tool with a clear query.
-3. Observation: Record what tool returned.
+1. Thought: Figure out what's needed.
+2. Action: (Optional) Call a tool with a precise query.
+3. Observation: Record tool output.
 
-If the first Observation is empty or irrelevant:
-4. Thought: The result was unclear. I should reframe and retry.
-5. Action: summarize_query with the original query.
-6. Action: DuckDuckGoSearchTool with the reframed query.
+If the first Observation is empty/irrelevant:
+4. Thought: Unclear result, reframe and retry.
+5. Action: summarize_query(original query).
+6. Action: DuckDuckGoSearchTool(reframed query).
 7. Observation: Record new result.
 
 Then:
-8. Thought: Reflect on all observations.
-9. FINAL ANSWER: Provide the answer.
-
-Formatting Rules:
-- Begin with FINAL ANSWER: [your answer]
-- Numbers: plain (no commas unless list)
-- Strings: no articles unless inside proper names
-- Lists: comma-separated without extra punctuation
+8. Thought: Reflect using all observations.
+9. FINAL ANSWER: Provide your answer.
+
+Formatting:
+- Start with FINAL ANSWER: [your answer]
+- Numbers plain (no commas unless list)
+- Strings no articles unless part of proper names
+- Lists comma-separated, no extra punctuation
 """
 
 # --- Build the Smart Agent ---
 
 smart_agent = CodeAgent(
     tools=[search_tool, summarize_query],
-    model=HfApiModel(),                 # no prompt here
-    system_message=system_message       # moved into CodeAgent
+    model=HfApiModel(),             # no prompt here
+    system_prompt=system_prompt     # prompt passed to CodeAgent
 )
 
-# --- Integrate into Gradio App ---
+# --- Hook into Gradio App ---
 
 class BasicAgent:
     def __init__(self):
-        print("SmolAgent with ReACT, Scratchpad & Retry initialized.")
+        print("SmolAgent (ReACT + Scratchpad + Retry) initialized.")
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        print(f"Q: {question[:50]}...")
         return smart_agent.run(question)
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
-    if profile:
-        username = profile.username
-    else:
-        return "Please log in to Hugging Face using the button above.", None
-
+    if not profile:
+        return "Please log in with Hugging Face.", None
+    username = profile.username
     agent = BasicAgent()
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
 
-    # Fetch questions
+    # 1. Fetch questions
     try:
         resp = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
         resp.raise_for_status()
@@ -90,7 +87,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Error fetching questions: {e}", None
 
-    # Run agent
+    # 2. Run agent
     logs, payload = [], []
     for item in questions:
         tid = item.get("task_id")
@@ -105,19 +102,20 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         payload.append({"task_id": tid, "submitted_answer": ans})
 
     if not payload:
-        return "Agent did not produce any answers to submit.", pd.DataFrame(logs)
+        return "Agent did not produce any answers.", pd.DataFrame(logs)
 
-    # Submit
+    # 3. Submit
     sub = {"username": username, "agent_code": agent_code, "answers": payload}
     try:
-        r = requests.post(f"{DEFAULT_API_URL}/submit", json=sub, timeout=60)
-        r.raise_for_status()
-        res = r.json()
+        post = requests.post(f"{DEFAULT_API_URL}/submit", json=sub, timeout=60)
+        post.raise_for_status()
+        res = post.json()
         status = (
             f"Submission Successful!\n"
             f"User: {res.get('username')}\n"
             f"Score: {res.get('score', 'N/A')}% "
-            f"({res.get('correct_count', '?')}/{res.get('total_attempted', '?')})"
+            f"({res.get('correct_count', '?')}/"
+            f"{res.get('total_attempted', '?')})"
         )
         return status, pd.DataFrame(logs)
     except Exception as e:
@@ -127,10 +125,10 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
 with gr.Blocks() as demo:
     gr.Markdown("# SmolAgent GAIA Evaluation Runner 🚀")
-    gr.Markdown("1. Clone this space  2. Log in  3. Click **Run Evaluation & Submit All Answers**")
+    gr.Markdown("1) Clone this space  2) Log in  3) Run Evaluation & Submit All Answers")
     gr.LoginButton()
     btn = gr.Button("Run Evaluation & Submit All Answers")
-    out_status = gr.Textbox(label="Status", lines=5, interactive=False)
+    out_status = gr.Textbox(label="Run Status", lines=5, interactive=False)
     out_table  = gr.DataFrame(label="Results")
     btn.click(fn=run_and_submit_all, outputs=[out_status, out_table])