Multiple image support

app.py

@@ -84,30 +84,30 @@ uploaded_files = st.file_uploader("Upload one or more images here", type=["png",
 # 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)
-
-    # Check if the user has entered a question
-    if 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 for {uploaded_file.name}:** {response.text}")