mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2025-03-31 12:20:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Set content_disposition_type and media_type on downloaded files (#8967)

Browse Source 
* changes

* add changeset

* changes

* add changeset

* add changeset

* changes

* changes

---------

Co-authored-by: Ali Abid <aliabid94@gmail.com>
Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com>
This commit is contained in:
aliabid94 2024-08-01 18:37:34 -07:00 committed by GitHub
parent c68eefba6c
commit 2f89877ac1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.changeset/shaky-mangos-yell.md Normal file
View File

@ -0,0 +1,5 @@
---
"gradio": patch
---
feat:Set `content_disposition_type` and `media_type` on downloaded files

12
gradio/routes.py
View File

@ -514,6 +514,7 @@ class App(FastAPI):
except PermissionError as err:
raise HTTPException(status_code=400, detail=str(err)) from err
rp_resp = await client.send(rp_req, stream=True)
rp_resp.headers.update({"Content-Disposition": "attachment"})
return StreamingResponse(
rp_resp.aiter_raw(),
status_code=rp_resp.status_code,
@ -579,15 +580,22 @@ class App(FastAPI):
if start.isnumeric() and end.isnumeric():
start = int(start)
end = int(end)
headers = dict(request.headers)
headers["Content-Disposition"] = "attachment"
response = ranged_response.RangedFileResponse(
abs_path,
ranged_response.OpenRange(start, end),
dict(request.headers),
headers,
stat_result=os.stat(abs_path),
)
return response
return FileResponse(abs_path, headers={"Accept-Ranges": "bytes"})
return FileResponse(
abs_path,
headers={"Accept-Ranges": "bytes"},
content_disposition_type="attachment",
media_type="application/octet-stream",
)
@app.get(
"/stream/{session_hash}/{run}/{component_id}",