Update app.py

app.py

@@ -1,37 +1,44 @@
-# Q&A Chatbot
-#from langchain.llms import OpenAI
 from dotenv import load_dotenv
 load_dotenv()  # take environment variables from .env.
 import streamlit as st
 import os
 from PIL import Image
 import google.generativeai as genai
+
+# Set the page configuration (must be the first Streamlit command)
+st.set_page_config(page_title="Gemini Image Demo")
+
 os.getenv("GOOGLE_API_KEY")
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-## Function to load OpenAI model and get respones
-def get_gemini_response(input,image):
+
+## Function to load OpenAI model and get response
+def get_gemini_response(input, image):
     model = genai.GenerativeModel('gemini-1.5-pro')
-    if input!="":
-       response = model.generate_content([input,image])
+    if input != "":
+        response = model.generate_content([input, image])
     else:
-       response = model.generate_content(image)
+        response = model.generate_content(image)
     return response.text
-##initialize our streamlit app
-st.set_page_config(page_title="Gemini Image Demo")
+
+## Initialize our Streamlit app
 st.header("Food Analysis")
 
 input1 = """
-You are a food analysis AI specialized in Indian cuisine and image recognition. Given an image of an Indian food thali, your job is to analyze and report on three key aspects:  Dish Detection: Identify and label each distinct dish in the thali. Recognize common Indian food items (such as roti, rice, dal, curries, vegetables, pickles, and sweets) with high accuracy. Use visual details like shape, color, and garnish to distinguish between similar-looking items.  Freshness and Condition Assessment: Assess the freshness of each item by examining visual cues such as color, texture, and moisture. Indicate whether each item appears fresh, slightly stale, overcooked, or rotten, and use descriptive terms to specify the condition of each dish.  Quantity Estimation: Estimate the proportion of each item in the thali as a percentage of the overall plate content. Provide an approximate percentage for each detected item to reflect its quantity relative to the entire plate.  Present the results in a structured table format with columns for the Detected Dish Name, Freshness Status (e.g., Fresh, Slightly Stale, Overcooked, Rotten), and Quantity Percentage (e.g., Dal: 15%, Rice: 20%, etc.). Ensure accuracy in identifying each dish, assessing freshness, and estimating quantity.
+You are a specialized in food analysis for Indian cuisine and image recognition. Given an image of an Indian food thali, your job is to analyze and report on the following aspects:  Dish Detection: Identify and label each distinct dish in the thali. Recognize common Indian food items (such as roti, rice, dal, curries, vegetables, pickles, and sweets) with high accuracy. Use visual details like shape, color, and garnish to distinguish between similar-looking items.  Freshness and Condition Assessment: Assess the freshness of each item by examining visual cues such as color, texture, and moisture. Indicate whether each item appears fresh, slightly stale, overcooked, rotten, or spoiled. Use descriptive terms to specify the condition of each dish.  Quantity Estimation: Estimate the quantity of each item in the thali in grams and provide an approximate percentage for each detected item to reflect its quantity relative to the entire plate.  Sufficiency Assessment: Evaluate if the total quantity and variety of food items in the thali are sufficient for a 14-year-old child, considering standard dietary needs and portion sizes. Present the results in a structured table format with columns for the Detected Dish Name, Freshness Status (e.g., Fresh, Slightly Stale, Overcooked, Rotten, Spoiled), Quantity in Grams, and Quantity Percentage (e.g., Dal: 150g, 15%). Ensure accuracy in identifying each dish, assessing freshness, estimating quantity, and evaluating sufficiency.
 """
-input=st.text_input("Input Prompt: ",key="input", placeholder="Optional")
+input = st.text_input("Input Prompt: ", key="input", placeholder="Optional")
 uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
-image=""
+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("Tell me about the image")
-## If ask button is clicked
+
+submit = st.button("Check the Food Quality")
+
+## If submit button is clicked
 if submit:
-    response=get_gemini_response(input1 + input,image)
-    st.subheader("The Response is")
-    st.write(response)
+    with st.spinner("Please wait while assessing the food quality..."):
+        response = get_gemini_response(input1 + input, image)
+        st.subheader("The Response is")
+        st.write(f"Prompt Is : {input1 + input}")
+        st.write(response)