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)
@@ -24,30 +25,12 @@ if 'chat_history' not in st.session_state:
 st.title("Gemini Chatbot")
 
 # Display the chat history
-for message in st.session_state['chat_history']:
-    role, text = message
+for idx, (role, text) in enumerate(st.session_state['chat_history']):
     st.markdown(f"**{role.title()}:** {text}")
 
-# Text input for the user to send messages
-user_input = st.text_input("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):
-    buffered = io.BytesIO()
-    image.save(buffered, format="JPEG")
-    img_str = base64.b64encode(buffered.getvalue()).decode()
-    return img_str
-
-# When the 'Send' button is clicked, process the input and generate a response
-if st.button("Send", key="send_button"):
+# Function to handle sending messages
+def send_message():
+    user_input = st.session_state.user_input
     if user_input:
         # Save user input to the chat history
         st.session_state['chat_history'].append(("User", user_input))
@@ -87,9 +70,26 @@ if st.button("Send", key="send_button"):
             st.session_state['chat_history'].append(("Gemini", response_text))
 
     # Clear the input box after sending the message
-    st.session_state['user_input'] = ""
+    st.session_state.user_input = ""
+
+# Text input for the user to send messages
+user_input = st.text_input("Enter your message here:", key="user_input", on_change=send_message)
+
+# 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):
+    buffered = io.BytesIO()
+    image.save(buffered, format="JPEG")
+    img_str = base64.b64encode(buffered.getvalue()).decode()
+    return img_str
 
-# Re-display the chat history to include the new messages
-for message in st.session_state['chat_history']:
-    role, text = message
-    st.markdown(f"**{role.title()}:** {text}")
+# Button to send the message
+# The button now doesn't need to do anything since the text input's on_change callback handles the logic
+st.button("Send", key="send_button", on_click=send_message)