Update app.py

app.py

@@ -3,22 +3,21 @@ import base64
 from huggingface_hub import InferenceClient
 import os
 
-# Initialize Hugging Face Inference clients using tokens from environment variables
-client_dish = InferenceClient(api_key=os.getenv("HF_API_TOKEN_DISH"))
-client_diet = InferenceClient(api_key=os.getenv("HF_API_TOKEN_DIET"))
-
+# Initialize Hugging Face Inference client using token from environment variables
+client = InferenceClient(api_key=os.getenv("HF_API_TOKEN_DISH"))
+client1 = InferenceClient(api_key=os.getenv("HF_API_TOKEN_DIET"))
 # 1. Function to identify dish from image
 def identify_dish(image_bytes):
     encoded_image = base64.b64encode(image_bytes).decode("utf-8")
     dish_name = ""
 
-    for message in client_dish.chat_completion(
-        model="meta-llama/Llama-3-11B-Vision-Instruct",
+    for message in client.chat_completion(
+        model="meta-llama/Llama-3.2-11B-Vision-Instruct",
         messages=[
             {
                 "role": "You are a highly specialized food identification AI with extensive knowledge of global cuisines. Your sole task is to accurately identify dishes from images. Adhere strictly to these guidelines:\n1. Analyze the image thoroughly, focusing on ingredients, presentation, and cultural context.\n2. Provide ONLY the name of the main dish or dishes visible. Do not list individual ingredients or components.\n3. Use the most specific and widely recognized name for the dish.\n4. If multiple distinct dishes are present, list them separated by commas.\n5. If you cannot identify a dish with high confidence (>90%), respond with 'Unidentified dish'.\n6. Do not provide any explanations, descriptions, or additional commentary.\n7. Respond in a concise, list-like format.\nYour response should contain nothing but the dish name(s) or 'Unidentified dish'.",
                 "content": [
-                    {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encoded_image}"}},
+                    {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encoded_image}" }},
                     {"type": "text", "text": "Identify the dishes in the image and return only the names of the dishes."},
                 ],
             }
@@ -85,8 +84,9 @@ def generate_diet_plan(dish_name, calorie_intake_per_day, goal):
     - Dish Name: {dish_name}
     - Caloric Intake per Day: {calorie_intake_per_day} calories
     - Goal: {goal}
+
     """
-    response = client_diet.chat_completion(
+    response = client1.chat_completion(
         model="meta-llama/Meta-Llama-3-8B-Instruct",
         messages=[{"role": "You are a certified Dietitian with 20 years of Experience", "content": user_input}],
         max_tokens=500
@@ -100,10 +100,10 @@ st.title("AI Diet Planner")
 # Sidebar for user input
 st.sidebar.title("User Input")
 image_file = st.sidebar.file_uploader("Upload an image of the dish", type=["jpeg", "png"])
-age = st.sidebar.number_input("Enter your age", min_value=18)
+age = st.sidebar.number_input("Enter your age", min_value=1)
 gender = st.sidebar.selectbox("Select your gender", ["male", "female"])
-height_cm = st.sidebar.number_input("Enter your height (cm)", min_value=150.0)
-weight_kg = st.sidebar.number_input("Enter your weight (kg)", min_value=50.0)
+height_cm = st.sidebar.number_input("Enter your height (cm)", min_value=1.0)
+weight_kg = st.sidebar.number_input("Enter your weight (kg)", min_value=1.0)
 weight_goal = st.sidebar.selectbox("Weight goal", ["loss", "gain", "maintain"])
 activity_level = st.sidebar.selectbox("Activity level", ["sedentary", "light", "moderate", "active", "very active"])
 time_frame = st.sidebar.number_input("Time frame to achieve goal (months)", min_value=1)