Update app.py

app.py

@@ -72,7 +72,8 @@ def bot(
     stop_sequences: str,
     top_k: int,
     top_p: float,
-    chatbot: CHAT_HISTORY
+    chatbot: CHAT_HISTORY,
+    selected_model: str
 ):
     if not GOOGLE_API_KEY:
         raise ValueError("GOOGLE_API_KEY is not set.")
@@ -89,8 +90,9 @@ def bot(
 
     text_prompt = [chatbot[-1][0]] if chatbot and chatbot[-1][0] and isinstance(chatbot[-1][0], str) else []
     image_prompt = [preprocess_image(Image.open(file).convert('RGB')) for file in files] if files else []
-    model_name = 'gemini-pro-vision' if files else 'gemini-pro'
-    model = genai.GenerativeModel(model_name)
+    
+    # Usar el modelo seleccionado
+    model = genai.GenerativeModel(selected_model)
     response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
 
     chatbot[-1][1] = ""
@@ -101,6 +103,7 @@ def bot(
             time.sleep(0.01)
             yield chatbot
 
+# Componentes de la interfaz
 chatbot_component = gr.Chatbot(
     label='Gemini',
     bubble_full_width=False,
@@ -149,11 +152,24 @@ top_p_component = gr.Slider(
     label="Top-P",
 )
 
-user_inputs = [
-    text_prompt_component,
-    chatbot_component
-]
+# Selector de modelos Gemini
+gemini_model_component = gr.Dropdown(
+    choices=[
+        'gemini-1.5-flash',        # Fast and versatile performance
+        'gemini-1.5-flash-8b',     # High volume, lower intelligence tasks
+        'gemini-1.5-pro',          # Complex reasoning tasks
+        'gemini-exp-1114'          # Quality improvements
+    ],
+    value='gemini-1.5-flash',        # Valor por defecto
+    label="Select Gemini Model",
+    interactive=True
+)
+
+# Función para actualizar el modelo
+def update_gemini_model(selected_model):
+    return selected_model
 
+# Inputs para la función 'bot'
 bot_inputs = [
     upload_button_component,
     temperature_component,
@@ -161,9 +177,11 @@ bot_inputs = [
     stop_sequences_component,
     top_k_component,
     top_p_component,
-    chatbot_component
+    chatbot_component,
+    gemini_model_component
 ]
 
+# Crear la interfaz Gradio
 with gr.Blocks() as demo:
     gr.HTML(TITLE)
     gr.HTML(SUBTITLE)
@@ -182,24 +200,27 @@ with gr.Blocks() as demo:
                 top_k_component.render()
                 top_p_component.render()
 
+    # Ejecutar cuando se presiona el botón de ejecución
     run_button_component.click(
         fn=user,
-        inputs=user_inputs,
+        inputs=[text_prompt_component, chatbot_component],
         outputs=[text_prompt_component, chatbot_component],
         queue=False
     ).then(
         fn=bot, inputs=bot_inputs, outputs=[chatbot_component],
     )
 
+    # Acción cuando se envía un texto
     text_prompt_component.submit(
         fn=user,
-        inputs=user_inputs,
+        inputs=[text_prompt_component, chatbot_component],
         outputs=[text_prompt_component, chatbot_component],
         queue=False
     ).then(
         fn=bot, inputs=bot_inputs, outputs=[chatbot_component],
     )
 
+    # Acción cuando se suben imágenes
     upload_button_component.upload(
         fn=upload,
         inputs=[upload_button_component, chatbot_component],