Update app.py

app.py

@@ -61,7 +61,7 @@ def handle_uploaded_images(files):
     """Procesa las imágenes cargadas, las redimensiona y las guarda"""
     image_paths = []
     for file in files:
-        image = Image.open(file).convert('RGB')  # Abrir y convertir la imagen a RGB
+        image = Image.open(file.name).convert('RGB')  # Abrir y convertir la imagen a RGB
         image_height = int(image.height * 512 / image.width)  # Mantener la proporción
         image_resized = image.resize((512, image_height))  # Redimensionar la imagen
         image_filename = f"{uuid.uuid4()}.jpeg"  # Crear un nombre único para la imagen
@@ -79,33 +79,34 @@ def count_images(message, history):
             total_images += 1
     return f"You just uploaded {num_images} images, total uploaded: {total_images + num_images}"
 
-# Función que se invoca al cargar imágenes
-def upload_images_handler(files, history):
-    # Procesar las imágenes cargadas
-    image_paths = handle_uploaded_images(files)
-
-    # Actualizar el historial con las imágenes procesadas
-    history.append((None, image_paths))
-
-    return history, count_images({"files": files}, history)
+# Función que maneja el texto y las imágenes
+def handle_message(message, history):
+    if message.get("files"):
+        # Procesar las imágenes cargadas
+        image_paths = handle_uploaded_images(message["files"])
+        # Añadir las imágenes al historial
+        history.append((None, image_paths))
+        return history, count_images(message, history)
+    
+    # Si no hay archivos, procesamos el mensaje como texto
+    return response(message, history)
 
 # Crear la interfaz de Gradio
-demo = gr.ChatInterface(
-    fn=response,  # Función de chat para manejar texto y archivos
+demo = gr.Interface(
+    fn=handle_message,  # Función de manejo de mensaje y archivos
+    inputs=[
+        gr.Textbox(placeholder="Escribe un mensaje..."),
+        gr.File(label="Subir imágenes", file_count="multiple", file_types=["image"])
+    ],
+    outputs=[
+        gr.Chatbot(type="messages"),  # Formato actualizado para mensajes con "role" y "content"
+        gr.Textbox(label="Conteo de imágenes")
+    ],
     examples=[  # Ejemplos iniciales de mensajes
         {"text": "No files", "files": []}
     ],
-    cache_examples=True,
-    multimodal=True,  # Activar la modalidad multimodal
-    textbox=gr.MultimodalTextbox(  # Configuración del cuadro de texto multimodal
-        file_count="multiple",  # Permitir múltiples archivos
-        file_types=["image"],  # Aceptar solo imágenes
-        sources=["upload", "microphone"]  # Fuentes de entrada: carga de archivos y micrófono
-    )
+    cache_examples=True
 )
 
-# Conectar la función de carga de imágenes con el evento de Gradio
-demo.add_event("file_upload", upload_images_handler)
-
 # Iniciar la interfaz
 demo.launch()