Salimshakeel commited on
Commit
936b370
·
verified ·
1 Parent(s): deda5ff

add download code

Browse files
Files changed (1) hide show
  1. routes/summarize.py +9 -0
routes/summarize.py CHANGED
@@ -9,6 +9,15 @@ import os
9
  router = APIRouter()
10
 
11
  @router.post("/summarize")
 
 
 
 
 
 
 
 
 
12
  def summarize_video(video: UploadFile = File(...)):
13
  if not video.filename.lower().endswith(('.mp4', '.avi', '.mov', '.mkv')):
14
  return JSONResponse(content={"error": "Unsupported file format"}, status_code=400)
 
9
  router = APIRouter()
10
 
11
  @router.post("/summarize")
12
+
13
+ @router.get("/download/{filename}")
14
+ def download_summary(filename: str):
15
+ file_path = os.path.join(os.getcwd(), "static", filename)
16
+ return FileResponse(
17
+ path=file_path,
18
+ filename=filename,
19
+ media_type='application/octet-stream'
20
+ )
21
  def summarize_video(video: UploadFile = File(...)):
22
  if not video.filename.lower().endswith(('.mp4', '.avi', '.mov', '.mkv')):
23
  return JSONResponse(content={"error": "Unsupported file format"}, status_code=400)