Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+
 from my_agent import agent
 
 # (Keep Constants as is)
@@ -18,7 +19,7 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         agent_response = agent.run(question)
         print(f"Agent returning fixed answer: {agent_response}")
-        return agent_response
+        return agent_response.strip()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -38,6 +39,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
+    file_url = f"{api_url}/files"
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
@@ -77,6 +79,35 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
+        file_name = item.get("file_name")
+
+        if file_name:
+            # Ensure the documents directory exists
+            documents_dir = "documents"
+            os.makedirs(documents_dir, exist_ok=True)
+
+            # Clear the documents directory
+            for file in os.listdir(documents_dir):
+                file_path = os.path.join(documents_dir, file)
+                if os.path.isfile(file_path):
+                    os.remove(file_path)
+
+            # Download the file
+            file_download_url = f"{file_url}/{task_id}"
+            try:
+                file_response = requests.get(file_download_url, timeout=15)
+                file_response.raise_for_status()
+                file_path = os.path.join(documents_dir, file_name)
+                with open(file_path, "wb") as f:
+                    f.write(file_response.content)
+                print(f"File downloaded and saved to: {file_path}")
+            except requests.exceptions.RequestException as e:
+                print(f"Error downloading file for task {task_id}: {e}")
+                continue
+
+            question_text += f" (File Name: {file_name})"
+
+
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue