from typing import Any

import litellm
from fastapi import APIRouter, Depends, File, Form, HTTPException, UploadFile
from pydantic import BaseModel
from sqlmodel import Session

from app.ai.nutrition import NutritionalInfo, analyze_nutrition_from_image, nutrition_module
from app.api import deps
from app.models.food import FoodLog  # Added FoodItem

router = APIRouter()


class AnalyzeRequest(BaseModel):
    description: str


@router.post("/analyze", response_model=NutritionalInfo)
def analyze_food(
    request: AnalyzeRequest,
) -> Any:
    """
    Analyze food description and return nutritional info using DSPy.
    """
    try:
        result = nutrition_module(description=request.description)
        return result.nutritional_info
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.post("/analyze/image", response_model=NutritionalInfo)
async def analyze_food_image(
    file: UploadFile = File(...),
    description: str = Form(""),
) -> Any:
    """
    Analyze food image and return nutritional info.
    """
    try:
        contents = await file.read()
        return analyze_nutrition_from_image(contents, description)
    except litellm.exceptions.BadRequestError as e:
        raise HTTPException(status_code=400, detail=f"Invalid image or request: {str(e)}")
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@router.post("/log", response_model=FoodLog)
def log_food(
    *,
    session: Session = Depends(deps.get_session),
    nutrition_info: NutritionalInfo,
    current_user: deps.CurrentUser,
) -> Any:
    """
    Save food log to database.
    """
    food_log = FoodLog(
        user_id=current_user.id,
        name=nutrition_info.name,
        calories=nutrition_info.calories,
        protein=nutrition_info.protein,
        carbs=nutrition_info.carbs,
        fats=nutrition_info.fats,
    )
    session.add(food_log)
    session.commit()
    session.refresh(food_log)
    return food_log


@router.get("/logs", response_model=list[FoodLog])
def read_logs(
    current_user: deps.CurrentUser,
    session: Session = Depends(deps.get_session),
    skip: int = 0,
    limit: int = 100,
) -> Any:
    """
    Get food logs for current user.
    """
    from sqlmodel import select

    statement = (
        select(FoodLog)
        .where(FoodLog.user_id == current_user.id)
        .order_by(FoodLog.timestamp.desc())
        .offset(skip)
        .limit(limit)
    )
    return session.exec(statement).all()