from fastapi import FastAPI from transformers import pipeline app = FastAPI(docs_url="/") @app.get("/calculate-food") def calculate_food_endpoint(activity: str, weight: int): score = 0 if activity == "Typical" : score = 110 elif activity == "Active" : score = 125 elif activity == "Overweight" : score = 70 elif activity == "Highly Active" : score = 175 elif activity == "Senior, neutered, inactive" : score = 90 elif activity == "Working Dog (light duty)" : score = 130 elif activity == "Working Dog (moderate duty)" : score = 150 elif activity == "Working Dog (heavy duty)" : score = 175 else : score = 110 return round( score / weight, 2) """ Calculates the recommended amount of dog food based on activity level and weight. Args: activity: The dog's activity level, as a number from 1 to 5. weight: The dog's weight in kilograms. Returns: A JSON object containing the recommended amount of food in cups. """ # Check if the activity and weight parameters are present in the API request. if activity is None or weight is None: return {"error": "Please provide both activity level and weight."} # Calculate the recommended amount of food. result = calculate_food(activity, weight) # Respond with the result as JSON. return result