Update app.py

app.py

@@ -2,10 +2,13 @@ import streamlit as st
 import base64
 from huggingface_hub import InferenceClient
 import os
+import requests
 
-# Initialize Hugging Face Inference client using token from environment variables
+# Initialize Hugging Face Inference client using tokens from environment variables
 client = InferenceClient(api_key=os.getenv("HF_API_TOKEN_DISH"))
 client1 = InferenceClient(api_key=os.getenv("HF_API_TOKEN_DIET"))
+groq_api_key = os.getenv("API_Groq")
+
 # 1. Function to identify dish from image
 def identify_dish(image_bytes):
     encoded_image = base64.b64encode(image_bytes).decode("utf-8")
@@ -15,7 +18,7 @@ def identify_dish(image_bytes):
         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'.",
+                "role": "You are a highly specialized food identification AI...",
                 "content": [
                     {"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."},
@@ -30,7 +33,7 @@ def identify_dish(image_bytes):
 
     return dish_name.strip()
 
-# 2. Function to get user inputs and calculate daily caloric needs
+# 2. Function to calculate daily caloric needs
 def calculate_metrics(age, gender, height_cm, weight_kg, weight_goal, activity_level, time_frame_months):
     bmi = weight_kg / ((height_cm / 100) ** 2)
 
@@ -78,13 +81,12 @@ def calculate_metrics(age, gender, height_cm, weight_kg, weight_goal, activity_l
 # 3. Function to generate diet plan
 def generate_diet_plan(dish_name, calorie_intake_per_day, goal):
     user_input = f"""
-    You are a certified Dietitian with 20 years of experience. Based on the following input, create an Indian diet plan that fits within the calculated calorie intake and assesses if the given dish is suitable for the user's goal.
+    You are a certified Dietitian with 20 years of experience. Based on the following input, create an Indian diet plan...
 
     Input:
     - Dish Name: {dish_name}
     - Caloric Intake per Day: {calorie_intake_per_day} calories
     - Goal: {goal}
-
     """
     response = client1.chat_completion(
         model="meta-llama/Meta-Llama-3-8B-Instruct",
@@ -94,6 +96,30 @@ def generate_diet_plan(dish_name, calorie_intake_per_day, goal):
 
     return response.choices[0].message.content
 
+# 4. Function to generate a recipe using Groq API
+def generate_recipe(recipe_name):
+    url = "https://api.groq.com/openai/v1/chat/completions"
+    headers = {
+        "Authorization": f"Bearer {groq_api_key}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "model": "gpt-3.5-turbo",
+        "messages": [
+            {"role": "system", "content": "You are a skilled chef."},
+            {"role": "user", "content": f"Create a detailed recipe for {recipe_name}"}
+        ]
+    }
+
+    response = requests.post(url, headers=headers, json=data)
+
+    if response.status_code == 200:
+        result = response.json()
+        generated_recipe = result['choices'][0]['message']['content']
+        return generated_recipe
+    else:
+        return f"Error: {response.status_code}, {response.text}"
+
 # Streamlit App Title
 st.title("AI Diet Planner")
 
@@ -143,6 +169,16 @@ if submit:
         st.write("#### Diet Plan Based on Dish & Goal:")
         st.markdown(f"<div style='background-color: #d1ecf1; color: #0c5460; padding: 10px; border-radius: 10px;'>{diet_plan}</div>", unsafe_allow_html=True)
 
+        # Step 4: Ask for recipe generation
+        st.markdown("<hr>", unsafe_allow_html=True)
+        recipe_option = st.radio("Would you like a recipe for a dish?", ["No", "Yes"])
+
+        if recipe_option == "Yes":
+            recipe_name = st.text_input("Enter the dish name for the recipe:")
+            if st.button("Generate Recipe"):
+                recipe = generate_recipe(recipe_name)
+                st.write("### Generated Recipe:")
+                st.markdown(f"<div style='background-color: #fff3cd; color: #856404; padding: 10px; border-radius: 10px;'>{recipe}</div>", unsafe_allow_html=True)
     else:
         st.error("Please upload a valid image in JPEG or PNG format.")