Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from PIL import Image
 import io
 import base64
+import uuid
 
 # Assuming google.generativeai as genai is the correct import based on your description
 import google.generativeai as genai
@@ -17,9 +18,11 @@ generation_config = genai.GenerationConfig(
 
 safety_settings = []
 
-# Initialize session state for chat history
+# Initialize session state for chat history and file uploader key
 if 'chat_history' not in st.session_state:
     st.session_state['chat_history'] = []
+if 'file_uploader_key' not in st.session_state:
+    st.session_state['file_uploader_key'] = str(uuid.uuid4())
 
 # UI layout
 st.title("Gemini Chatbot")
@@ -34,8 +37,8 @@ def get_image_base64(image):
 
 # Function to send message and clear input
 def send_message():
-    user_input = st.session_state['user_input']
-    uploaded_files = st.session_state['file_uploader']
+    user_input = st.session_state.user_input
+    uploaded_files = st.session_state.uploaded_files
     
     if user_input or uploaded_files:
         # Save user input to the chat history
@@ -65,9 +68,9 @@ def send_message():
         response_text = response.text if hasattr(response, "text") else "No response text found."
         st.session_state['chat_history'].append({"role": "model", "parts": [{"text": response_text}]})
 
-        # Clear the input box after sending the message
-        st.session_state.user_input = ""
-        st.session_state.file_uploader = None  # Clear uploaded files
+        # Clear the user input and generate a new key for the file uploader widget to reset it
+        st.session_state.user_input = ''
+        st.session_state.file_uploader_key = str(uuid.uuid4())
 
 # Multiline text input for the user to send messages
 user_input = st.text_area("Enter your message here:", key="user_input", value="")
@@ -77,7 +80,7 @@ uploaded_files = st.file_uploader(
     "Upload images:",
     type=["png", "jpg", "jpeg"],
     accept_multiple_files=True,
-    key="file_uploader"
+    key=st.session_state.file_uploader_key
 )
 
 # Button to send the message
@@ -90,4 +93,7 @@ for entry in st.session_state['chat_history']:
     if 'text' in parts:
         st.markdown(f"**{role.title()}**: {parts['text']}")
     elif 'data' in parts:
-        st.markdown(f"**{role.title()}**: (Image)")
+        st.markdown(f"**{role.title()}**: (Image)")
+
+# Ensure the file_uploader widget state is tied to the randomly generated key
+st.session_state.uploaded_files = uploaded_files