Update app.py

app.py

@@ -3,7 +3,6 @@ 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)
@@ -47,8 +46,8 @@ def get_image_base64(image):
     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"):
+# Callback function to send message and clear input
+def send_message():
     if user_input or uploaded_files:
         # Save user input to the chat history
         if user_input:
@@ -85,7 +84,10 @@ 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 = ""
+
+# When the 'Send' button is clicked, call the send_message callback
+send_button = st.button("Send", key="send_button", on_click=send_message)
 
 # Re-display the chat history to include the new messages
 for message in st.session_state['chat_history']: