Update app.py

app.py

@@ -1,7 +1,6 @@
 import os
-import time
 import uuid
-from typing import List, Tuple, Optional, Union
+from typing import List, Optional, Union
 from PIL import Image
 import google.generativeai as genai
 import gradio as gr
@@ -21,19 +20,12 @@ TITLE = """<h1 align="center">Gemini Playground ✨</h1>"""
 SUBTITLE = """<h2 align="center">Play with Gemini Pro and Gemini Pro Vision</h2>"""
 
 # 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]:
+    chatbot: List[dict],
+) -> tuple:
     if not text_prompt.strip():
         return chatbot, "Por favor, escribe un mensaje válido."
 
@@ -46,13 +38,14 @@ def bot_response(
             "max_output_tokens": 8192,
         },
     )
-    chat = model.start_chat(history=transform_history(chatbot))
+    chat = model.start_chat(history=chatbot)
     chat.system_instruction = system_instruction
 
     response = chat.send_message(text_prompt)
     response.resolve()
 
-    chatbot.append((text_prompt, response.text))
+    chatbot.append({"role": "user", "content": text_prompt})
+    chatbot.append({"role": "assistant", "content": response.text})
     return chatbot, ""
 
 # Código de la pestaña 2
@@ -71,21 +64,21 @@ def cache_pil_image(image: Image.Image) -> str:
     image.save(image_path, "JPEG")
     return image_path
 
-def upload(files: Optional[List[str]], chatbot: list) -> list:
+def upload(files: Optional[List[str]], chatbot: List[dict]) -> List[dict]:
     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))
+        chatbot.append({"role": "user", "content": f"Uploaded image: {image_path}"})
     return chatbot
 
 def advanced_response(
     files: Optional[List[str]],
     model_choice: str,
     system_instruction: str,
-    chatbot: list,
+    chatbot: List[dict],
 ):
     if not files:
         return chatbot
@@ -105,9 +98,9 @@ def advanced_response(
     images = [cache_pil_image(preprocess_image(Image.open(file))) for file in files]
     response = chat.generate_content(images, stream=True)
 
-    chatbot[-1][1] = ""
+    chatbot.append({"role": "assistant", "content": ""})
     for chunk in response:
-        chatbot[-1][1] += chunk.text
+        chatbot[-1]["content"] += chunk.text
         yield chatbot
 
 # Construcción de la interfaz con las dos pestañas originales
@@ -121,7 +114,7 @@ with gr.Blocks() as demo:
             value="gemini-1.5-flash",
             label="Selecciona el modelo",
         )
-        chatbot_1 = gr.Chatbot(label="Gemini", scale=2, height=300)
+        chatbot_1 = gr.Chatbot(label="Gemini", scale=2, height=300, type="messages")
         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(
@@ -143,7 +136,7 @@ with gr.Blocks() as demo:
             value="gemini-1.5-flash",
             label="Select Model",
         )
-        chatbot_2 = gr.Chatbot(label="Gemini", height=300)
+        chatbot_2 = gr.Chatbot(label="Gemini", height=300, type="messages")
         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)
@@ -165,4 +158,4 @@ with gr.Blocks() as demo:
         )
 
 if __name__ == "__main__":
-    demo.launch(debug=True, queue=True)
+    demo.launch(debug=True)