Update app.py

app.py

@@ -58,8 +58,14 @@ async def download_dependencies(requirements_file: UploadFile = File(...)):
                     tar.add(file_path, arcname=file)
             tar.add(log_file_path, arcname="download.log")
 
+        # Get the file size
+        file_size = os.path.getsize(tar_path)
+
         # Return the tar file
-        return StreamingResponse(open(tar_path, "rb"), media_type="application/gzip", headers={"Content-Disposition": f"attachment; filename=dependencies.tar.gz"})
+        return StreamingResponse(open(tar_path, "rb"), media_type="application/gzip", headers={
+            "Content-Disposition": f"attachment; filename=dependencies.tar.gz",
+            "Content-Length": str(file_size)
+        })
 
     except subprocess.CalledProcessError as e:
         raise HTTPException(status_code=500, detail=f"Error downloading dependencies: {str(e)}")
@@ -71,8 +77,7 @@ async def download_dependencies(requirements_file: UploadFile = File(...)):
         if os.path.exists(log_file_path):
             os.remove(log_file_path)
         if os.path.exists(tar_path):
-            os.remove(tar_path)
-            
+            os.remove(tar_path)        
 def download_docker_image(image_name, tag='latest', destination='/tmp/docker-images'):
     try:
         os.makedirs(destination, exist_ok=True)