Update app.py

app.py

@@ -8,13 +8,11 @@ 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):
+# Function to load OpenAI model and get responses
+def get_gemini_response(input, image, prompt):
     model = genai.GenerativeModel('gemini-pro-vision')
-    response = model.generate_content([input,image[0],prompt])
+    response = model.generate_content([input, image[0], prompt])
     return response.text
-   
 
 def input_image_setup(uploaded_file):
     # Check if a file has been uploaded
@@ -32,43 +30,37 @@ def input_image_setup(uploaded_file):
     else:
         raise FileNotFoundError("No file uploaded")
 
-
-##initialize our streamlit app
-
+# Initialize our Streamlit app
 st.set_page_config(page_title="Gemini Image Demo")
 
 st.header("Generative AI : Business card Reader")
+input_prompt = textwrap.dedent("""
+    You are an expert in understanding business cards.
+    You will receive input images of business card & you will have to answer questions based on the input image.
+    You have to extract information from business card images and give correct tag to the output text 
+    like person name, company name, occupation, address, telephone number, mobile number, email, website, etc. Give output in json format.
+    """)
+
 # Display sample input images
 sample_images_folder = pathlib.Path(__file__).parent / "examples"
-sample_images = [sample_image for sample_image in sample_images_folder.glob("*.jpg") if sample_image.is_file()]
+sample_images = [sample_image.name for sample_image in sample_images_folder.glob("*.jpg") if sample_image.is_file()]
+
+sample_image_uploaded = st.sidebar.file_uploader("Choose a sample image", type=["jpg", "jpeg"], options=sample_images)
 
-if len(sample_images) > 0:
-    st.subheader("Sample Input Images")
-    for i, sample_image in enumerate(sample_images):
-        image = Image.open(sample_image)
-        st.image(image, caption=f"Sample Image {i+1}", use_column_width=True)
-        
-input=st.text_input("Input Prompt: ",key="input")
-uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
-image=""  
+if sample_image_uploaded:
+    uploaded_file = sample_image_uploaded
+else:
+    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+
+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("Submit")
-
-input_prompt = """
-               You are an expert in understanding business cards.
-               You will receive input images of business card & you will have to answer questions based on the input image.
-               You have to extract information from business card images and give correct tag to the output text 
-               like person name, company name, occupation, address, telephone number, mobile number, email, website, etc. Give output in json format.
-               """
-
-## If ask button is clicked
+submit = st.button("Submit")
 
 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, input)
     st.subheader("Output :")
     st.write(response)