Updated the code to use less time to run

app.py

@@ -1,4 +1,5 @@
 import os
+import signal
 import gradio as gr
 import requests
 import pandas as pd
@@ -12,6 +13,20 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
+# =========================
+# Timeout Utilities
+# =========================
+class TimeoutException(Exception):
+    pass
+
+
+def timeout_handler(signum, frame):
+    raise TimeoutException("Agent timed out")
+
+
+signal.signal(signal.SIGALRM, timeout_handler)
+
+
 # =========================
 # Basic Agent Definition
 # =========================
@@ -19,31 +34,30 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
 
-        # Get OpenAI API key from environment (HF Secrets)
         OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
         if not OPENAI_API_KEY:
             raise ValueError(
-                "OPENAI_API_KEY environment variable not set. "
+                "OPENAI_API_KEY not set. "
                 "Add it in Hugging Face → Settings → Secrets."
             )
 
-        # Initialize OpenAI model
         model = OpenAIServerModel(
-            model_id="gpt-4o-mini",   # You can switch to "gpt-4o" if you want
+            model_id="gpt-4o-mini",   # fast + cheap + sufficient
             api_key=OPENAI_API_KEY
         )
 
-        # Initialize tools
         search_tool = DuckDuckGoSearchTool()
 
-        # Initialize agent
+        # IMPORTANT: step limit prevents infinite loops
         self.agent = CodeAgent(
             model=model,
-            tools=[search_tool]
+            tools=[search_tool],
+            max_steps=3,
+            verbosity=0
         )
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question: {question[:50]}...")
+        print(f"Agent received question: {question[:60]}...")
         return self.agent.run(question)
 
 
@@ -53,12 +67,12 @@ class BasicAgent:
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
 
-    if profile:
-        username = profile.username
-        print(f"User logged in: {username}")
-    else:
+    if not profile:
         return "Please login to Hugging Face first.", None
 
+    username = profile.username
+    print(f"User logged in: {username}")
+
     questions_url = f"{DEFAULT_API_URL}/questions"
     submit_url = f"{DEFAULT_API_URL}/submit"
 
@@ -81,18 +95,21 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
 
-    # Run agent
+    # Run agent on each question
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
 
-        if not task_id or question_text is None:
+        if not task_id or not question_text:
             continue
 
         try:
+            signal.alarm(40)  # ⏱ max 40 seconds per question
             answer = agent(question_text)
         except Exception as e:
             answer = f"AGENT ERROR: {e}"
+        finally:
+            signal.alarm(0)
 
         answers_payload.append(
             {"task_id": task_id, "submitted_answer": answer}
@@ -147,16 +164,18 @@ with gr.Blocks() as demo:
         **Instructions**
         1. Login with your Hugging Face account.
         2. Click **Run Evaluation & Submit All Answers**.
-        3. Wait while the agent answers and submits.
+        3. Wait for the agent to finish (this can take a few minutes).
 
         ---
-        **Note:**  
-        This app uses **OpenAI (gpt-4o-mini)** and requires an `OPENAI_API_KEY`
-        set in **Hugging Face → Settings → Secrets**.
+        **Requirements**
+        - Uses **OpenAI (gpt-4o-mini)**
+        - Requires `OPENAI_API_KEY` in HF Space Secrets
+        - Agent is step-limited and timeout-safe
         """
     )
 
     gr.LoginButton()
+
     run_button = gr.Button("Run Evaluation & Submit All Answers")
 
     status_output = gr.Textbox(
@@ -177,7 +196,7 @@ with gr.Blocks() as demo:
 
 
 # =========================
-# App Entry Point
+# Entry Point
 # =========================
 if __name__ == "__main__":
     print("Launching Gradio app...")