Update app.py

app.py

@@ -148,7 +148,7 @@ def chat(message, history):
     except Exception as e:
         return f"Error: {e}"
 
-# Remove the first chat function as it's duplicated
+# Eliminamos la primera función chat duplicada
 def process_file(file):
     file_path = file.name
     file_extension = os.path.splitext(file_path)[1].lower()
@@ -158,33 +158,44 @@ def process_file(file):
         with open(file_path, 'r', encoding='utf-8') as f:
             content = f.read()
     else:
-        content = f"Archivo subido: {os.path.basename(file_path)} (Este tipo de archivo no puede ser procesado para mostrar su contenido)"
+        content = f"Archivo subido: {os.path.basename(file_path)}"
     
     return content
 
 def chat(message, history, file=None):
     try:
-        file_content = ""
-        if file is not None:
-            file_content = f"\n\nArchivo subido: {os.path.basename(file.name)}\n"
-            file_content += process_file(file)
-        
-        # Combinar el mensaje del usuario con el contenido del archivo
+        # Preparamos el mensaje combinado solo si hay un archivo y el mensaje está vacío
         combined_message = message
-        if file_content:
-            combined_message += f"\n\nContenido del archivo:\n{file_content}"
+        if file is not None and message.strip() == "":
+            combined_message = "Analiza este archivo"
+        elif file is not None:
+            # Solo añadimos una nota sobre el archivo, no su contenido completo
+            combined_message += f"\n\n(Se ha subido un archivo: {os.path.basename(file.name)})"
+        
+        # Construimos el historial de mensajes para el modelo
+        formatted_history = []
+        for msg in history:
+            if isinstance(msg, dict):
+                if msg["role"] == "user":
+                    formatted_history.append({"role": "user", "parts": [msg["content"]]})
+                else:
+                    formatted_history.append({"role": "model", "parts": [msg["content"]]})
+            else:
+                formatted_history.append({"role": "user", "parts": [msg[0]]})
+                formatted_history.append({"role": "model", "parts": [msg[1]]})
         
         messages = [
             {"role": "user", "parts": [system_prompt]},
-            *[{"role": "user", "parts": [msg["content"]]} if isinstance(msg, dict) else {"role": "user", "parts": [msg[0]]} for msg in history],
+            *formatted_history,
             {"role": "user", "parts": [combined_message]}
         ]
+        
         response = model.generate_content(messages)
         
-        # Return in the format expected by gr.Chatbot with type="messages"
-        return [{"role": "user", "content": combined_message}, {"role": "assistant", "content": response.text}]
+        # Devolvemos en el formato esperado por gr.Chatbot con type="messages"
+        return history + [{"role": "user", "content": message}, {"role": "assistant", "content": response.text}]
     except Exception as e:
-        return [{"role": "user", "content": combined_message}, {"role": "assistant", "content": f"Error: {e}"}]
+        return history + [{"role": "user", "content": message}, {"role": "assistant", "content": f"Error: {e}"}]
 
 with gr.Blocks(title="🤖Chucho Bot - CopyXpert Sales Assistant") as demo:
     gr.Markdown("# 🤖Chucho Bot - CopyXpert Sales Assistant")
@@ -192,7 +203,7 @@ with gr.Blocks(title="🤖Chucho Bot - CopyXpert Sales Assistant") as demo:
     
     with gr.Row():
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(type="messages")
+            chatbot = gr.Chatbot(type="messages", height=500)
             msg = gr.Textbox(
                 placeholder="Escribe tu pregunta aquí...",
                 container=False,
@@ -207,7 +218,8 @@ with gr.Blocks(title="🤖Chucho Bot - CopyXpert Sales Assistant") as demo:
             file_upload = gr.File(
                 label="Sube un archivo (opcional)",
                 file_types=["text", "pdf", "docx", "csv", "xlsx", "image"],
-                type="filepath"
+                type="filepath",
+                visible=True
             )
             
             examples = gr.Examples(
@@ -234,7 +246,7 @@ with gr.Blocks(title="🤖Chucho Bot - CopyXpert Sales Assistant") as demo:
         queue=False
     )
     
-    clear.click(lambda: None, None, chatbot, queue=False)
+    clear.click(lambda: [], None, chatbot, queue=False)
     msg.submit(
         chat,
         inputs=[msg, chatbot, file_upload],