Update app.py

app.py

@@ -148,19 +148,101 @@ def chat(message, history):
     except Exception as e:
         return f"Error: {e}"
 
-demo = gr.ChatInterface(
-    fn=chat,
-    examples=[
-        "¿Qué incluye el curso CopyXpert?",
-        "¿Cuál es el precio del curso?",
-        "¿Cómo puedo inscribirme?",
-        "¿Qué beneficios obtendré?",
-        "¿Cuál es la metodología del curso?",
-        "¿Necesito experiencia previa?"
-    ],
-    title="🤖Chucho Bot - CopyXpert Sales Assistant",
-    description="¡Hola! Soy Chucho Bot, tu asistente personal para el curso CopyXpert. ¿Cómo puedo ayudarte hoy?"
-)
+def process_file(file):
+    file_path = file.name
+    file_extension = os.path.splitext(file_path)[1].lower()
+    
+    # Leer el contenido del archivo según su extensión
+    if file_extension in ['.txt', '.md', '.csv', '.html', '.css', '.js', '.py', '.json']:
+        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)"
+    
+    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
+        combined_message = message
+        if file_content:
+            combined_message += f"\n\nContenido del archivo:\n{file_content}"
+        
+        messages = [
+            {"role": "user", "parts": [system_prompt]},
+            *[{"role": "user", "parts": [msg[0]]} for msg in history],
+            {"role": "user", "parts": [combined_message]}
+        ]
+        response = model.generate_content(messages)
+        return response.text
+    except Exception as e:
+        return f"Error: {e}"
+
+with gr.Blocks(title="🤖Chucho Bot - CopyXpert Sales Assistant") as demo:
+    gr.Markdown("# 🤖Chucho Bot - CopyXpert Sales Assistant")
+    gr.Markdown("¡Hola! Soy Chucho Bot, tu asistente personal para el curso CopyXpert. ¿Cómo puedo ayudarte hoy?")
+    
+    with gr.Row():
+        with gr.Column(scale=4):
+            chatbot = gr.Chatbot()
+            msg = gr.Textbox(
+                placeholder="Escribe tu pregunta aquí...",
+                container=False,
+                scale=7
+            )
+            
+            with gr.Row():
+                submit = gr.Button("Enviar", variant="primary")
+                clear = gr.Button("Limpiar")
+                
+        with gr.Column(scale=1):
+            file_upload = gr.File(
+                label="Sube un archivo (opcional)",
+                file_types=["text", "pdf", "docx", "csv", "xlsx", "image"],
+                type="file"
+            )
+            
+            examples = gr.Examples(
+                examples=[
+                    "¿Qué incluye el curso CopyXpert?",
+                    "¿Cuál es el precio del curso?",
+                    "¿Cómo puedo inscribirme?",
+                    "¿Qué beneficios obtendré?",
+                    "¿Cuál es la metodología del curso?",
+                    "¿Necesito experiencia previa?"
+                ],
+                inputs=msg
+            )
+    
+    submit.click(
+        chat,
+        inputs=[msg, chatbot, file_upload],
+        outputs=chatbot,
+        queue=True
+    ).then(
+        lambda: "",
+        None,
+        msg,
+        queue=False
+    )
+    
+    clear.click(lambda: None, None, chatbot, queue=False)
+    msg.submit(
+        chat,
+        inputs=[msg, chatbot, file_upload],
+        outputs=chatbot,
+        queue=True
+    ).then(
+        lambda: "",
+        None,
+        msg,
+        queue=False
+    )
 
 demo.launch()