Christian Krakau-Louis
27 lines
902 B
Python
27 lines
902 B
Python
# In main.py or a new dedicated file, e.g. app/routes_ui.py
|
|
from fastapi import APIRouter, UploadFile, File, HTTPException
|
|
from app.tasks.upload_to_s3 import upload_to_s3
|
|
import os
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("/ui-upload")
|
|
async def ui_upload(file: UploadFile = File(...)):
|
|
# You can store this file in your 'workdir' (like how /process does) or a tmp dir
|
|
workdir = "/workdir"
|
|
target_path = os.path.join(workdir, file.filename)
|
|
|
|
try:
|
|
with open(target_path, "wb") as f:
|
|
content = await file.read()
|
|
f.write(content)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"Failed to save file: {e}")
|
|
|
|
# Now you can call your existing Celery flow:
|
|
task = upload_to_s3.delay(target_path)
|
|
return {"task_id": task.id, "status": "queued"}
|
|
|
|
# Then include this in your main app as well:
|
|
# app.include_router(router)
|