Update app.py

app.py

@@ -8,13 +8,8 @@ import gradio as gr
 # ===============================
 
 BASE_URL = os.environ.get("GAIA_API_URL")
-
-# Your username (auto-detected, fallback added)
 HF_USERNAME = os.environ.get("SPACE_AUTHOR_NAME", "jatinror")
-
-# Your Space ID hardcoded as fallback
 SPACE_ID = os.environ.get("SPACE_ID", "jatinror/Final_Assignment_Template")
-
 AGENT_CODE_URL = f"https://huggingface.co/spaces/{SPACE_ID}/tree/main"
 
 print("Using SPACE_ID:", SPACE_ID)
@@ -38,7 +33,6 @@ def web_search(query, max_results=3):
 def download_task_file(task_id):
     url = f"{BASE_URL}/files/{task_id}"
     response = requests.get(url)
-
     if response.status_code == 200:
         file_path = f"/tmp/{task_id}"
         with open(file_path, "wb") as f:
@@ -47,14 +41,12 @@ def download_task_file(task_id):
     return None
 
 # ===============================
-# BASIC REASONING (LEVEL-1 READY)
+# BASIC REASONING
 # ===============================
 
 def solve_question(question, task_id):
     file_path = download_task_file(task_id)
-
     context = ""
-
     if file_path and os.path.exists(file_path):
         try:
             with open(file_path, "r", errors="ignore") as f:
@@ -63,22 +55,17 @@ def solve_question(question, task_id):
             context = ""
     else:
         context = web_search(question)
-
     return extract_answer(context)
 
 # ===============================
-# ANSWER EXTRACTION (SHORT ONLY!)
+# ANSWER EXTRACTION
 # ===============================
 
 def extract_answer(text):
     import re
-
-    # Try returning first number (many GAIA answers are numeric)
     numbers = re.findall(r"\b\d+(?:\.\d+)?\b", text)
     if numbers:
         return numbers[0]
-
-    # Otherwise return short phrase
     words = text.split()
     return " ".join(words[:6]).strip()
 
@@ -101,7 +88,6 @@ def submit_answers(answers):
         "agent_code": AGENT_CODE_URL,
         "answers": answers
     }
-
     print("Submitting payload...")
     response = requests.post(f"{BASE_URL}/submit", json=payload)
     print("Server response:", response.text)
@@ -113,46 +99,41 @@ def submit_answers(answers):
 def run_agent():
     print("Fetching GAIA questions...")
     questions = get_questions()
-
     answers = []
-
     for q in questions:
         task_id = q["task_id"]
         question = q["question"]
-
         print("Solving:", task_id)
-
         try:
             result = solve_question(question, task_id)
         except Exception as e:
             print("Error:", e)
             result = ""
-
         answers.append({
             "task_id": task_id,
             "submitted_answer": result.strip()
         })
-
     submit_answers(answers)
     print("Finished submission.")
 
 # ===============================
-# GRADIO UI (Required for Run Button)
+# GRADIO INTERFACE (GUARANTEES RUN BUTTON)
 # ===============================
 
-def run_pipeline():
+def run_pipeline(dummy_button):
     try:
         run_agent()
         return "✅ GAIA submission completed. Check leaderboard."
     except Exception as e:
         return f"❌ Error occurred: {str(e)}"
 
+# Use a dummy button as input to force Run button to appear
 demo = gr.Interface(
     fn=run_pipeline,
-    inputs=None,
+    inputs=gr.Button("Run GAIA Agent"),  # <- this creates the Run button
     outputs="text",
     title="GAIA Final Assignment Agent",
-    description="Click Run to execute your GAIA agent and submit answers."
+    description="Click the button to execute your GAIA agent and submit answers."
 )
 
 if __name__ == "__main__":