Update app.py

app.py

@@ -78,29 +78,36 @@ conn.close()
 # Separate section for image uploading
 st.title("Image Description Generator")
 
-uploaded_file = st.file_uploader("Upload an image here", type=["png", "jpg", "jpeg"])
-
-# Text input for asking questions about the image
-image_question = st.text_input("Ask something about the image:")
-
-if uploaded_file and image_question:
-   image_parts = [
-     {
-       "mime_type": uploaded_file.type,
-       "data": uploaded_file.read() 
-     },
-   ]
-
-   prompt_parts = [
-     image_question,
-     image_parts[0],
-   ]
-
-   model = genai.GenerativeModel(
-     model_name="gemini-pro-vision", 
-     generation_config=generation_config, 
-     safety_settings=safety_settings 
-   )
-   
-   response = model.generate_content(prompt_parts)
-   st.markdown(f"**Model's answer:** {response.text}")
+# Change the file_uploader to accept multiple files
+uploaded_files = st.file_uploader("Upload one or more images here", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
+
+# Text input for asking questions about the images
+image_question = st.text_input("Ask something about the images:")
+
+# Check if the user has entered a question
+if image_question:
+    # Create a list of image parts from the uploaded files
+    image_parts = []
+    for uploaded_file in uploaded_files:
+        image_parts.append({
+            "mime_type": uploaded_file.type,
+            "data": uploaded_file.read() 
+        })
+    
+    # Create a prompt parts list with the question and the image parts
+    prompt_parts = [image_question] + image_parts
+
+    # Use the gemini-pro-vision model to generate a response
+    model = genai.GenerativeModel(
+        model_name="gemini-pro-vision", 
+        generation_config=generation_config, 
+        safety_settings=safety_settings 
+    )
+    
+    response = model.generate_content(prompt_parts)
+    st.markdown(f"**Model's answer:** {response.text}")
+
+# Loop through the uploaded files and display them
+for uploaded_file in uploaded_files:
+    # Display the image
+    st.image(uploaded_file)