update file handling

app.py

@@ -66,52 +66,24 @@ class MyAgent(TypedDict):
 
 
 # =========================
-# File Handling Functions
+# Simplified File Handling
 # =========================
 def process_question_with_files(question_data: dict) -> str:
+    """
+    Simple file handling - just pass the file info to the agent
+    and let it use its tools to handle the file.
+    """
     question_text = question_data.get('question', '')
     file_name = question_data.get('file_name', '')
 
     if not file_name:
         return question_text
 
-    print(f"📎 Processing question with attached file: {file_name}")
-    try:
-        # Download the file
-        file_url = f"{DEFAULT_API_URL}/files/{file_name}"
-        local_file_path = f"/tmp/{file_name}"
-        download_result = download_file(file_url, local_file_path)
-        if "Failed to download" in download_result:
-            return f"{question_text}\n\n[Note: Could not download attached file {file_name}]"
-
-        # Route based on file type
-        ext = file_name.lower().split('.')[-1]
-        if ext in ['mp3', 'wav', 'm4a', 'flac', 'ogg']:
-            result = audio_processing_tool.invoke(local_file_path)
-            tag = "Audio Transcription"
-        elif ext in ['png', 'jpg', 'jpeg', 'gif', 'bmp']:
-            result = image_recognition_tool.invoke(local_file_path)
-            tag = "Image Analysis"
-        elif ext in ['csv', 'xls', 'xlsx']:
-            result = read_file_tool.invoke(local_file_path)
-            tag = "Spreadsheet Content"
-        elif ext in ['txt', 'md', 'py', 'json']:
-            result = read_file_tool.invoke(local_file_path)
-            tag = "File Content"
-        else:
-            result = read_file_tool.invoke(local_file_path)
-            tag = "File Content"
-
-        # Clean up
-        try:
-            os.remove(local_file_path)
-        except Exception:
-            pass
-
-        return f"{question_text}\n\n[{tag}: {result}]"
+    # Simple approach: just tell the agent about the file
+    # Let the agent use its tools to download and process the file
+    file_url = f"{DEFAULT_API_URL}/files/{file_name}"
 
-    except Exception as e:
-        return f"{question_text}\n\n[Note: Error processing attached file {file_name}: {str(e)}]"
+    return f"{question_text}\n\n[There is an attached file: {file_name}. You can download it from: {file_url}]"
 
 
 def extract_final_answer(text: str) -> str:
@@ -166,10 +138,22 @@ def assistant(state: MyAgent):
     # Add system message to instruct the agent to use the tool
     system_message = SystemMessage(content="""
 You are a helpful assistant tasked with answering questions using a set of tools. 
-Now, I will ask you a question. Think step by step and report your answer with the following template: 
+
+IMPORTANT: When a question mentions an attached file, you should:
+1. Use the download_file_from_url_tool to download the file
+2. Use the appropriate tool to analyze the file content:
+   - For images: use image_recognition_tool or extract_text_from_image_tool
+   - For audio: use audio_processing_tool
+   - For spreadsheets: use analyze_csv_file_tool or analyze_excel_file_tool
+   - For text files: use read_file_tool
+   - For code files: use python_execution_tool or code_execution_tool
+
+Think step by step and report your answer with the following template: 
 FINAL ANSWER: [YOUR FINAL ANSWER]. 
+
 YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, Apply the rules above for each element (number or string), ensure there is exactly one space after each comma.
-Your answer should only have the answer. 
+
+Your answer should only have the answer.
 """)
 
     # Combine system message with user messages