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)
@@ -25,55 +24,12 @@ if 'chat_history' not in st.session_state:
 st.title("Gemini Chatbot")
 
 # Display the chat history
-for idx, (role, text) in enumerate(st.session_state['chat_history']):
-    st.markdown(f"**{role.title()}:** {text}")
+for message in st.session_state['chat_history']:
+    role, text = message
+    st.markdown(f"**{role.title()}**: {text}")
 
-# 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))
-
-        # Assuming the generative model accepts text input
-        prompts = [{"role": "user", "parts": [{"text": user_input}]}]
-
-        # Use the appropriate Gemini model based on the inputs
-        model_name = 'gemini-pro-vision' if uploaded_files else 'gemini-pro'
-        model = genai.GenerativeModel(
-            model_name=model_name,
-            generation_config=generation_config,
-            safety_settings=safety_settings
-        )
-
-        # Generate the response
-        response = model.generate_content(prompts)
-        response_text = response.text if hasattr(response, "text") else "No response text found."
-
-        # Save the model response to the chat history
-        st.session_state['chat_history'].append(("Gemini", response_text))
-
-    # Process and save uploaded images to the chat history
-    if uploaded_files:
-        for uploaded_file in uploaded_files:
-            bytes_data = uploaded_file.read()
-            image = Image.open(io.BytesIO(bytes_data))
-            image_base64 = get_image_base64(image)
-            st.session_state['chat_history'].append(("User", f"Uploaded image: {uploaded_file.name}"))
-            
-            # Assuming the generative model accepts image input
-            image_prompt = {"role": "user", "parts": [{"mime_type": uploaded_file.type, "data": image_base64}]}
-            response = model.generate_content([image_prompt])
-            response_text = response.text if hasattr(response, "text") else "No response text found."
-            
-            # Save the model response to the chat history
-            st.session_state['chat_history'].append(("Gemini", response_text))
-
-    # Clear the input box after sending the message
-    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)
+# Multiline text input for the user to send messages
+user_input = st.text_area("Enter your message here:", key="user_input")
 
 # File uploader for images
 uploaded_files = st.file_uploader(
@@ -90,6 +46,50 @@ def get_image_base64(image):
     img_str = base64.b64encode(buffered.getvalue()).decode()
     return img_str
 
-# 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)
+# When the 'Send' button is clicked, process the input and generate a response
+if st.button("Send", key="send_button"):
+    if user_input or uploaded_files:
+        # Save user input to the chat history
+        if user_input:
+            st.session_state['chat_history'].append(("User", user_input))
+
+        if uploaded_files:
+            images_prompts = []
+            for uploaded_file in uploaded_files:
+                bytes_data = uploaded_file.read()
+                image = Image.open(io.BytesIO(bytes_data))
+                image_base64 = get_image_base64(image)
+                st.session_state['chat_history'].append(("User", f"Uploaded image: {uploaded_file.name}"))
+                image_prompt = {"role": "user", "parts": [{"mime_type": uploaded_file.type, "data": image_base64}]}
+                images_prompts.append(image_prompt)
+            
+            # Use the Gemini model for vision
+            model_name = 'gemini-pro-vision'
+            model = genai.GenerativeModel(
+                model_name=model_name,
+                generation_config=generation_config,
+                safety_settings=safety_settings
+            )
+            response = model.generate_content(images_prompts)
+            response_text = response.text if hasattr(response, "text") else "No response text found."
+            st.session_state['chat_history'].append(("Gemini", response_text))
+        elif user_input:
+            # Use the Gemini model for text
+            model_name = 'gemini-pro'
+            model = genai.GenerativeModel(
+                model_name=model_name,
+                generation_config=generation_config,
+                safety_settings=safety_settings
+            )
+            prompts = [{"role": "user", "parts": [{"text": user_input}]}]
+            response = model.generate_content(prompts)
+            response_text = response.text if hasattr(response, "text") else "No response text found."
+            st.session_state['chat_history'].append(("Gemini", response_text))
+
+    # Clear the input box after sending the message
+    st.session_state['user_input'] = ""
+
+# 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}")