Update app.py

app.py

@@ -31,43 +31,45 @@ async def check_n_load_attach(session: aiohttp.ClientSession, task_id
     try:
         async with session.get(file_url, timeout=15) as response:
             if response.status == 200:
-                # Get filename from Content-Disposition
-                filename = None
-                content_disposition = response.headers.get("Content-Disposition")
-                if content_disposition and "filename=" in content_disposition:
-                    filename = content_disposition.split("filename=")[-1].strip('"')
-                if not filename:
-                    # Determine extension from Content-Type
-                    content_type = str(response.headers.get("Content-Type", "")).lower()
-                    extension = ""
-                    if "image/png" in content_type:
-                        extension = ".png"
-                    elif "image/jpeg" in content_type:
-                        extension = ".jpg"
-                    elif "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet" in content_type:
-                        extension = ".xlsx"
-                    elif "audio/mpeg" in content_type:
-                        extension = ".mp3"
-                    elif "application/pdf" in content_type:
-                        extension = ".pdf"
-                    elif "text/x-python" in content_type:
-                        extension = ".py"
-                    else:
-                        extension = ""
-                    filename = f"{task_id}{extension}"
+
+                # Determine file extension from Content-Type
+                content_type = str(response.headers.get("Content-Type", "")).lower()
+                extension = ""
+                if "image/png" in content_type:
+                    extension = ".png"
+                elif "image/jpeg" in content_type:
+                    extension = ".jpg"
+                elif "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet" in content_type:
+                    extension = ".xlsx"
+                elif "audio/mpeg" in content_type:
+                    extension = ".mp3"
+                elif "application/pdf" in content_type:
+                    extension = ".pdf"
+                elif "text/x-python" in content_type:
+                    extension = ".py"
+                else:
+                    print(f"Unsupported content type: {content_type} for task {task_id}")
+                    return None
                 
-                # Save the file
+                # Use task_id as the filename to ensure uniqueness
+                filename = f"{task_id}{extension}"
                 local_file_path = os.path.join("downloads", filename)
                 os.makedirs("downloads", exist_ok=True)
+                
+                # Save the file
                 async with aiofiles.open(local_file_path, "wb") as file:
                     async for chunk in response.content.iter_chunked(8192):
                         await file.write(chunk)
                 print(f"File downloaded successfully: {local_file_path}")
                 return local_file_path
+            else:
+                print(f"Failed to download file for task {task_id}: HTTP {response.status}")
+                return None
     except aiohttp.ClientError as e:
         print(f"Error downloading attachment for task {task_id}: {str(e)}")
         return None
 
+
 async def fetch_questions(session: aiohttp.ClientSession, questions_url: str) -> list:
     """Fetch questions asynchronously."""
     try: