Update app.py

app.py

@@ -1,73 +1,73 @@
-from dotenv import load_dotenv
-load_dotenv()
-
-import streamlit as st
-import google.generativeai as genai 
-import os
-from PIL import Image
-#from PyPDF2 import PdfReader
-
-genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-
-def get_gemini_response(input_prompt,image):
-    model=genai.GenerativeModel("gemini-pro-vision")
-    response=model.generate_content([input_prompt,image[0]])
-    return response.text
-
-
-def get_image_content(uploaded_file):
-    if uploaded_file is not None:
-        image_byte_data=uploaded_file.getvalue()
-
-        image_parts = [
-            {   
-                "mime_type":uploaded_file.type, 
-                "data":image_byte_data
-            }
-        ]
-        return image_parts
-    else:
-        raise FileNotFoundError("File not uploaded") 
-    
-
-st.set_page_config(page_title="FoodAnalyzer")
-st.markdown("<h1 style='text-align: center;'>FoodAnalyzer App</h1>", unsafe_allow_html=True)
-
-age = st.number_input("Enter Age", min_value=0, max_value=120, step=1)
-gender = st.selectbox("Select Gender", options=["Male", "Female", "Other"])
-weight = st.number_input("Enter Weight in (kg)", min_value=0.0, step=0.1) 
-Height = st.number_input("Enter Height in (cm)", min_value=0.0, step=0.1) 
-
-uploaded_file=st.file_uploader("Upload an Image",type=["jpg","png","jpeg"])   
-image=''
-if uploaded_file is not None:
-    image=Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
-
-col1, col2, col3 = st.columns([7, 2, 7])
-with col2:
-    submit = st.button("Analyze")
-
-input_prompt = f"""
-You are an expert nutritionist. Please analyze the food items from the uploaded image and calculate the
-total calories. Also provide the details of every food item with calorie intake in the following format:
-
-         1. Item 1 - Number of calories
-         2. Item 2 - Number of calories  
-         3. Item 3 - Number of calories
-         4. Item 4 - Number of calories  
-......
-......
-
-Also, consider the user inputted {age}, {gender}, {weight}, {Height} and mention whether the food is healthy or not 
-for the inputted {age}, {gender}, {weight},{Height}. Additionally, mention the percentage split of the ratio of proteins, carbohydrates, fats, fiber, sugar, minerals, vitamins, and
-other important nutrients required in our diet. Do not confuse and show {weight} with {Height} or {Height} with {weight}"""
-
-
-if submit:
-    image_date = get_image_content(uploaded_file)
-    response=get_gemini_response(input_prompt,image_date)
-    st.subheader("The Response is")
-    st.write(response)
-
+from dotenv import load_dotenv
+load_dotenv()
+
+import streamlit as st
+import google.generativeai as genai 
+import os
+from PIL import Image
+#from PyPDF2 import PdfReader
+
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+
+def get_gemini_response(input_prompt,image):
+    model=genai.GenerativeModel("gemini-pro-1.5-latest")
+    response=model.generate_content([input_prompt,image[0]])
+    return response.text
+
+
+def get_image_content(uploaded_file):
+    if uploaded_file is not None:
+        image_byte_data=uploaded_file.getvalue()
+
+        image_parts = [
+            {   
+                "mime_type":uploaded_file.type, 
+                "data":image_byte_data
+            }
+        ]
+        return image_parts
+    else:
+        raise FileNotFoundError("File not uploaded") 
+    
+
+st.set_page_config(page_title="FoodAnalyzer")
+st.markdown("<h1 style='text-align: center;'>FoodAnalyzer App</h1>", unsafe_allow_html=True)
+
+age = st.number_input("Enter Age", min_value=0, max_value=120, step=1)
+gender = st.selectbox("Select Gender", options=["Male", "Female", "Other"])
+weight = st.number_input("Enter Weight in (kg)", min_value=0.0, step=0.1) 
+Height = st.number_input("Enter Height in (cm)", min_value=0.0, step=0.1) 
+
+uploaded_file=st.file_uploader("Upload an Image",type=["jpg","png","jpeg"])   
+image=''
+if uploaded_file is not None:
+    image=Image.open(uploaded_file)
+    st.image(image, caption="Uploaded Image", use_column_width=True)
+
+col1, col2, col3 = st.columns([7, 2, 7])
+with col2:
+    submit = st.button("Analyze")
+
+input_prompt = f"""
+You are an expert nutritionist. Please analyze the food items from the uploaded image and calculate the
+total calories. Also provide the details of every food item with calorie intake in the following format:
+
+         1. Item 1 - Number of calories
+         2. Item 2 - Number of calories  
+         3. Item 3 - Number of calories
+         4. Item 4 - Number of calories  
+......
+......
+
+Also, consider the user inputted {age}, {gender}, {weight}, {Height} and mention whether the food is healthy or not 
+for the inputted {age}, {gender}, {weight},{Height}. Additionally, mention the percentage split of the ratio of proteins, carbohydrates, fats, fiber, sugar, minerals, vitamins, and
+other important nutrients required in our diet. Do not confuse and show {weight} with {Height} or {Height} with {weight}"""
+
+
+if submit:
+    image_date = get_image_content(uploaded_file)
+    response=get_gemini_response(input_prompt,image_date)
+    st.subheader("The Response is")
+    st.write(response)
+