Update app.py

app.py

@@ -3,14 +3,17 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+import tempfile
+from pathlib import Path
 from final_agent import FinalAgent
+from utils import process_file_for_task_v2
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
-# ----- see final_agent.py ------
+# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -69,17 +72,28 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     print(f"Running agent on {len(questions_data)} questions...")
     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:
+        original_question_text = item.get("question")
+        if not task_id or original_question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
+        question_text_for_agent, downloaded_file_path = process_file_for_task_v2(task_id, original_question_text, DEFAULT_API_URL)
+        
+        local_file_to_clean = downloaded_file_path # Use the returned path for cleanup
+
         try:
-            submitted_answer = agent(question_text, task_id=task_id)
+            submitted_answer = agent(question_text_for_agent, task_id=task_id)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
-            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
+            results_log.append({"Task ID": task_id, "Question": original_question_text, "Processed Question": question_text_for_agent, "Submitted Answer": submitted_answer})
         except Exception as e:
             print(f"Error running agent on task {task_id}: {e}")
-            results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
+            results_log.append({"Task ID": task_id, "Question": original_question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
+        finally:
+            if local_file_to_clean and local_file_to_clean.exists():
+                try:
+                    local_file_to_clean.unlink()
+                    print(f"Cleaned up temporary file: {local_file_to_clean}")
+                except OSError as e_clean:
+                    print(f"Error cleaning up temp file {local_file_to_clean}: {e_clean}")
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")
@@ -186,3 +200,4 @@ if __name__ == "__main__":
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
     demo.launch(debug=True, share=False)
+