Update app.py

app.py

@@ -1,28 +1,26 @@
-TITLE = """<h1 align="center">Gemini Playground ✨</h1>"""
-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, Union
+
+from dotenv import load_dotenv
 from PIL import Image
+import numpy as np
 import google.generativeai as genai
 import gradio as gr
-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
+# Configuración de claves
 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.")
 
-# Configuración del modelo Gemini
+# Configuración de Gemini
+genai.configure(api_key=GOOGLE_API_KEY)
+
+# Parámetros comunes para el modelo
 generation_config = {
     "temperature": 1,
     "top_p": 0.95,
@@ -31,98 +29,120 @@ generation_config = {
     "response_mime_type": "text/plain",
 }
 
-genai.configure(api_key=GOOGLE_API_KEY)
-
-model = genai.GenerativeModel(
-    model_name="gemini-1.5-flash", 
-    generation_config=generation_config
-)
-
-# Inicializar la sesión de chat
-chat = model.start_chat(history=[])
-
-# Función para transformar el historial de Gradio al formato de Gemini
-def transform_history(history):
-    new_history = []
-    for chat_entry in history:
-        new_history.append({"parts": [{"text": chat_entry[0]}], "role": "user"})
-        new_history.append({"parts": [{"text": chat_entry[1]}], "role": "model"})
-    return new_history
-
-# Función de respuesta que maneja el historial
-def bot_response(
-    model_choice: str,
-    system_instruction: str,
-    text_prompt: str,
-    chatbot: list,
-) -> Tuple[list, str]:
-    """
-    Envía el mensaje al modelo, obtiene la respuesta y actualiza el historial.
-    """
-    if not text_prompt.strip():
-        return chatbot, "Por favor, escribe un mensaje válido."
+# Inicialización del modelo
+model = genai.GenerativeModel(model_name="gemini-1.5-flash", generation_config=generation_config)
 
-    # Transformar el historial al formato que espera Gemini
-    transformed_history = transform_history(chatbot)
+# Título y subtítulo
+TITLE = """<h1 align="center">Gemini Playground ✨</h1>"""
+SUBTITLE = """<h2 align="center">Play with Gemini Pro and Gemini Pro Vision</h2>"""
 
-    # Configurar el modelo
-    chat.history = transformed_history
+# Función para transformar el historial de chat
+def transform_history(history: List[Tuple[str, str]]) -> list:
+    return [
+        {"parts": [{"text": entry[0]}], "role": "user"} if i % 2 == 0 else {"parts": [{"text": entry[1]}], "role": "model"}
+        for i, entry in enumerate(history)
+    ]
 
-    # Enviar el mensaje y obtener la respuesta
+# Código para la pestaña 1: Chat básico con Gemini
+def bot_response(model_choice: str, system_instruction: str, text_prompt: str, chatbot: list) -> Tuple[list, str]:
+    if not text_prompt.strip():
+        return chatbot, "Por favor, escribe un mensaje válido."
+    chat = model.start_chat(history=transform_history(chatbot))
+    chat.system_instruction = system_instruction
     response = chat.send_message(text_prompt)
     response.resolve()
-
-    # Obtener el texto generado por el modelo
-    generated_text = response.text
-
-    # Actualizar el historial con la pregunta y la respuesta
-    chatbot.append((text_prompt, generated_text))
-
+    chatbot.append((text_prompt, response.text))
     return chatbot, ""
 
-# Componentes de la interfaz
-chatbot_component = gr.Chatbot(label="Gemini", scale=2, height=300)
-text_input_component = gr.Textbox(placeholder="Escribe un mensaje...", show_label=False, scale=8)
-run_button_component = gr.Button(value="Enviar", variant="primary", scale=1)
-model_dropdown_component = gr.Dropdown(
-    choices=["gemini-1.5-flash", "gemini-2.0-flash-exp", "gemini-1.5-pro"],
-    value="gemini-1.5-flash",
-    label="Selecciona el modelo",
-    scale=2
-)
-system_instruction_component = gr.Textbox(
-    placeholder="Escribe una instrucción para el sistema...",
-    label="Instrucción del sistema",
-    scale=8,
-    value="You are an assistant."
-)
-
-# Definir la interfaz
+# Código para la pestaña 2: Chat con imágenes y configuración avanzada
+IMAGE_CACHE_DIRECTORY = "/tmp"
+IMAGE_WIDTH = 512
+
+def preprocess_image(image: Image.Image) -> Optional[Image.Image]:
+    if image:
+        image_height = int(image.height * IMAGE_WIDTH / image.width)
+        return image.resize((IMAGE_WIDTH, image_height))
+
+def cache_pil_image(image: Image.Image) -> str:
+    image_filename = f"{uuid.uuid4()}.jpeg"
+    os.makedirs(IMAGE_CACHE_DIRECTORY, exist_ok=True)
+    image_path = os.path.join(IMAGE_CACHE_DIRECTORY, image_filename)
+    image.save(image_path, "JPEG")
+    return image_path
+
+def upload(files: Optional[List[str]], chatbot: List[Tuple[Optional[Union[Tuple[str], str]], Optional[str]]]) -> list:
+    for file in files:
+        image = Image.open(file).convert("RGB")
+        image_preview = preprocess_image(image)
+        if image_preview:
+            gr.Image(image_preview).render()
+        image_path = cache_pil_image(image)
+        chatbot.append(((image_path,), None))
+    return chatbot
+
+def advanced_response(
+    files: Optional[List[str]], model_choice: str, system_instruction: str, chatbot: list
+):
+    if not files:
+        return chatbot
+    chat = model.start_chat(history=transform_history(chatbot))
+    chat.system_instruction = system_instruction
+    images = [cache_pil_image(preprocess_image(Image.open(file))) for file in files]
+    response = chat.generate_content(images, stream=True, generation_config=generation_config)
+    chatbot[-1][1] = ""
+    for chunk in response:
+        chatbot[-1][1] += chunk.text
+        yield chatbot
+
+# Construcción de la interfaz
 with gr.Blocks() as demo:
     gr.HTML(TITLE)
     gr.HTML(SUBTITLE)
-    with gr.Column():
-        model_dropdown_component.render()
-        chatbot_component.render()
-        with gr.Row():
-            text_input_component.render()
-            run_button_component.render()
-        with gr.Accordion("Instrucción del sistema", open=False):
-            system_instruction_component.render()
-
-    # Configurar eventos
-    run_button_component.click(
-        fn=bot_response,
-        inputs=[model_dropdown_component, system_instruction_component, text_input_component, chatbot_component],
-        outputs=[chatbot_component, text_input_component],
-    )
-
-    text_input_component.submit(
-        fn=bot_response,
-        inputs=[model_dropdown_component, system_instruction_component, text_input_component, chatbot_component],
-        outputs=[chatbot_component, text_input_component],
-    )
-
-# Lanzar la aplicación
+    with gr.Tab("Chat Básico"):
+        gr.Markdown("### Chat con Gemini Pro")
+        model_dropdown = gr.Dropdown(
+            choices=["gemini-1.5-flash", "gemini-2.0-flash-exp", "gemini-1.5-pro"],
+            value="gemini-1.5-flash",
+            label="Selecciona el modelo"
+        )
+        chatbot_basic = gr.Chatbot(label="Chat Básico", height=300)
+        text_input = gr.Textbox(placeholder="Escribe tu mensaje...", show_label=False)
+        system_instruction = gr.Textbox(
+            placeholder="Instrucción del sistema...",
+            label="Instrucción del sistema",
+            value="You are an assistant."
+        )
+        send_button = gr.Button("Enviar")
+        send_button.click(
+            bot_response,
+            inputs=[model_dropdown, system_instruction, text_input, chatbot_basic],
+            outputs=[chatbot_basic, text_input]
+        )
+    with gr.Tab("Chat Avanzado con Imágenes"):
+        gr.Markdown("### Interacción con imágenes y Gemini Pro Vision")
+        model_dropdown_adv = gr.Dropdown(
+            choices=["gemini-1.5-flash", "gemini-2.0-flash-exp", "gemini-1.5-pro"],
+            value="gemini-1.5-flash",
+            label="Selecciona el modelo"
+        )
+        chatbot_advanced = gr.Chatbot(label="Chat Avanzado", height=300)
+        text_input_adv = gr.Textbox(placeholder="Mensaje o descripción...")
+        system_instruction_adv = gr.Textbox(
+            placeholder="Instrucción del sistema...",
+            label="Instrucción del sistema"
+        )
+        upload_button = gr.UploadButton(label="Subir Imágenes", file_types=["image"])
+        process_button = gr.Button("Procesar")
+        process_button.click(
+            advanced_response,
+            inputs=[upload_button, model_dropdown_adv, system_instruction_adv, chatbot_advanced],
+            outputs=[chatbot_advanced]
+        )
+        upload_button.upload(
+            upload,
+            inputs=[upload_button, chatbot_advanced],
+            outputs=[chatbot_advanced]
+        )
+
 if __name__ == "__main__":
-    demo.queue(max_size=99).launch(debug=True, show_error=True)
+    demo.launch(debug=True)