Update app.py

app.py

@@ -14,12 +14,8 @@ from dotenv import load_dotenv
 # Cargar las variables de entorno desde el archivo .env
 load_dotenv()
 
-print("google-generativeai:", genai.__version__)
-
-# Obtener la clave de la API de las variables de entorno
+# Verificar la clave de API
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
-
-# Verificar que la clave de la API esté configurada
 if not GOOGLE_API_KEY:
     raise ValueError("GOOGLE_API_KEY is not set in environment variables.")
 
@@ -58,7 +54,7 @@ def user(text_prompt: str, chatbot: CHAT_HISTORY):
 def bot(
     files: Optional[List[str]],
     model_choice: str,
-    system_instruction: Optional[str],  # Parámetro para la instrucción del sistema
+    system_instruction: str,  # Instrucción del sistema extraída directamente
     chatbot: CHAT_HISTORY
 ):
     if not GOOGLE_API_KEY:
@@ -67,16 +63,16 @@ def bot(
     # Configurar la API con la clave
     genai.configure(api_key=GOOGLE_API_KEY)
     generation_config = genai.types.GenerationConfig(
-        temperature=0.7,  # Valor predeterminado
-        max_output_tokens=8192,  # Fijar el límite de tokens a 8,192
-        top_k=10,  # Valor predeterminado
-        top_p=0.9   # Valor predeterminado
+        temperature=0.7,
+        max_output_tokens=8192,
+        top_k=10,
+        top_p=0.9
     )
 
     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 []
-    
-    # Crear el modelo con la instrucción del sistema si está definida
+
+    # Usar la instrucción del sistema si está presente
     model = genai.GenerativeModel(
         model_name=model_choice,
         generation_config=generation_config,
@@ -93,17 +89,20 @@ def bot(
             time.sleep(0.01)
             yield chatbot
 
-# Componente para ingresar la instrucción del sistema
-system_instruction_component = gr.Textbox(
-    placeholder="Enter system instruction...", 
-    show_label=True, 
-    scale=8,
-    visible=False  # Por defecto no visible
+# Componente para la instrucción del sistema dentro de un desplegable
+system_instruction_dropdown = gr.Dropdown(
+    choices=[
+        "Keep responses concise and professional.",
+        "Use a friendly and engaging tone.",
+        "Focus on technical explanations.",
+        "Encourage creative ideas.",
+        "Simplify complex concepts for a beginner audience."
+    ],
+    label="System Instruction",
+    placeholder="Select or leave empty",
+    scale=2
 )
 
-def toggle_system_instruction(option: str):
-    return gr.update(visible=option == "Enable")
-
 # Definir los componentes de entrada y salida
 chatbot_component = gr.Chatbot(
     label='Gemini',
@@ -124,12 +123,6 @@ model_choice_component = gr.Dropdown(
     label="Select Model",
     scale=2
 )
-system_instruction_toggle = gr.Dropdown(
-    label="System Instruction",
-    choices=["Disable", "Enable"],
-    value="Disable",
-    scale=1
-)
 
 user_inputs = [
     text_prompt_component,
@@ -138,8 +131,8 @@ user_inputs = [
 
 bot_inputs = [
     upload_button_component,
-    model_choice_component,  # El campo de modelo está ahora arriba
-    system_instruction_component,  # Instrucción del sistema sigue siendo separada
+    model_choice_component,
+    system_instruction_dropdown,  # Desplegable de System Instruction
     chatbot_component
 ]
 
@@ -148,20 +141,13 @@ with gr.Blocks() as demo:
     gr.HTML(TITLE)
     gr.HTML(SUBTITLE)
     with gr.Column():
-        # Campo de selección de modelo arriba
         model_choice_component.render()
         chatbot_component.render()
         with gr.Row():
             text_prompt_component.render()
             upload_button_component.render()
             run_button_component.render()
-        system_instruction_toggle.render()
-        system_instruction_toggle.change(
-            fn=toggle_system_instruction,
-            inputs=[system_instruction_toggle],
-            outputs=[system_instruction_component]
-        )
-        system_instruction_component.render()
+        system_instruction_dropdown.render()  # Agregar desplegable para System Instruction
 
     run_button_component.click(
         fn=user,