Update app.py

app.py

@@ -1,3 +1,4 @@
+# --- app.py ---
 import os
 import gradio as gr
 import requests
@@ -6,6 +7,8 @@ from agent import GaiaAgent
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+agent = GaiaAgent()
+
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
 
@@ -21,13 +24,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     submit_url = f"{api_url}/submit"
 
     try:
-        agent = GaiaAgent()
-    except Exception as e:
-        return f"Error initializing agent: {e}", None
-
-    agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
-
-    try:
+        agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
         response = requests.get(questions_url, timeout=15)
         response.raise_for_status()
         questions_data = response.json()
@@ -45,14 +42,9 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             continue
         try:
             final_answer, trace = agent(question_text)
-
-            print("\n--- QUESTION ---")
-            print(f"Task ID: {task_id}")
-            print(f"Question: {question_text}")
-            print("\n--- REASONING TRACE ---")
-            print(trace)
-            print("\n--- FINAL ANSWER (SUBMITTED) ---")
-            print(final_answer)
+            print(f"\n--- QUESTION ---\nTask ID: {task_id}\nQuestion: {question_text}")
+            print(f"\n--- REASONING TRACE ---\n{trace}")
+            print(f"\n--- FINAL ANSWER (SUBMITTED) ---\n{final_answer}")
 
             answers_payload.append({
                 "task_id": task_id,
@@ -83,25 +75,39 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)\n"
             f"Message: {result_data.get('message', 'No message received.')}"
         )
-        results_df = pd.DataFrame(results_log)
-        return final_status, results_df
+        return final_status, pd.DataFrame(results_log)
     except Exception as e:
         return f"Submission Failed: {e}", pd.DataFrame(results_log)
 
+def evaluate(question, files):
+    uploaded_files = {}
+    if files:
+        for file in files:
+            file_path = file.name
+            file.save(file_path)
+            uploaded_files[file.name] = file_path
 
-with gr.Blocks() as demo:
-    gr.Markdown("# GAIA Agent Submission Interface")
-    gr.Markdown("""
-        Logga in och kör agenten.\n
-        Du behöver INTE en OpenAI API-nyckel längre. Agenten kör en lokal modell.
-    """)
-    gr.LoginButton()
-
-    run_button = gr.Button("Run Evaluation & Submit All Answers")
-    status_output = gr.Textbox(label="Submission Result")
-    results_table = gr.DataFrame(label="Answers")
+    prediction, reasoning = agent(question, uploaded_files)
+    return prediction, reasoning
 
-    run_button.click(fn=run_and_submit_all, outputs=[status_output, results_table])
+with gr.Blocks() as demo:
+    gr.Markdown("# GAIA Agent Interface")
+    gr.Markdown("Logga in och kör agenten på alla frågor eller testa enskilda.")
+
+    with gr.Tab("✅ Run Full Evaluation"):
+        gr.LoginButton()
+        run_button = gr.Button("Run Evaluation & Submit All Answers")
+        status_output = gr.Textbox(label="Submission Result")
+        results_table = gr.DataFrame(label="Answers")
+        run_button.click(fn=run_and_submit_all, outputs=[status_output, results_table])
+
+    with gr.Tab("🔍 Test Manual Question"):
+        question = gr.Textbox(label="Question")
+        files = gr.File(label="Optional files (mp3, wav, xlsx)", file_types=['.mp3', '.wav', '.xlsx'], type="file", file_count="multiple")
+        submit = gr.Button("Run Agent")
+        answer = gr.Textbox(label="Answer")
+        reasoning = gr.Textbox(label="Reasoning Trace")
+        submit.click(fn=evaluate, inputs=[question, files], outputs=[answer, reasoning])
 
 if __name__ == "__main__":
     print("\n" + "-"*30 + " App Starting " + "-"*30)
@@ -119,10 +125,10 @@ if __name__ == "__main__":
         print(f"   Repo URL: https://huggingface.co/spaces/{space_id_startup}")
         print(f"   Repo Tree URL: https://huggingface.co/spaces/{space_id_startup}/tree/main")
     else:
-        print("ℹ️  SPACE_ID environment variable not found (running locally?). Repo URL cannot be determined.")
+        print("ℹ️  SPACE_ID environment variable not found (running locally?).")
 
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
     demo.launch(debug=True, share=False)
-    
+