Update app.py

app.py

@@ -2,12 +2,10 @@ 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()
@@ -17,44 +15,47 @@ GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 if not GOOGLE_API_KEY:
     raise ValueError("GOOGLE_API_KEY is not set in environment variables.")
 
-# 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,
-    "top_k": 40,
-    "max_output_tokens": 8192,
-    "response_mime_type": "text/plain",
-}
-
-# Inicialización del modelo
-model = genai.GenerativeModel(model_name="gemini-1.5-flash", generation_config=generation_config)
-
-# 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>"""
 
-# 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)
-    ]
-
-# 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]:
+# Código de la pestaña 1
+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
+
+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."
+
+    model = genai.GenerativeModel(
+        model_name=model_choice,
+        generation_config={
+            "temperature": 1,
+            "top_p": 0.95,
+            "top_k": 40,
+            "max_output_tokens": 8192,
+        },
+    )
     chat = model.start_chat(history=transform_history(chatbot))
     chat.system_instruction = system_instruction
+
     response = chat.send_message(text_prompt)
     response.resolve()
+
     chatbot.append((text_prompt, response.text))
     return chatbot, ""
 
-# Código para la pestaña 2: Chat con imágenes y configuración avanzada
+# Código de la pestaña 2
 IMAGE_CACHE_DIRECTORY = "/tmp"
 IMAGE_WIDTH = 512
 
@@ -70,7 +71,7 @@ def cache_pil_image(image: Image.Image) -> str:
     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:
+def upload(files: Optional[List[str]], chatbot: list) -> list:
     for file in files:
         image = Image.open(file).convert("RGB")
         image_preview = preprocess_image(image)
@@ -81,68 +82,87 @@ def upload(files: Optional[List[str]], chatbot: List[Tuple[Optional[Union[Tuple[
     return chatbot
 
 def advanced_response(
-    files: Optional[List[str]], model_choice: str, system_instruction: str, chatbot: list
+    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))
+
+    model = genai.GenerativeModel(
+        model_name=model_choice,
+        generation_config={
+            "temperature": 0.7,
+            "max_output_tokens": 8192,
+            "top_k": 10,
+            "top_p": 0.9,
+        },
+    )
+    chat = model.start_chat(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)
+    response = chat.generate_content(images, stream=True)
+
     chatbot[-1][1] = ""
     for chunk in response:
         chatbot[-1][1] += chunk.text
         yield chatbot
 
-# Construcción de la interfaz
+# Construcción de la interfaz con las dos pestañas originales
 with gr.Blocks() as demo:
     gr.HTML(TITLE)
     gr.HTML(SUBTITLE)
-    with gr.Tab("Chat Básico"):
-        gr.Markdown("### Chat con Gemini Pro")
-        model_dropdown = gr.Dropdown(
+
+    with gr.Tab("Pestaña 1"):
+        model_dropdown_1 = 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"
+            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...",
+        chatbot_1 = gr.Chatbot(label="Gemini", scale=2, height=300)
+        text_input_1 = gr.Textbox(placeholder="Escribe un mensaje...", show_label=False, scale=8)
+        run_button_1 = gr.Button(value="Enviar", variant="primary", scale=1)
+        system_instruction_1 = gr.Textbox(
+            placeholder="Escribe una instrucción para el sistema...",
             label="Instrucción del sistema",
-            value="You are an assistant."
+            scale=8,
+            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]
+
+        run_button_1.click(
+            fn=bot_response,
+            inputs=[model_dropdown_1, system_instruction_1, text_input_1, chatbot_1],
+            outputs=[chatbot_1, text_input_1],
         )
-    with gr.Tab("Chat Avanzado con Imágenes"):
-        gr.Markdown("### Interacción con imágenes y Gemini Pro Vision")
-        model_dropdown_adv = gr.Dropdown(
+
+    with gr.Tab("Pestaña 2"):
+        model_dropdown_2 = 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"
+            label="Select Model",
         )
-        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"
+        chatbot_2 = gr.Chatbot(label="Gemini", height=300)
+        text_input_2 = gr.Textbox(placeholder="Message or description...", show_label=False, scale=8)
+        upload_button = gr.UploadButton(label="Upload Images", file_count="multiple", file_types=["image"])
+        run_button_2 = gr.Button(value="Run", variant="primary", scale=1)
+        system_instruction_2 = gr.Textbox(
+            placeholder="Enter system instruction...",
+            label="System Instruction",
+            scale=8,
         )
-        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]
+
+        run_button_2.click(
+            fn=advanced_response,
+            inputs=[upload_button, model_dropdown_2, system_instruction_2, chatbot_2],
+            outputs=[chatbot_2],
         )
         upload_button.upload(
-            upload,
-            inputs=[upload_button, chatbot_advanced],
-            outputs=[chatbot_advanced]
+            fn=upload,
+            inputs=[upload_button, chatbot_2],
+            outputs=[chatbot_2],
         )
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch(debug=True, queue=True)