39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
from fastapi import APIRouter, UploadFile, File, Form, HTTPException
|
|
from fastapi import APIRouter, Depends
|
|
from infra.token.token_manager import TokenManager
|
|
from starlette.status import HTTP_401_UNAUTHORIZED
|
|
from fastapi.encoders import jsonable_encoder
|
|
from fastapi.responses import JSONResponse
|
|
from app.central_storage.backend.application.document_hub import DocumentHub
|
|
|
|
|
|
router = APIRouter()
|
|
token_manager = TokenManager()
|
|
|
|
|
|
@router.post(
|
|
"/upload-file",
|
|
summary="upload a document with a given associated_with id.",
|
|
description="upload a document. If success, returning the document id and file name",
|
|
)
|
|
async def upload_file(associated_with: str = Form(...), file: UploadFile = File(None)):
|
|
|
|
document_hub = DocumentHub()
|
|
# File processing
|
|
try:
|
|
file_data = await file.read() # You can use async chunking for larger files
|
|
document_id = await document_hub.upload_document(
|
|
associated_with, file.filename, file_data
|
|
)
|
|
|
|
if document_id:
|
|
result = {"document_id": str(document_id), "file_name": file.filename}
|
|
return JSONResponse(content=jsonable_encoder(result))
|
|
else:
|
|
return JSONResponse(
|
|
status_code=500, content={"error": "File upload failed"}
|
|
)
|
|
|
|
except Exception as e:
|
|
return JSONResponse(status_code=500, content={"error": "Internal server error"})
|