Update app.py

app.py

@@ -1,35 +1,28 @@
-#Import libraries
+# 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 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):
+# Function to load Google Gemini Vision model and get response
+def get_response_code(image, prompt):
     model = genai.GenerativeModel('gemini-pro-vision')
     response = model.generate_content([image[0], prompt])
     return response.text
 
-#Preprocess image data
+# Function to preprocess image data
 def prep_image(uploaded_file):
-    #Check if there is any data
+    # Check if there is any data
     if uploaded_file is not None:
-        #Read the file as bytes
+        # Read the file as bytes
         bytes_data = uploaded_file.getvalue()
 
-        #get the image part information
+        # Get the image part information
         image_parts = [
             {
                 "mime_type": uploaded_file.type,
@@ -39,65 +32,29 @@ def prep_image(uploaded_file):
         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('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.
-    Score yourself in terms of Accuracy and Precision on the calories you predict
-    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.
-    Score yourself in terms of Accuracy and Precision on calories you predict
-
-    Return the response using markdown. Also score yourself in terms of accuracy and Precision 
-
-    """
-    ##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)
+# Configuring Streamlit App
+st.set_page_config(page_title="Code Interpreter")
+st.image('LOGO.jpg', width=70)
+st.header("Code Interpreter")
+
+# Section for Code Interpreter
+upload_code_file = st.file_uploader("Choose an image of code...", type=["jpg", "jpeg", "png"])
+if upload_code_file is not None:
+    # Show the uploaded code image
+    code_image = Image.open(upload_code_file)
+    st.image(code_image, caption="Uploaded Code Image", use_column_width=True)
+
+# Prompt Template for code interpretation
+input_prompt_code = """
+You are an expert in coding. The image contains a piece of code. Your task is to interpret the code and solve the problem it presents.
+Provide an analysis of the code, its functionality, and any potential improvements or solutions.
+"""
+
+# Button for code interpretation
+submit = st.button("Interpret Code!")
+if submit:
+    code_image_data = prep_image(upload_code_file)
+    response = get_response_code(code_image_data, input_prompt_code)
+    st.subheader("Code AI: ")
+    st.write(response)