Update app.py

app.py

@@ -1,3 +1,6 @@
+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
@@ -30,19 +33,13 @@ generation_config = {
 
 genai.configure(api_key=GOOGLE_API_KEY)
 
-# Inicializar los modelos para ambas pestañas
-model_with_images = genai.GenerativeModel(
-    model_name="gemini-1.5-flash",
-    generation_config=generation_config
-)
-
-model_text_only = genai.GenerativeModel(
-    model_name="gemini-1.5-flash",
+model = genai.GenerativeModel(
+    model_name="gemini-1.5-flash", 
     generation_config=generation_config
 )
 
-# Inicializar la sesión de chat para el chatbot sin imágenes
-chat_text_only = model_text_only.start_chat(history=[])
+# 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):
@@ -52,7 +49,7 @@ def transform_history(history):
         new_history.append({"parts": [{"text": chat_entry[1]}], "role": "model"})
     return new_history
 
-# Función de respuesta que maneja el historial para el chatbot sin imágenes
+# Función de respuesta que maneja el historial
 def bot_response(
     model_choice: str,
     system_instruction: str,
@@ -69,10 +66,10 @@ def bot_response(
     transformed_history = transform_history(chatbot)
 
     # Configurar el modelo
-    chat_text_only.history = transformed_history
+    chat.history = transformed_history
 
     # Enviar el mensaje y obtener la respuesta
-    response = chat_text_only.send_message(text_prompt)
+    response = chat.send_message(text_prompt)
     response.resolve()
 
     # Obtener el texto generado por el modelo
@@ -83,158 +80,49 @@ def bot_response(
 
     return chatbot, ""
 
-# Funciones para manejar el chatbot con imágenes
-def preprocess_image(image: Image.Image) -> Optional[Image.Image]:
-    if image:
-        image_height = int(image.height * 512 / image.width)
-        return image.resize((512, image_height))
-
-def cache_pil_image(image: Image.Image) -> str:
-    image_filename = f"{uuid.uuid4()}.jpeg"
-    os.makedirs("/tmp", exist_ok=True)
-    image_path = os.path.join("/tmp", image_filename)
-    image.save(image_path, "JPEG")
-    return image_path
-
-def upload(files: Optional[List[str]], chatbot: list) -> 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 user(text_prompt: str, chatbot: list):
-    if text_prompt:
-        chatbot.append((text_prompt, None))
-    return "", chatbot
-
-def bot(
-    files: Optional[List[str]],
-    model_choice: str,
-    system_instruction: Optional[str],
-    chatbot: list
-):
-    if not GOOGLE_API_KEY:
-        raise ValueError("GOOGLE_API_KEY is not set.")
-
-    genai.configure(api_key=GOOGLE_API_KEY)
-    generation_config = genai.types.GenerationConfig(
-        temperature=0.7,
-        max_output_tokens=8192,
-        top_k=10,
-        top_p=0.9
-    )
-
-    if not system_instruction:
-        system_instruction = "1"
-
-    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_with_images = genai.GenerativeModel(
-        model_name=model_choice,
-        generation_config=generation_config,
-        system_instruction=system_instruction
-    )
-
-    response = model_with_images.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
-
-    chatbot[-1][1] = ""
-    for chunk in response:
-        for i in range(0, len(chunk.text), 10):
-            section = chunk.text[i:i + 10]
-            chatbot[-1][1] += section
-            time.sleep(0.01)
-            yield chatbot
-
-# Interfaces
-TITLE = """<h1 align="center">Gemini Playground ✨</h1>"""
-SUBTITLE = """<h2 align="center">Play with Gemini Pro and Gemini Pro Vision</h2>"""
-
-# Componentes comunes
-chatbot_component_with_images = gr.Chatbot(label='Gemini with Images', scale=2, height=300)
-chatbot_component_text_only = gr.Chatbot(label='Gemini Text Only', scale=2, height=300)
-text_prompt_component = gr.Textbox(placeholder="Message...", show_label=False, autofocus=True, scale=8)
-run_button_component = gr.Button(value="Run", variant="primary", scale=1)
-upload_button_component = gr.UploadButton(label="Upload Images", file_count="multiple", file_types=["image"], scale=1)
-
-# Componentes separados para cada pestaña
-model_choice_component_text_only = 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
-)
-
-model_choice_component_with_images = gr.Dropdown(
+# 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="Select Model",
+    label="Selecciona el modelo",
     scale=2
 )
-
 system_instruction_component = gr.Textbox(
-    placeholder="Enter system instruction...",
-    show_label=True,
-    scale=8
+    placeholder="Escribe una instrucción para el sistema...",
+    label="Instrucción del sistema",
+    scale=8,
+    value="You are an assistant."
 )
 
+# Definir la interfaz
 with gr.Blocks() as demo:
     gr.HTML(TITLE)
     gr.HTML(SUBTITLE)
-    with gr.Tabs():
-        with gr.TabItem("Chatbot with Images"):
-            with gr.Column():
-                model_choice_component_with_images
-                chatbot_component_with_images
-                with gr.Row():
-                    text_prompt_component
-                    upload_button_component
-                    run_button_component
-                with gr.Accordion("System Instruction", open=False):
-                    system_instruction_component
-
-            run_button_component.click(
-                fn=user,
-                inputs=[text_prompt_component, chatbot_component_with_images],
-                outputs=[text_prompt_component, chatbot_component_with_images],
-                queue=False
-            ).then(
-                fn=bot,
-                inputs=[upload_button_component, model_choice_component_with_images, system_instruction_component, chatbot_component_with_images],
-                outputs=[chatbot_component_with_images],
-            )
-
-            upload_button_component.upload(
-                fn=upload,
-                inputs=[upload_button_component, chatbot_component_with_images],
-                outputs=[chatbot_component_with_images],
-                queue=False
-            )
-
-        with gr.TabItem("Chatbot Text Only"):
-            with gr.Column():
-                model_choice_component_text_only
-                chatbot_component_text_only
-                with gr.Row():
-                    text_prompt_component
-                    run_button_component
-                with gr.Accordion("System Instruction", open=False):
-                    system_instruction_component
-
-            run_button_component.click(
-                fn=bot_response,
-                inputs=[model_choice_component_text_only, system_instruction_component, text_prompt_component, chatbot_component_text_only],
-                outputs=[chatbot_component_text_only, text_prompt_component],
-            )
-
-            text_prompt_component.submit(
-                fn=bot_response,
-                inputs=[model_choice_component_text_only, system_instruction_component, text_prompt_component, chatbot_component_text_only],
-                outputs=[chatbot_component_text_only, text_prompt_component],
-            )
-
-demo.queue(max_size=99).launch(debug=True, show_error=True)
+    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
+if __name__ == "__main__":
+    demo.queue(max_size=99).launch(debug=True, show_error=True)