Update main.py

main.py

@@ -236,7 +236,18 @@ async def startup_event():
 
 
 # --- Your Endpoints ---
-@app.get("/files/{task_id}", ...)
+@app.get("/files/{task_id}",
+         summary="Get Associated File by Task ID",
+         description="Downloads the file associated with the given task_id, if one exists and is mapped.",
+         responses={
+             200: {
+                 "description": "File content.",
+                 "content": {"*/*": {}} # Indicates response can be any file type
+             },
+             403: {"model": ErrorResponse, "description": "Access denied (e.g., path traversal attempt)."},
+             404: {"model": ErrorResponse, "description": "Task ID not found, no file associated, or file missing on server."},
+             500: {"model": ErrorResponse, "description": "Server error reading file."}
+         })
 async def get_task_file(task_id: str):
     # ... (endpoint logic) ...
     try: