Spaces:
Sleeping
Sleeping
| 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) | |