Spaces:
Sleeping
Sleeping
| #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 model and get response | |
| def get_response_diet(prompt, input): | |
| model = genai.GenerativeModel('gemini-pro') | |
| response = model.generate_content([prompt, input]) | |
| return response.text | |
| #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 | |
| #Preprocess image data | |
| def prep_image(uploaded_file): | |
| #Check if there is any data | |
| if uploaded_file is not None: | |
| #Read the file as bytes | |
| bytes_data = uploaded_file.getvalue() | |
| #get the image part information | |
| image_parts = [ | |
| { | |
| "mime_type": uploaded_file.type, | |
| "data": bytes_data | |
| } | |
| ] | |
| return image_parts | |
| else: | |
| raise FileNotFoundError("No File is uploaded!") | |
| #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) |