Update app.py

app.py

@@ -3,38 +3,41 @@ from PIL import Image
 import io
 import base64
 
-# Placeholder for google.generativeai package
 import google.generativeai as genai
 
 # Configure the API key (should be set as an environment variable or secure storage in production)
-api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
+api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"  # Replace with your actual API key
 genai.configure(api_key=api_key)
 
-generation_config = {
-    "temperature": 0.9,
-    "max_output_tokens": 3000
-}
+generation_config = genai.GenerationConfig(
+    temperature=0.9,
+    max_output_tokens=3000
+)
 
 safety_settings = []
 
-# Initialize session state
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
+# Initialize session state for chat history
+if 'chat_history' not in st.session_state:
+    st.session_state['chat_history'] = []
 
 # UI layout
 st.title("Gemini Chatbot")
 
 # Display the chat history
-for idx, message in enumerate(st.session_state.chat_history):
+for message in st.session_state['chat_history']:
     role, text = message
-    key = f"message_{idx}_{role}"  # Unique key for each message
-    st.text_area(f"{role.title()} says:", value=text, height=75, disabled=True, key=key)
+    st.markdown(f"**{role.title()}:** {text}")
 
 # Text input for the user to send messages
 user_input = st.text_input("Enter your message here:", key="user_input")
 
 # File uploader for images
-uploaded_files = st.file_uploader("Upload images:", type=["png", "jpg", "jpeg"], accept_multiple_files=True, key="file_uploader")
+uploaded_files = st.file_uploader(
+    "Upload images:",
+    type=["png", "jpg", "jpeg"],
+    accept_multiple_files=True,
+    key="file_uploader"
+)
 
 # Function to convert image to base64
 def get_image_base64(image):
@@ -45,47 +48,48 @@ def get_image_base64(image):
 
 # When the 'Send' button is clicked, process the input and generate a response
 if st.button("Send", key="send_button"):
-    # Save user input to the chat history if it's not empty
-    if user_input.strip():
-        st.session_state.chat_history.append(("user", 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
-    image_parts = []
     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)
-            image_parts.append({
-                "mime_type": uploaded_file.type,
-                "data": image_base64
-            })
-            st.session_state.chat_history.append(("user", f"Uploaded image: {uploaded_file.name}"))
-
-    # Prepare the prompts for the model
-    prompts = [{"role": "user", "parts": [{"text": user_input}]}] if user_input.strip() else []
-    for image_part in image_parts:
-        prompts.append({"role": "user", "parts": [image_part]})
-
-    # 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)
-    
-    # Corrected response handling
-    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(("model", response_text))
-
-# Redisplay the updated chat history
-for idx, message in enumerate(st.session_state.chat_history):
+            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'] = ""
+
+# Re-display the chat history to include the new messages
+for message in st.session_state['chat_history']:
     role, text = message
-    key = f"message_{idx}_{role}"  # Unique key for each message
-    st.text_area(f"{role.title()} says:", value=text, height=75, disabled=True, key=key)
+    st.markdown(f"**{role.title()}:** {text}")