Merge feature/file-upload

.env

@@ -32,7 +32,7 @@ SMTP_SSL=False
 SMTP_PORT=587
 
 # Postgres
-POSTGRES_SERVER=localhost
+POSTGRES_SERVER=db
 POSTGRES_PORT=5432
 POSTGRES_DB=app
 POSTGRES_USER=postgres

backend/.python-version

@@ -0,0 +1 @@
+3.11.9

backend/app/api/main.py

@@ -1,13 +1,13 @@
 from fastapi import APIRouter
 
-from app.api.routes import items, login, private, users, utils
+from app.api.routes import documents, login, private, users, utils
 from app.core.config import settings
 
 api_router = APIRouter()
 api_router.include_router(login.router)
 api_router.include_router(users.router)
 api_router.include_router(utils.router)
-api_router.include_router(items.router)
+api_router.include_router(documents.router)
 
 
 if settings.ENVIRONMENT == "local":

backend/app/api/routes/documents.py

@@ -0,0 +1,45 @@
+from typing import Any
+from app.api.deps import CurrentUser, SessionDep
+from app.models import Document, DocumentCreate, DocumentPublic
+from fastapi import APIRouter, BackgroundTasks, File, UploadFile, HTTPException
+from app.s3 import upload_file_to_s3, generate_s3_url
+
+router = APIRouter(prefix="/documents", tags=["documents"])
+
+@router.post("/", response_model=DocumentPublic)
+def create_document(
+    *, session: SessionDep, current_user: CurrentUser,
+      background_tasks: BackgroundTasks,
+      file: UploadFile = File(...), 
+):
+    key = None
+    try:
+        user_id = current_user.id
+        key = upload_file_to_s3(file, current_user.id)
+    except Exception as e:
+        raise HTTPException(500, f"Failed to upload file: {key}. Error: {str(e)}")
+
+    try:
+        url = generate_s3_url(key)
+    except Exception as e:
+        raise HTTPException(500, f"Could not generate URL for file key: {key}")
+
+    document_in = DocumentCreate(
+        filename=file.filename,
+        content_type=file.content_type,
+        size=file.size,  
+        s3_url=url,
+    )
+
+    document = Document.model_validate(document_in, update={"owner_id": current_user.id})
+
+
+    session.add(document)
+    session.commit()
+    session.refresh(document)
+
+    # 3. Kick off background job
+    print("Document created, starting background task...")
+    # background_tasks.add_task(generate_questions, document.id)
+
+    return document