Update app.py

app.py

@@ -5,10 +5,10 @@ import base64
 import uuid
 from gtts import gTTS
 import google.generativeai as genai
-from io import BytesIO  # Import BytesIO
+from io import BytesIO
 
 # Set your API key
-api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"  # Replace with your actual API key
+api_key = "AIzaSyAHD0FwX-Ds6Y3eI-i5Oz7IdbJqR6rN7pg"  # Replace with your actual API key
 genai.configure(api_key=api_key)
 
 # Configure the generative AI model
@@ -45,6 +45,9 @@ if 'file_uploader_key' not in st.session_state:
 
 st.title("Gemini Chatbot")
 
+# Model Selection Dropdown
+selected_model = st.selectbox("Select a Gemini 1.5 model:", ["gemini-1.5-flash-latest", "gemini-1.5-pro-latest"])
+
 # Helper functions for image processing and chat history management
 def get_image_base64(image):
     image = image.convert("RGB")
@@ -110,17 +113,23 @@ def send_message():
                 "parts": [{"mime_type": uploaded_file.type, "data": base64_image}]
             })
 
-    # Determine if vision model should be used
+    # Determine if vision model should be used 
     use_vision_model = any(part.get('mime_type') == 'image/jpeg' for part in prompt_parts)
 
-    # Set up the model and generate a response
-    model_name = 'gemini-pro-vision' if use_vision_model else 'gemini-pro'
+    # Use the selected model 
+    model_name = selected_model
+    if use_vision_model and "pro" not in model_name:
+        st.warning(f"The selected model ({model_name}) does not support image inputs. Choose a 'pro' model for image capabilities.")
+        return
+    
     model = genai.GenerativeModel(
         model_name=model_name,
         generation_config=generation_config,
         safety_settings=safety_settings
     )
+
     chat_history_str = "\n".join(prompts)
+
     if use_vision_model:
         # Include text and images for vision model
         generated_prompt = {"role": "user", "parts": prompt_parts}