Update app.py

app.py

@@ -1539,17 +1539,46 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         
         try:
             file_response = requests.get(file_download_url, timeout=15)
-            if file_response.status_code == 200:
-                # File exists, save it
+            if file_response.status_code == 200:        
+                # Get filename from Content-Disposition header if available
+                filename = None
+                if 'Content-Disposition' in file_response.headers:
+                    cd = file_response.headers['Content-Disposition']
+                    filename_match = re.findall('filename="?([^"]+)"?', cd)
+                    if filename_match:
+                        filename = filename_match[0]
+                
+                # If no filename, use task_id with extension from Content-Type
+                if not filename:
+                    content_type = file_response.headers.get('Content-Type', '')
+                    ext_map = {
+                        'image/png': '.png',
+                        'image/jpeg': '.jpg',
+                        'image/gif': '.gif',
+                        'audio/mpeg': '.mp3',
+                        'audio/wav': '.wav',
+                        'text/plain': '.txt',
+                        'text/csv': '.csv',
+                        'application/pdf': '.pdf',
+                        'text/x-python': '.py',
+                        'application/x-python-code': '.py',
+                    }
+                    ext = ext_map.get(content_type, '')
+                    filename = f"{task_id}{ext}"
+                
+                # Save to current directory
+                local_file_path = filename
+                
                 with open(local_file_path, 'wb') as f:
-                    f.write(file_response.content)
-                print(f"✅ Downloaded file for {task_id}")
-            else:
-                # No file for this task, that's fine
-                local_file_path = None
-        except:
-            local_file_path = None
-            
+                    for chunk in response.iter_content(chunk_size=8192):
+                        f.write(chunk)
+                
+                file_size = os.path.getsize(local_file_path)
+                abs_path = os.path.abspath(local_file_path)
+                
+                print(f"✅ Downloaded: {filename} ({file_size} bytes)")
+                print(f"   Saved to: {abs_path}")      
+           
         try:
             # Pass file_path to agent
             submitted_answer = agent(question_text, local_file_path)