update

app.py

@@ -285,13 +285,17 @@ def get_wikipedia_section_tables(
 def download_file(question_id: str, file_name: str) -> str:
     """
     A tool that downloads file that was mentioned in a question and store it as local file.
-    Note that user still need to read file from local file path returned from the function. This function DOES NOT return file content.
+    Returns a JSON string containing the file path and optionally the text content if the file has a text MIME type.
+
     Args:
         question_id: Question ID.
         file_name: File name.
     Returns:
-        str: Local path where the file was stored.
+        str: JSON string containing file information. Structure:
+             - For text files: {"path": "local_path", "content": "file_content"}
+             - For non-text files: {"path": "local_path"}
     """
+    import json
     import os
 
     import requests
@@ -317,15 +321,41 @@ def download_file(question_id: str, file_name: str) -> str:
         # Use original filename directly
         local_path = file_name
 
-        # Save the text file locally
+        # Save the file locally
         with open(local_path, "wb") as f:
             f.write(response.content)
 
         print(f"File saved to: {local_path}")
-        return local_path
+
+        # Check if the file has a text MIME type
+        text_mime_types = [
+            "text/",
+            "application/json",
+            "application/xml",
+            "application/javascript",
+            "application/csv",
+            "application/x-csv",
+            "text/csv",
+        ]
+
+        is_text_file = any(
+            content_type.startswith(mime_type) for mime_type in text_mime_types
+        )
+
+        result = {"path": local_path}
+
+        if is_text_file:
+            # Decode response content directly as text using UTF-8
+            text_content = response.content.decode("utf-8")
+            result["content"] = text_content
+            print(
+                f"Added text content to result (length: {len(text_content)} characters)"
+            )
+
+        return json.dumps(result)
 
     except requests.exceptions.RequestException as e:
-        raise ValueError(f"Failed to text file from {url}: {e}")
+        raise ValueError(f"Failed to download file from {url}: {e}")
     except Exception as e:
         # Print first 200 characters of response content for debugging
         content_preview = (
@@ -334,7 +364,7 @@ def download_file(question_id: str, file_name: str) -> str:
             else b"No response"
         )
         print(f"Error downloading file. Content preview: {content_preview}")
-        raise ValueError(f"Failed to download text file from {url}: {e}")
+        raise ValueError(f"Failed to download file from {url}: {e}")
 
 
 # --- Basic Agent Definition ---
@@ -373,7 +403,6 @@ class BasicAgent:
                 "pytubefix",
                 "pytubefix.cli",
                 "youtube_transcript_api",
-                "open",
             ],
             name="multimodal_agent",
             description="""
@@ -415,7 +444,6 @@ class BasicAgent:
                 "pytubefix",
                 "pytubefix.cli",
                 "youtube_transcript_api",
-                "open",
             ],
             planning_interval=2,
             max_steps=10,