Update app.py

app.py

@@ -8,8 +8,6 @@ import google.generativeai as genai
 
 genai.configure(api_key='AIzaSyCeNgXfZx0kJ736XFVtxXxev_RdscB0i5s')
 
-## Function to load OpenAI model and get respones
-
 def get_gemini_response(input,image,prompt):
     model = genai.GenerativeModel('gemini-pro-vision')
     response = model.generate_content([input,image[0],prompt])
@@ -19,12 +17,12 @@ def get_gemini_response(input,image,prompt):
 def input_image_setup(uploaded_file):
     # Check if a file has been uploaded
     if uploaded_file is not None:
-        # Read the file into bytes
+        
         bytes_data = uploaded_file.getvalue()
 
         image_parts = [
             {
-                "mime_type": uploaded_file.type,  # Get the mime type of the uploaded file
+                "mime_type": uploaded_file.type,  
                 "data": bytes_data
             }
         ]
@@ -33,7 +31,7 @@ def input_image_setup(uploaded_file):
         raise FileNotFoundError("No file uploaded")
 
 
-##initialize our streamlit app
+##streamlit app
 
 st.set_page_config(page_title="Gemini Image Demo")
 
@@ -43,7 +41,7 @@ image=""
 if uploaded_file is not None:
     image = Image.open(uploaded_file)
     st.image(image, caption="Uploaded Image.", use_column_width=True)
-input=st.text_input("Input Prompt(optional): ",key="input")
+#input=st.text_input("Input Prompt(optional): ",key="input")
 
 
 submit=st.button("Submit")
@@ -56,10 +54,8 @@ input_prompt = """
                Constraints: Do not include missing information. Maintain appropriate data types (e.g., phone numbers as strings).
                """
 
-## If ask button is clicked
-
 if submit:
     image_data = input_image_setup(uploaded_file)
-    response=get_gemini_response(input_prompt,image_data,input)
+    response=get_gemini_response(input_prompt,image_data)
     st.subheader("Output :")
     st.write(response)