Update app/routes/files.py

app/routes/files.py

@@ -25,14 +25,33 @@ async def list_files():
     }
 
 @router.get("/download")
-async def download_file_by_name(filename: str = Query(..., description="The exact name of the uploaded PDF file")):
+async def download_file_by_name(filename: str = Query(..., description="The exact name of the file")):
     """
-    Takes a file name (e.g., 'document.pdf') as input and returns the 
-    download URL for its generated Markdown report.
+    Takes a file name (e.g., 'document.pdf' or 'document_report.md') 
+    and returns the download URL for its generated report.
     """
+    
+    # --- FIX START: Handle Filename Mismatch ---
+    search_name = filename
+
+    # If frontend asks for '..._report.md', revert it to '.pdf' to search the DB
+    if filename.endswith("_report.md"):
+        search_name = filename.replace("_report.md", ".pdf")
+    # -------------------------------------------
+
     # 1. Find the job associated with this filename
-    job = await run_in_threadpool(get_job_by_filename, filename)
+    job = await run_in_threadpool(get_job_by_filename, search_name)
     
+    # If not found via PDF logic, it might be a ZIP batch that didn't get renamed
+    if not job and filename.endswith("_report.md"):
+         # Try replacing with .zip just in case it was a batch report
+         search_name_zip = filename.replace("_report.md", ".zip")
+         job = await run_in_threadpool(get_job_by_filename, search_name_zip)
+
+    if not job:
+        # Final fallback: maybe the user really uploaded a file named '..._report.md'
+        job = await run_in_threadpool(get_job_by_filename, filename)
+
     if not job:
         raise HTTPException(status_code=404, detail=f"File '{filename}' not found in records.")