main.py

\"\"\"
OmniParse AI — FastAPI Backend (HTML UI Ready).
Tento soubor slouží jako most mezi CORE logikou v app.py a budoucím HTML/JS frontendem.
Všechny business funkce se importují z app.py.
\"\"\"

import os
from fastapi import FastAPI, UploadFile, File, Header, HTTPException, Depends
from fastapi.responses import JSONResponse
from fastapi.middleware.cors import CORSMiddleware
import app as core

app = FastAPI(title="OmniParse AI API")

# CORS pro HTML/JS frontend
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"],
)

# --- AUTH ENDPOINTY ---

@app.post("/api/auth/signup")
async def api_signup(data: dict):
    ok, msg, token = core.signup(data.get("name"), data.get("email"), data.get("password"), data.get("accepted_terms"))
    if not ok:
        raise HTTPException(status_code=400, detail=msg)
    return {"message": msg, "token": token}

@app.post("/api/auth/login")
async def api_login(data: dict):
    ok, msg, token = core.login(data.get("email"), data.get("password"))
    if not ok:
        raise HTTPException(status_code=401, detail=msg)
    return {"message": msg, "token": token}

# --- BUSINESS ENDPOINTY (vyžadují token nebo API klíč) ---

async def get_current_user(authorization: str = Header(None)):
    if not authorization:
        raise HTTPException(status_code=401, detail="Missing Authorization header")
    
    token = authorization.replace("Bearer ", "")
    
    # Zkusíme session token (pro UI)
    user = core.resolve_session(token)
    if user:
        return user
        
    # Zkusíme API klíč (pro externí integrace)
    # Poznámka: v SQLiteDB/SupabaseDB by se musela přidat funkce get_user_by_api_key
    # Pro demo účely budeme hledat v DB ručně
    db_user = core.DB.get_user_by_email("demo@omniparse.ai") # Placeholder pro ukázku
    if db_user and db_user.get("api_key") == token:
        return db_user
        
    raise HTTPException(status_code=401, detail="Invalid token or API key")

@app.post("/api/extract")
async def api_extract(file: UploadFile = File(...), user: dict = Depends(get_current_user)):
    # Uložíme dočasně soubor
    temp_path = f"/tmp/{file.filename}"
    with open(temp_path, "wb") as buffer:
        buffer.write(await file.read())
    
    # Zavoláme CORE pipeline
    result = core.process_invoice_file(user, temp_path, file.filename)
    
    if "error" in result:
        raise HTTPException(status_code=500, detail=result["error"])
        
    return result

@app.get("/api/invoices")
async def api_get_invoices(user: dict = Depends(get_current_user)):
    return core.DB.get_invoices(user["id"])

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000)