Update app.py

app.py

@@ -4,7 +4,7 @@ SUBTITLE = """<h2 align="center">Play with Gemini Pro and Gemini Pro Vision</h2>
 import os
 import time
 import uuid
-from typing import List, Tuple, Optional, Dict, Union
+from typing import List, Tuple, Optional, Union
 
 import google.generativeai as genai
 import gradio as gr
@@ -44,6 +44,7 @@ def upload(files: Optional[List[str]], chatbot: CHAT_HISTORY) -> CHAT_HISTORY:
         image = Image.open(file).convert('RGB')
         image_preview = preprocess_image(image)
         if image_preview:
+            # Display a preview of the uploaded image
             gr.Image(image_preview).render()
         image_path = cache_pil_image(image)
         chatbot.append(((image_path,), None))
@@ -57,26 +58,26 @@ def user(text_prompt: str, chatbot: CHAT_HISTORY):
 def bot(
     files: Optional[List[str]],
     temperature: float,
-    max_output_tokens: int,
     top_k: int,
     top_p: float,
+    model_choice: str,
     chatbot: CHAT_HISTORY
 ):
     if not GOOGLE_API_KEY:
         raise ValueError("GOOGLE_API_KEY is not set.")
 
+    # Configurar la API con la clave
     genai.configure(api_key=GOOGLE_API_KEY)
     generation_config = genai.types.GenerationConfig(
         temperature=temperature,
-        max_output_tokens=max_output_tokens,
+        max_output_tokens=8192,  # Fijar el límite de tokens a 8,192
         top_k=top_k,
         top_p=top_p
     )
 
     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-1.5-flash'
-    model = genai.GenerativeModel(model_name)
+    model = genai.GenerativeModel(model_choice)
     response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
 
     chatbot[-1][1] = ""
@@ -107,13 +108,6 @@ temperature_component = gr.Slider(
     step=0.05,
     label="Temperature",
 )
-max_output_tokens_component = gr.Slider(
-    minimum=1,
-    maximum=2048,
-    value=1024,
-    step=1,
-    label="Token limit",
-)
 top_k_component = gr.Slider(
     minimum=1,
     maximum=40,
@@ -128,6 +122,12 @@ top_p_component = gr.Slider(
     step=0.01,
     label="Top-P",
 )
+model_choice_component = gr.Dropdown(
+    choices=["gemini-1.5-flash", "gemini-2.0-flash-exp", "gemini-1.5-pro"],
+    value="gemini-1.5-flash",
+    label="Select Model",
+    scale=2
+)
 
 user_inputs = [
     text_prompt_component,
@@ -137,9 +137,9 @@ user_inputs = [
 bot_inputs = [
     upload_button_component,
     temperature_component,
-    max_output_tokens_component,
     top_k_component,
     top_p_component,
+    model_choice_component,
     chatbot_component
 ]
 
@@ -154,10 +154,10 @@ with gr.Blocks() as demo:
             run_button_component.render()
         with gr.Accordion("Parameters", open=False):
             temperature_component.render()
-            max_output_tokens_component.render()
             with gr.Accordion("Advanced", open=False):
                 top_k_component.render()
                 top_p_component.render()
+                model_choice_component.render()
 
     run_button_component.click(
         fn=user,