Update app.py

app.py

@@ -3,6 +3,7 @@ from PIL import Image
 import io
 import base64
 
+# Assuming google.generativeai as genai is the correct import based on your description
 import google.generativeai as genai
 
 # Configure the API key (should be set as an environment variable or secure storage in production)
@@ -53,41 +54,38 @@ if st.button("Send", key="send_button"):
         if user_input:
             st.session_state['chat_history'].append(("User", user_input))
 
-        if uploaded_files:
-            images_prompts = []
-            for uploaded_file in uploaded_files:
-                bytes_data = uploaded_file.read()
-                image = Image.open(io.BytesIO(bytes_data))
-                image_base64 = get_image_base64(image)
-                st.session_state['chat_history'].append(("User", f"Uploaded image: {uploaded_file.name}"))
-                image_prompt = {"role": "user", "parts": [{"mime_type": uploaded_file.type, "data": image_base64}]}
-                images_prompts.append(image_prompt)
-            
-            # Use the Gemini model for vision
-            model_name = 'gemini-pro-vision'
-            model = genai.GenerativeModel(
-                model_name=model_name,
-                generation_config=generation_config,
-                safety_settings=safety_settings
-            )
-            response = model.generate_content(images_prompts)
-            response_text = response.text if hasattr(response, "text") else "No response text found."
-            st.session_state['chat_history'].append(("Gemini", response_text))
-        elif user_input:
-            # Use the Gemini model for text
-            model_name = 'gemini-pro'
+        # Process text input
+        if user_input and not uploaded_files:
+            prompts = [{"role": "user", "parts": [{"text": user_input}]}]
             model = genai.GenerativeModel(
-                model_name=model_name,
+                model_name='gemini-pro',
                 generation_config=generation_config,
                 safety_settings=safety_settings
             )
-            prompts = [{"role": "user", "parts": [{"text": user_input}]}]
             response = model.generate_content(prompts)
             response_text = response.text if hasattr(response, "text") else "No response text found."
             st.session_state['chat_history'].append(("Gemini", response_text))
 
-    # Clear the input box after sending the message
-    st.session_state['user_input'] = ""
+        # Process and save uploaded images to the chat history
+        if uploaded_files:
+            for uploaded_file in uploaded_files:
+                bytes_data = uploaded_file.read()
+                image = Image.open(io.BytesIO(bytes_data))
+                image_base64 = get_image_base64(image)
+                st.session_state['chat_history'].append(("User", f"Uploaded image: {uploaded_file.name}"))
+
+                image_prompt = {"role": "user", "parts": [{"mime_type": uploaded_file.type, "data": image_base64}]}
+                model = genai.GenerativeModel(
+                    model_name='gemini-pro-vision',
+                    generation_config=generation_config,
+                    safety_settings=safety_settings
+                )
+                response = model.generate_content([image_prompt])
+                response_text = response.text if hasattr(response, "text") else "No response text found."
+                st.session_state['chat_history'].append(("Gemini", response_text))
+
+        # Clear the input box after sending the message
+        st.session_state['user_input'] = ""
 
 # Re-display the chat history to include the new messages
 for message in st.session_state['chat_history']: