Update app.py

app.py

@@ -1,37 +1,34 @@
 from dotenv import load_dotenv
-load_dotenv() ##loading all the envrionment variables from the .env file
+load_dotenv()  # loading all the envrionment variables from the .env file
 
-import streamlit as st
+import gradio as gr
 import os
 import google.generativeai as genai
 from PIL import Image
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-## function to load Gemimi pro model and get responses
+# function to load Gemini Pro model and get responses
 model = genai.GenerativeModel("gemini-1.5-flash")
 
 def get_gemini_response(input, image):
-    if input !="":
+    if input != "":
         response = model.generate_content([input, image])
     else:
         response = model.generate_content(image)
     return response.text
 
-st.set_page_config(page_title="Gemini Image Demo", page_icon=":robot:")
-st.header("Gemini Mulimodal Bot")
-input=st.text_input("Enter your question here:", key="input")
-
-uploaded_file = st.file_uploader("Choose an image...", type=["png", "jpg", "jpeg"])
-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 submit is clickec
-if submit:
-    response=get_gemini_response(input, image)
-    st.subheader("The response is")
-    st.write(response)
+# Define the Gradio interface
+iface = gr.Interface(
+    fn=get_gemini_response,
+    inputs=[
+        gr.Textbox(label="Enter your question here:"),
+        gr.Image(label="Choose an image...", type="pil"),
+    ],
+    outputs=gr.Textbox(label="The response is"),
+    title="Gemini Multimodal Bot",
+    description="Ask Gemini questions about images",
+)
+
+# Launch the Gradio interface
+iface.launch(share=True)