Update app.py

app.py

@@ -1529,61 +1529,32 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         question_text = item.get("question")
         correct_answer = item.get("answer", "N/A")  # Get correct answer from API
         
-        # Initialize file variables for the current question
-        # Try to download file for EVERY task (not just if file_path exists)
-        file_download_url = f"{DEFAULT_API_URL}/files/{task_id}"
-        local_file_path = None 
+        # Look for file locally in files/ directory
+        local_file_path = None
+        files_dir = "files"
         
         try:
-            file_response = requests.get(file_download_url, timeout=15)
-            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]
+            # Check if files directory exists
+            if os.path.exists(files_dir):
+                # Look for any file that starts with the task_id
+                matching_files = [f for f in os.listdir(files_dir) if f.startswith(task_id)]
                 
-                # 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:
-                    for chunk in file_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}")
-            elif file_response.status_code == 404:
-                print(f"ℹ️ No file found for task {task_id} (404), proceeding without file.")
+                if matching_files:
+                    # Use the first matching file
+                    local_file_path = os.path.join(files_dir, matching_files[0])
+                    file_size = os.path.getsize(local_file_path)
+                    abs_path = os.path.abspath(local_file_path)
+                    
+                    print(f"✅ Found file: {matching_files[0]} ({file_size} bytes)")
+                    print(f"   Path: {abs_path}")
+                else:
+                    print(f"ℹ️ No file found for task {task_id}, proceeding without file.")
             else:
-                print(f"⚠️ Warning: File download for {task_id} failed with status {file_response.status_code}")   
+                print(f"⚠️ Warning: '{files_dir}' directory not found.")
                 
         except Exception as e:
-            # Handles any other unexpected errors
-            print(f"\n❌ An unexpected error occurred: {e}")      
-            
+            print(f"❌ Error looking for file: {e}")     
+                    
         try:
             # Pass file_path to agent
             submitted_answer = agent(question_text, local_file_path)