Update app.py

app.py

@@ -72,8 +72,7 @@ def bot(
     stop_sequences: str,
     top_k: int,
     top_p: float,
-    chatbot: CHAT_HISTORY,
-    selected_model: str
+    chatbot: CHAT_HISTORY
 ):
     if not GOOGLE_API_KEY:
         raise ValueError("GOOGLE_API_KEY is not set.")
@@ -90,9 +89,8 @@ 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 []
-    
-    # Usar el modelo seleccionado
-    model = genai.GenerativeModel(selected_model)
+    model_name = 'gemini-pro-vision' if files else 'gemini-pro'
+    model = genai.GenerativeModel(model_name)
     response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
 
     chatbot[-1][1] = ""
@@ -103,7 +101,6 @@ def bot(
             time.sleep(0.01)
             yield chatbot
 
-# Componentes de la interfaz
 chatbot_component = gr.Chatbot(
     label='Gemini',
     bubble_full_width=False,
@@ -152,24 +149,11 @@ top_p_component = gr.Slider(
     label="Top-P",
 )
 
-# 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
+user_inputs = [
+    text_prompt_component,
+    chatbot_component
+]
 
-# Inputs para la función 'bot'
 bot_inputs = [
     upload_button_component,
     temperature_component,
@@ -177,11 +161,9 @@ bot_inputs = [
     stop_sequences_component,
     top_k_component,
     top_p_component,
-    chatbot_component,
-    gemini_model_component
+    chatbot_component
 ]
 
-# Crear la interfaz Gradio
 with gr.Blocks() as demo:
     gr.HTML(TITLE)
     gr.HTML(SUBTITLE)
@@ -200,27 +182,24 @@ 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=[text_prompt_component, chatbot_component],
+        inputs=user_inputs,
         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=[text_prompt_component, chatbot_component],
+        inputs=user_inputs,
         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],