Update app.py

app.py

@@ -1,67 +1,102 @@
-
-import streamlit as st
+#Import libraries
+from dotenv import load_dotenv
 import os
 import google.generativeai as genai
 from PIL import Image
+import streamlit as st
+
 
+#Load API Key
+load_dotenv()
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-## Function to load Google Gemini Pro Vision API And get response
+#Function to load Google Gemini Pro model and get response
+def get_response_diet(prompt, input):
+    model = genai.GenerativeModel('gemini-pro')
+    response = model.generate_content([prompt, input])
+    return response.text
 
-def get_gemini_repsonse(input,image,prompt):
-    model=genai.GenerativeModel('gemini-pro-vision')
-    response=model.generate_content([input,image[0],prompt])
+#Function to load Google Gemini Vision model and get response
+def get_response_nutrition(image, prompt):
+    model = genai.GenerativeModel('gemini-pro-vision')
+    response = model.generate_content([image[0], prompt])
     return response.text
 
-def input_image_setup(uploaded_file):
-    # Check if a file has been uploaded
+#Preprocess image data
+def prep_image(uploaded_file):
+    #Check if there is any data
     if uploaded_file is not None:
-        # Read the file into bytes, image into numbers
+        #Read the file as bytes
         bytes_data = uploaded_file.getvalue()
 
-        image_parts = [                     #In dic format
+        #get the image part information
+        image_parts = [
             {
-                "mime_type": uploaded_file.type,  # Get the mime type of the uploaded file
+                "mime_type": uploaded_file.type,
                 "data": bytes_data
             }
         ]
         return image_parts
     else:
-        raise FileNotFoundError("No file uploaded")
+        raise FileNotFoundError("No File is uploaded!")
     
-##initialize our streamlit app
-
-st.set_page_config(page_title="Nutrition and Calories Manager App")
-
-st.header("Nutrition and Calories Manager App")
-input=st.text_input("Input Prompt: ",key="input")
-uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
-image=""   
-if uploaded_file is not None:
-    image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image.", use_column_width=True)
-
-
-submit=st.button("CHECK")
-
-input_prompt="""
-You are an expert in nutritionist where you need to see the food items from the image
-               and calculate the total calories, also provide the details of every food item with calories intake
-               is below format
-               1. Item 1 -no of calories
-               2. Item 2 -no of calories
-               ----
-               ----
-               Total Calories
-Also mention about the important nutrients like Protein , Carbohydrates and Fats  in each item
-        Put the  output in a tabular format with each item in a separate row and details in columns
-"""
-
-## If submit button is clicked
-
-if submit:
-    image_data=input_image_setup(uploaded_file)
-    response=get_gemini_repsonse(input_prompt,image_data,input)
-    st.subheader("The Response is")
-    st.write(response)
+#Configuring Streamlit App
+#st.set_page_config(page_title="Health Management: Nutrition Calculator & Diet Planner")
+st.image('Nutritionist/logo.jpg', width=70)
+st.header("Health: Nutrition Calculator & Diet Planner")
+
+######################################################################################
+section_choice1 = st.radio("Choose Section:", ("Nutrition Calculator","Diet Planner"))
+
+#If choice is nutrition calculator
+if section_choice1 == "Nutrition Calculator":
+    upload_file = st.file_uploader("Choose an image...", type=["jpg","jpeg","png"])
+    image = ""
+    if upload_file is not None:
+        #Show the image
+        image = Image.open(upload_file)
+        st.image(image, caption="Uploaded Image", use_column_width=True)
+
+
+    #Prompt Template
+    input_prompt_nutrition = """
+    You are an expert Nutritionist. As a skilled nutritionist, you're required to analyze the food iems
+    in the image and determine the total nutrition value. 
+    Additionally, you need to furnish a breakdown of each food item along with its respective content.
+
+    Food item, Serving size, Tatal Cal., Protien (g), Fat,
+    Carb (g), Fiber (g), Vit B-12, Vit B-6,
+    Iron, Zinc, Mang.
+
+    Use a table to show above informaion.
+    """
+    ##if the button is clicked
+    submit = st.button("Calculate Nutrition value!")
+    if submit:
+        image_data = prep_image(upload_file)
+        response = get_response_nutrition(image_data, input_prompt_nutrition)
+        st.subheader("Nutrition AI: ")
+        st.write(response)
+
+#If choice is diet planner
+if section_choice1 == "Diet Planner":
+
+    #Prompt Template
+    input_prompt_diet = """
+    You are an expert Nutritionist. 
+    If the input contains list of items like fruits or vegetables, you have to give diet plan and suggest
+    breakfast, lunch, dinner wrt given item.
+    If the input contains numbers, you have to suggest diet plan for breakfast, luncg=h, dinner within
+    given number of calorie for the whole day.
+
+    Return the response using markdown.
 
+    """
+    ##if the button is clicked
+    input_diet = st.text_area(" Input the list of items that you have at home and get diet plan! OR \
+                              Input how much calorie you want to intake perday?:")
+    submit1 = st.button("Plan my Diet!")
+    if submit1:
+        response = get_response_diet(input_prompt_diet, input_diet)
+        st.subheader("Diet AI: ")
+        st.write(response)