Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import io
 import random
 import pandas as pd
 import chainlit as cl
@@ -15,15 +14,10 @@ EXCEL_PATH = "MasterData SISO - Actualizada - Jan-29-2025.xlsx"
 AUDIO_PATH = "Manual del Operador de Tienda.wav"
 CERTIFICATE_TEMPLATE = "certificate_template_.pdf"
 
-# Convertimos la ruta del audio a un path tipo URL (opcional)
+# Convertimos la ruta del audio a un path tipo URL (opcional, útil localmente)
 abs_path = os.path.abspath(AUDIO_PATH).replace("\\", "/")
 AUDIO_URL = f"file:///{quote(abs_path)}"
 
-# Carpeta de salida para “generar” el PDF
-OUTPUT_FOLDER = "Certificates"
-if not os.path.exists(OUTPUT_FOLDER):
-    os.makedirs(OUTPUT_FOLDER)
-
 # ===== PREGUNTAS DEL EXAMEN =====
 PREGUNTAS = [
     {
@@ -93,7 +87,7 @@ def cargar_datos():
             ],
             dtype={'SAPId': 'int32'}
         )
-        # Creamos columnas extra si no existen
+        # Si faltan columnas, las agregamos
         for col in ['Fecha', 'Calificacion', 'Curso']:
             if col not in df.columns:
                 df[col] = None
@@ -102,7 +96,7 @@ def cargar_datos():
         print(f"Error cargando datos: {str(e)}")
         return {}
 
-def actualizar_excel(sap_id, aprobado: bool):
+def actualizar_excel(sap_id, aprobado):
     """Marca en Excel la fecha, el resultado y el curso para el SAP ID dado."""
     try:
         df = pd.read_excel(EXCEL_PATH)
@@ -114,34 +108,7 @@ def actualizar_excel(sap_id, aprobado: bool):
     except Exception as e:
         print(f"Error actualizando Excel: {str(e)}")
 
-# ===== FUNCIÓN PARA "GENERAR" (COPIAR) EL PDF =====
-def generar_pdf_sin_texto(sap_id: int) -> str:
-    """
-    Retorna la ruta de un PDF “generado” a partir de la plantilla original.
-    Aquí NO dibujamos ni agregamos texto. Simplemente copiamos la plantilla
-    y la renombramos con el ID de la persona.
-    """
-    # Lee la plantilla
-    try:
-        with open(CERTIFICATE_TEMPLATE, "rb") as f_in:
-            pdf_data = f_in.read()
-    except Exception as e:
-        print(f"Error leyendo la plantilla: {e}")
-        return None
-
-    # Crea un nuevo archivo .pdf para este SAP
-    final_name = f"{sap_id}_certificate.pdf"
-    final_path = os.path.join(OUTPUT_FOLDER, final_name)
-    try:
-        with open(final_path, "wb") as f_out:
-            f_out.write(pdf_data)
-    except Exception as e:
-        print(f"Error escribiendo el PDF final: {e}")
-        return None
-
-    return final_path
-
-# ===== CHAINLIT: LÓGICA DE CONVERSACIÓN =====
+# ===== CHAINLIT =====
 @cl.on_chat_start
 async def inicio():
     SAP_DICT = cargar_datos()
@@ -153,12 +120,12 @@ async def manejar_mensaje(message: cl.Message):
     SAP_DICT = cl.user_session.get("SAP_DICT")
     user_data = cl.user_session.get("user_data")
     
-    # Si ya estamos en el examen, procesamos la respuesta
+    # Si el usuario ya está en examen, verificamos la respuesta
     if user_data and user_data.get("en_examen"):
         await procesar_respuesta(message)
         return
         
-    # Si no, interpretamos el mensaje como SAP ID
+    # Sino, interpretamos como SAP ID
     try:
         sap_id = int(message.content)
         if sap_id not in SAP_DICT:
@@ -173,7 +140,7 @@ async def manejar_mensaje(message: cl.Message):
         └ Centro Coste: {usuario['Centro de coste']}
         """
         
-        # Guardamos en sesión
+        # Guardamos datos en la sesión
         cl.user_session.set("user_data", {
             "sap_id": sap_id,
             "puntaje": 0,
@@ -191,13 +158,13 @@ async def mostrar_pregunta():
     user_data = cl.user_session.get("user_data")
     num_pregunta = user_data["pregunta_actual"]
     
-    # Si ya no hay más preguntas, mostramos resultado
+    # Si ya no hay más preguntas, mostramos resultado final
     if num_pregunta >= len(PREGUNTAS):
         await mostrar_resultado()
         return
     
-    pregunta = PREGUNTAS[num_pregunta]
-    mensaje = f"📝 **Pregunta {num_pregunta + 1}**\n{pregunta['pregunta']}\n\n" + "\n".join(pregunta['opciones'])
+    p = PREGUNTAS[num_pregunta]
+    mensaje = f"📝 **Pregunta {num_pregunta + 1}**\n{p['pregunta']}\n\n" + "\n".join(p['opciones'])
     await cl.Message(content=mensaje + "\n\n🔘 Respuesta (A/B/C/D):").send()
 
 async def procesar_respuesta(message: cl.Message):
@@ -206,12 +173,9 @@ async def procesar_respuesta(message: cl.Message):
     
     respuesta = message.content.strip().upper()
     
-    # Validar la respuesta (debe ser A, B, C o D)
+    # Validar la respuesta
     if len(respuesta) != 1 or respuesta not in ['A', 'B', 'C', 'D']:
-        await cl.Message(content="""
-        ⚠️ **Formato incorrecto** ⚠️
-        Por favor ingrese solo la letra correspondiente (A/B/C/D).
-        """).send()
+        await cl.Message(content="⚠️ **Formato incorrecto**. Respuesta = A/B/C/D").send()
         return
     
     correcta = PREGUNTAS[num_pregunta]['respuesta']
@@ -224,7 +188,7 @@ async def procesar_respuesta(message: cl.Message):
     user_data["pregunta_actual"] += 1
     cl.user_session.set("user_data", user_data)
     
-    # Si quedan preguntas, mostramos la siguiente; si no, resultado final
+    # Si quedan preguntas, mostramos la siguiente
     if user_data["pregunta_actual"] < len(PREGUNTAS):
         await mostrar_pregunta()
     else:
@@ -233,8 +197,8 @@ async def procesar_respuesta(message: cl.Message):
 async def mostrar_resultado():
     user_data = cl.user_session.get("user_data")
     SAP_DICT = cl.user_session.get("SAP_DICT")
-
-    # Nota de aprobación => 3 correctas (de 5)
+    
+    # Nota de aprobación (3/5)
     aprobado = user_data['puntaje'] >= 3
     actualizar_excel(user_data['sap_id'], aprobado)
     
@@ -244,23 +208,18 @@ async def mostrar_resultado():
         "❌ **LO SENTIMOS, REPROBADO**\n"
     )
     mensaje += f"Puntaje Final: {user_data['puntaje']}/5"
-
+    
     if aprobado:
-        # Generar PDF sin texto adicional
-        pdf_path = generar_pdf_sin_texto(user_data['sap_id'])
-        if pdf_path:
-            # Lo enviamos como archivo descargable
-            await cl.Message(
-                content=mensaje,
-                elements=[
-                    cl.File(path=pdf_path, name="Certificado de Aprobación")
-                ]
-            ).send()
-        else:
-            # En caso de error leyendo/escribiendo PDF
-            await cl.Message(content=mensaje + "\n\n(⚠️ Error al generar el PDF)").send()
+        # Solo mostramos la plantilla PDF tal cual
+        await cl.Message(
+            content=mensaje,
+            elements=[
+                # Se envía la plantilla como archivo descargable
+                cl.File(path=CERTIFICATE_TEMPLATE, name="Certificado de Aprobación")
+            ]
+        ).send()
     else:
-        # Si reprueba => Mostrar audio
+        # Si no aprueba, muestra el audio
         await cl.Message(
             content=mensaje,
             elements=[
@@ -272,8 +231,8 @@ async def mostrar_resultado():
                 )
             ]
         ).send()
-
-    # Reiniciamos la sesión
+    
+    # Reiniciamos para un nuevo examen
     cl.user_session.set("user_data", None)
     await cl.Message(content="\nIngrese nuevo SAP ID para otro examen:").send()