Backend/schema/user_input.py

from pydantic import BaseModel, Field
from typing import Literal

class WorkoutFeatures(BaseModel):
    # 1. Demographic Features
    Age: int = Field(..., ge=0)
    Gender: int
    Session_Duration_hours: float = Field(..., alias="Session_Duration (hours)", ge=0)
    Calories_Burned: float = Field(..., ge=0)
    Fat_Percentage: float = Field(..., ge=0)
    Water_Intake_liters: float = Field(..., alias="Water_Intake (liters)", ge=0)
    workout_frequency : int = Field(..., alias="Workout_Frequency (days/week)", ge=0)
    experience_level : int = Field(..., alias="Experience_Level", ge=0)
    bmi : float = Field(..., alias="BMI", ge=0)
    daily_meals_frequency : int = Field(..., alias="Daily meals frequency", ge=0)
    carbs : float = Field(..., alias="Carbs", ge=0)
    proteins : float = Field(..., alias="Proteins", ge=0)
    fats : float = Field(..., alias="Fats", ge=0)
    calories : float = Field(..., alias="Calories", ge=0)
    diet_type: Literal['Vegan', 'Vegetarian', 'Paleo', 'Keto', 'Low-Carb', 'Balanced']

    class Config:
        populate_by_name = True