add download code
Browse files- 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)
|