Update app.py

app.py

@@ -58,6 +58,7 @@ def user(text_prompt: str, chatbot: CHAT_HISTORY):
 def bot(
     files: Optional[List[str]],
     model_choice: str,
+    system_instruction: str,  # Añadido el parámetro para la instrucción del sistema
     chatbot: CHAT_HISTORY
 ):
     if not GOOGLE_API_KEY:
@@ -74,7 +75,14 @@ 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 = genai.GenerativeModel(model_choice)
+    
+    # Crear el modelo con la instrucción del sistema
+    model = genai.GenerativeModel(
+        model_name=model_choice,
+        generation_config=generation_config,
+        system_instruction=system_instruction  # Se pasa la instrucción del sistema
+    )
+
     response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
 
     chatbot[-1][1] = ""
@@ -85,6 +93,12 @@ 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
+)
+
+# Definir los componentes de entrada y salida
 chatbot_component = gr.Chatbot(
     label='Gemini',
     bubble_full_width=False,
@@ -113,9 +127,11 @@ user_inputs = [
 bot_inputs = [
     upload_button_component,
     model_choice_component,
+    system_instruction_component,  # Añadido el campo de instrucción del sistema
     chatbot_component
 ]
 
+# Definir la interfaz de usuario
 with gr.Blocks() as demo:
     gr.HTML(TITLE)
     gr.HTML(SUBTITLE)
@@ -126,6 +142,7 @@ with gr.Blocks() as demo:
             upload_button_component.render()
             run_button_component.render()
         model_choice_component.render()
+        system_instruction_component.render()  # Mostrar el campo de la instrucción del sistema
 
     run_button_component.click(
         fn=user,
@@ -152,4 +169,5 @@ with gr.Blocks() as demo:
         queue=False
     )
 
+# Lanzar la aplicación
 demo.queue(max_size=99).launch(debug=False, show_error=True)