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"
+api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"  # Replace this with your actual API key
 genai.configure(api_key=api_key)
 
 # Configure the generative AI model
@@ -39,9 +39,11 @@ safety_settings = [
 
 # Initialize session state
 if 'chat_history' not in st.session_state:
-    st.session_state['chat_history'] = []
+    st.session_state.chat_history = []
 if 'file_uploader_key' not in st.session_state:
-    st.session_state['file_uploader_key'] = str(uuid.uuid4())
+    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")
 
@@ -50,11 +52,6 @@ st.markdown("""
 **AI Planner System Prompt:** As the AI Planner, your primary task is to assist in the development of a coherent and engaging book. You will be responsible for organizing the overall structure, defining the plot or narrative, and outlining the chapters or sections. To accomplish this, you will need to use your understanding of storytelling principles and genre conventions, as well as any specific information provided by the user, to create a well-structured framework for the book.
 """)
 
-# System message to be included in the prompts sent to the AI model
-ai_planner_system_prompt = """
-AI Planner System Prompt: As the AI Planner, your primary task is to assist in the development of a coherent and engaging book. You will be responsible for organizing the overall structure, defining the plot or narrative, and outlining the chapters or sections. To accomplish this, you will need to use your understanding of storytelling principles and genre conventions, as well as any specific information provided by the user, to create a well-structured framework for the book.
-"""
-
 # Helper functions for image processing and chat history management
 def get_image_base64(image):
     image = image.convert("RGB")
@@ -64,11 +61,11 @@ def get_image_base64(image):
     return img_str
 
 def clear_conversation():
-    st.session_state['chat_history'] = []
-    st.session_state['file_uploader_key'] = str(uuid.uuid4())
+    st.session_state.chat_history = []
+    st.session_state.file_uploader_key = str(uuid.uuid4())
 
 def display_chat_history():
-    for entry in st.session_state['chat_history']:
+    for entry in st.session_state.chat_history:
         role = entry["role"]
         parts = entry["parts"][0]
         if 'text' in parts:
@@ -80,7 +77,7 @@ def get_chat_history_str():
     chat_history_str = "\n".join(
         f"{entry['role'].title()}: {part['text']}" if 'text' in part
         else f"{entry['role'].title()}: (Image)"
-        for entry in st.session_state['chat_history']
+        for entry in st.session_state.chat_history
         for part in entry['parts']
     )
     return chat_history_str
@@ -88,31 +85,22 @@ def get_chat_history_str():
 # Modified send_message function
 def send_message():
     user_input = st.session_state.user_input
-    uploaded_files = st.session_state.uploaded_files
-    prompts = [ai_planner_system_prompt]  # Start with the AI Planner System Prompt
+    prompts = []
     prompt_parts = []
 
-    # Populate the prompts list with the existing chat history
-    for entry in st.session_state['chat_history']:
-        for part in entry['parts']:
-            if 'text' in part:
-                prompts.append(part['text'])
-            elif 'data' in part:
-                # Add the image in base64 format to prompt_parts for vision model
-                prompt_parts.append({"data": part['data'], "mime_type": "image/jpeg"})
-                prompts.append("[Image]")
-
     if user_input:
         prompts.append(user_input)
-        st.session_state['chat_history'].append({"role": "user", "parts": [{"text": user_input}]})
+        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
     if uploaded_files:
         for uploaded_file in uploaded_files:
             base64_image = get_image_base64(Image.open(uploaded_file))
             prompts.append("[Image]")
             prompt_parts.append({"data": base64_image, "mime_type": "image/jpeg"})
-            st.session_state['chat_history'].append({
+            st.session_state.chat_history.append({
                 "role": "user",
                 "parts": [{"mime_type": uploaded_file.type, "data": base64_image}]
             })
@@ -134,7 +122,7 @@ def send_message():
     response_text = response.text if hasattr(response, "text") else "No response text found."
 
     if response_text:
-        st.session_state['chat_history'].append({"role": "model", "parts":[{"text": response_text}]})
+        st.session_state.chat_history.append({"role": "model", "parts":[{"text": response_text}]})
         tts = gTTS(text=response_text, lang='en')
         tts_file = BytesIO()
         tts.write_to_fp(tts_file)
@@ -146,9 +134,9 @@ def send_message():
     st.session_state.file_uploader_key = str(uuid.uuid4())
     display_chat_history()
 
-# UI components
+# 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=st.session_state.file_uploader_key)
+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))
 send_button = st.button("Send", on_click=send_message)
 clear_button = st.button("Clear Conversation", on_click=clear_conversation)