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)
@@ -23,24 +24,9 @@ if 'chat_history' not in st.session_state:
 # UI layout
 st.title("Gemini Chatbot")
 
-# Display the chat history
-for message in st.session_state['chat_history']:
-    role, text = message
-    st.markdown(f"**{role.title()}**: {text}")
-
-# Multiline text input for the user to send messages
-user_input = st.text_area("Enter your message here:", key="user_input")
-
-# File uploader for images
-uploaded_files = st.file_uploader(
-    "Upload images:",
-    type=["png", "jpg", "jpeg"],
-    accept_multiple_files=True,
-    key="file_uploader"
-)
-
 # Function to convert image to base64
 def get_image_base64(image):
+    image = image.convert("RGB")  # Convert to RGB to remove alpha channel if present
     buffered = io.BytesIO()
     image.save(buffered, format="JPEG")
     img_str = base64.b64encode(buffered.getvalue()).decode()
@@ -48,6 +34,9 @@ def get_image_base64(image):
 
 # Callback function to send message and clear input
 def send_message():
+    user_input = st.session_state['user_input']
+    uploaded_files = st.session_state['file_uploader']
+    
     if user_input or uploaded_files:
         # Save user input to the chat history
         if user_input:
@@ -68,8 +57,7 @@ def send_message():
         # 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 = Image.open(uploaded_file).convert("RGB")  # Ensure image is in RGB
                 image_base64 = get_image_base64(image)
                 st.session_state['chat_history'].append(("User", f"Uploaded image: {uploaded_file.name}"))
 
@@ -86,10 +74,21 @@ def send_message():
         # Clear the input box after sending the message
         st.session_state.user_input = ""
 
-# When the 'Send' button is clicked, call the send_message callback
-send_button = st.button("Send", key="send_button", on_click=send_message)
+# Multiline text input for the user to send messages
+user_input = st.text_area("Enter your message here:", key="user_input", value="")
+
+# File uploader for images
+uploaded_files = st.file_uploader(
+    "Upload images:",
+    type=["png", "jpg", "jpeg"],
+    accept_multiple_files=True,
+    key="file_uploader"
+)
+
+# Button to send the message
+send_button = st.button("Send", on_click=send_message)
 
-# Re-display the chat history to include the new messages
+# Display the chat history
 for message in st.session_state['chat_history']:
     role, text = message
     st.markdown(f"**{role.title()}**: {text}")