Update pages/5_Gemini-Chat.py

pages/5_Gemini-Chat.py

@@ -8,7 +8,7 @@ import google.generativeai as genai
 from io import BytesIO
 
 # Set your API key
-api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"  # Replace this with your actual API key
+api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"  # Replace with your actual API key
 genai.configure(api_key=api_key)
 
 # Configure the generative AI model
@@ -42,8 +42,6 @@ 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())
-if 'uploaded_files' not in st.session_state:  # Initialize uploaded_files in session state
-    st.session_state.uploaded_files = []
 
 st.title("Gemini Chatbot")
 
@@ -82,7 +80,7 @@ def get_chat_history_str():
     )
     return chat_history_str
 
-# Modified send_message function
+# Function to send messages
 def send_message():
     user_input = st.session_state.user_input
     prompts = []
@@ -93,8 +91,7 @@ def send_message():
         st.session_state.chat_history.append({"role": "user", "parts": [{"text": user_input}]})
         prompt_parts.append({"text": user_input})
 
-    # Handling uploaded files
-    uploaded_files = st.session_state.uploaded_files
+    # Handling uploaded files directly
     if uploaded_files:
         for uploaded_file in uploaded_files:
             base64_image = get_image_base64(Image.open(uploaded_file))
@@ -130,13 +127,11 @@ def send_message():
         st.audio(tts_file, format='audio/mp3')
 
     st.session_state.user_input = ''
-    st.session_state.uploaded_files = []
-    st.session_state.file_uploader_key = str(uuid.uuid4())
     display_chat_history()
 
 # UI components for user input, file uploader, send and clear buttons
 user_input = st.text_area("Enter your message here:", value="", key="user_input")
-uploaded_files = st.file_uploader("Upload images:", type=["png", "jpg", "jpeg"], accept_multiple_files=True, key="file_uploader_key", on_change=lambda: st.session_state.update(uploaded_files=uploaded_files))
+uploaded_files = st.file_uploader("Upload images:", type=["png", "jpg", "jpeg"], accept_multiple_files=True, key="file_uploader_key")
 send_button = st.button("Send", on_click=send_message)
 clear_button = st.button("Clear Conversation", on_click=clear_conversation)