mirror of
https://github.com/escalante29/healthy-fit.git
synced 2026-03-21 09:08:46 +01:00
184c8330a7
Introduces DSPy-based nutrition and plan generation modules, including image analysis for nutritional info and personalized diet/exercise plans. Adds new API endpoints for health metrics/goals, nutrition image analysis, and plan management. Updates models, schemas, and backend structure to support these features, and includes initial training data and configuration for prompt optimization.
73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
import litellm
|
|
import dspy
|
|
from typing import Any
|
|
|
|
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
|