Update app.py

app.py

@@ -84,6 +84,7 @@ def user(text_prompt: str, chatbot: CHAT_HISTORY):
 
 def bot(
     google_key: str,
+    model_name: str,
     files: Optional[List[str]],
     temperature: float,
     max_output_tokens: int,
@@ -114,14 +115,14 @@ def bot(
             if chatbot[-1][0] and isinstance(chatbot[-1][0], str) \
             else []
         image_prompt = [Image.open(file).convert('RGB') for file in files]
-        model = genai.GenerativeModel('gemini-1.5-pro-exp-0801')
+        model = genai.GenerativeModel(model_name)
         response = model.generate_content(
             text_prompt + image_prompt,
             stream=True,
             generation_config=generation_config)
     else:
         messages = preprocess_chat_history(chatbot)
-        model = genai.GenerativeModel('gemini-1.5-pro-exp-0801')
+        model = genai.GenerativeModel(model_name)
         response = model.generate_content(
             messages,
             stream=True,
@@ -136,6 +137,15 @@ def bot(
             time.sleep(0.01)
             yield chatbot
 
+model_selection = gr.Dropdown(
+    ["gemini-1.0-pro",
+     "gemini-1.5-flash",
+     "gemini-1.5-pro",
+     "gemini-1.5-pro-exp-0801"
+    ],
+    label="Select Gemini Model",
+    value="gemini-1.5-pro"
+)
 
 google_key_component = gr.Textbox(
     label="GOOGLE API KEY",
@@ -224,6 +234,7 @@ user_inputs = [
 
 bot_inputs = [
     google_key_component,
+    model_selection,
     upload_button_component,
     temperature_component,
     max_output_tokens_component,