Update app.py

app.py

@@ -64,27 +64,24 @@ def cache_pil_image(image: Image.Image) -> str:
     image.save(image_path, "JPEG")
     return image_path
 
-def upload(files: Optional[List[gr.File]], chatbot: List[dict]) -> List[dict]:
-    if files:
-        for file in files:
-            image = Image.open(file.name).convert("RGB")
-            image_preview = preprocess_image(image)
-            if image_preview:
-                image_path = cache_pil_image(image)
-                chatbot.append({"role": "user", "content": f"Uploaded image: {image_path}"})
+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({"role": "user", "content": f"Uploaded image: {image_path}"})
     return chatbot
 
 def advanced_response(
-    text_prompt: str,
-    files: Optional[List[gr.File]],
+    files: Optional[List[str]],
     model_choice: str,
     system_instruction: str,
     chatbot: List[dict],
 ):
-    if not text_prompt.strip() and not files:
-        chatbot.append({"role": "assistant", "content": "Please provide a prompt or upload an image."})
-        yield chatbot
-        return
+    if not files:
+        return chatbot
 
     model = genai.GenerativeModel(
         model_name=model_choice,
@@ -98,17 +95,13 @@ def advanced_response(
     chat = model.start_chat(history=chatbot)
     chat.system_instruction = system_instruction
 
-    if text_prompt:
-        chatbot.append({"role": "user", "content": text_prompt})
-    if files:
-        images = [cache_pil_image(preprocess_image(Image.open(file.name))) for file in files]
-        chatbot.append({"role": "user", "content": f"Uploaded images: {', '.join(images)}"})
-
-    response = chat.send_message(text_prompt)
-    response.resolve()
+    images = [cache_pil_image(preprocess_image(Image.open(file))) for file in files]
+    response = chat.generate_content(images, stream=True)
 
-    chatbot.append({"role": "assistant", "content": response.text})
-    yield chatbot
+    chatbot.append({"role": "assistant", "content": ""})
+    for chunk in response:
+        chatbot[-1]["content"] += chunk.text
+        yield chatbot
 
 # Construcción de la interfaz con las dos pestañas originales
 with gr.Blocks() as demo:
@@ -122,6 +115,8 @@ with gr.Blocks() as demo:
             label="Selecciona el modelo",
         )
         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(
             placeholder="Escribe una instrucción para el sistema...",
             label="Instrucción del sistema",
@@ -129,10 +124,6 @@ with gr.Blocks() as demo:
             value="You are an assistant.",
         )
 
-        with gr.Row():
-            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)
-
         run_button_1.click(
             fn=bot_response,
             inputs=[model_dropdown_1, system_instruction_1, text_input_1, chatbot_1],
@@ -146,20 +137,18 @@ with gr.Blocks() as demo:
             label="Select Model",
         )
         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)
         system_instruction_2 = gr.Textbox(
             placeholder="Enter system instruction...",
             label="System Instruction",
             scale=8,
         )
 
-        with gr.Row():
-            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)
-
         run_button_2.click(
             fn=advanced_response,
-            inputs=[text_input_2, upload_button, model_dropdown_2, system_instruction_2, chatbot_2],
+            inputs=[upload_button, model_dropdown_2, system_instruction_2, chatbot_2],
             outputs=[chatbot_2],
         )
         upload_button.upload(