main.py

from fastapi import FastAPI, Request
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
from fastapi.responses import HTMLResponse, RedirectResponse
import os

from config import settings
from database import init_db, SessionLocal, User

app = FastAPI(title=settings.APP_NAME, description=settings.APP_DESCRIPTION)

app.mount("/static", StaticFiles(directory="static"), name="static")
app.mount("/uploads", StaticFiles(directory=settings.UPLOAD_DIR), name="uploads")

templates = Jinja2Templates(directory="templates")

from routes_auth import router as auth_router
from routes_profiles import router as profiles_router
from routes_chat import router as chat_router
from routes_search import router as search_router
from routes_admin import router as admin_router
from routes_voice import router as voice_router

app.include_router(auth_router)
app.include_router(profiles_router)
app.include_router(chat_router)
app.include_router(search_router)
app.include_router(admin_router)
app.include_router(voice_router)


@app.on_event("startup")
async def startup():
    init_db()
    os.makedirs(settings.UPLOAD_DIR, exist_ok=True)
    os.makedirs("static", exist_ok=True)
    os.makedirs("static/js", exist_ok=True)
    os.makedirs("static/css", exist_ok=True)
    os.makedirs("templates", exist_ok=True)


@app.get("/", response_class=HTMLResponse)
async def index(request: Request):
    from auth import get_user_from_request
    db = SessionLocal()
    try:
        user = get_user_from_request(request, db)
        if user:
            return RedirectResponse(url="/dashboard", status_code=302)
        return RedirectResponse(url="/auth/login", status_code=302)
    finally:
        db.close()


@app.get("/dashboard", response_class=HTMLResponse)
async def dashboard(request: Request):
    from auth import get_user_from_request
    db = SessionLocal()
    try:
        user = get_user_from_request(request, db)
        if not user:
            return RedirectResponse(url="/auth/login", status_code=302)

        from database import MemoryProfile, Conversation, MemoryFile
        profiles = db.query(MemoryProfile).filter(MemoryProfile.user_id == user.id).all()
        total_conversations = db.query(Conversation).filter(Conversation.user_id == user.id).count()
        total_files = db.query(MemoryFile).join(MemoryProfile).filter(MemoryProfile.user_id == user.id).count()
        recent_conversations = db.query(Conversation).filter(
            Conversation.user_id == user.id
        ).order_by(Conversation.updated_at.desc()).limit(5).all()

        recent_conv_profiles = {}
        for rc in recent_conversations:
            p = db.query(MemoryProfile).filter(MemoryProfile.id == rc.profile_id).first()
            if p:
                recent_conv_profiles[rc.id] = p.name

        return templates.TemplateResponse("dashboard.html", {
            "request": request, "user": user, "profiles": profiles,
            "total_conversations": total_conversations,
            "total_files": total_files,
            "recent_conversations": recent_conversations,
            "recent_conv_profiles": recent_conv_profiles,
        })
    finally:
        db.close()


@app.exception_handler(302)
async def redirect_handler(request: Request, exc):
    return RedirectResponse(url=exc.headers.get("Location", "/"), status_code=302)


if __name__ == "__main__":
    import uvicorn
    uvicorn.run("main:app", host="127.0.0.1", port=8000, reload=True)