Update app.py

app.py

@@ -265,9 +265,10 @@ def main():
             process_text(st.session_state.current_user['name'], text_input)
     elif option == "Image":
         text_input = st.text_input("Enter text prompt to use with Image context:")
-        image_input = st.file_uploader("Upload an image", type=["png"])
-        if image_input:
-            process_image(st.session_state.current_user['name'], image_input, text_input)
+        uploaded_files = st.file_uploader("Upload images", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
+        for image_input in uploaded_files:
+            image_bytes = image_input.read()
+            process_image(st.session_state.current_user['name'], image_bytes, text_input)
     elif option == "Audio":
         text_input = st.text_input("Enter text prompt to use with Audio context:")
         uploaded_files = st.file_uploader("Upload an audio file", type=["mp3", "wav"], accept_multiple_files=True)