Update app.py

app.py

@@ -3,76 +3,78 @@ import os
 import json
 import requests
 from datetime import datetime
-
-# --- Importar la librería de Supabase ---
+import uuid # Necesario para generar IDs únicos para cada conversación
 from supabase import create_client, Client
-# --- FIN de las importaciones de Supabase ---
 
+# --- CONFIGURACIÓN ---
+# Clave de la API de Groq
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
+
+# Configuración del modelo y el sistema
 SYSTEM_MESSAGE = os.environ.get("System_Prompt")
-MODEL_NAME = "meta-llama/llama-4-maverick-17b-128e-instruct"
+MODEL_NAME = "llama3-70b-8192" # Modelo actualizado y recomendado
 MAX_TOKENS = 4096
 TEMPERATURE = 0.7
 TOP_P = 0.95
 
-# --- Configuración de Supabase ---
-# Es CRÍTICO que estas variables de entorno se configuren en tu Hugging Face Space
-# Ve a Settings -> Repository secrets y añade SUPABASE_URL y SUPABASE_KEY
+# --- NUEVA CONFIGURACIÓN DE SUPERBASE ---
 SUPABASE_URL = os.environ.get("SUPABASE_URL")
 SUPABASE_KEY = os.environ.get("SUPABASE_KEY")
+TABLE_NAME = "conversations" # Asegúrate que este sea el nombre de tu tabla
 
-supabase: Client = None
+# Inicializar el cliente de Supabase
+# Se verifica que las credenciales existan antes de crear el cliente
 if SUPABASE_URL and SUPABASE_KEY:
-    try:
-        supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
-        print("Supabase cliente inicializado correctamente.")
-    except Exception as e:
-        print(f"Error al inicializar Supabase cliente: {e}")
+    supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
 else:
-    print("Advertencia: SUPABASE_URL o SUPABASE_KEY no están configuradas.")
-
-# --- FIN de la configuración de Supabase ---
-
-def persist_data(session_data, user_id="anonymous_user"):
+    supabase = None
+    print("Advertencia: Las credenciales de Supabase no están configuradas. El historial de chat no se guardará.")
+
+# --- FUNCIÓN DE PERSISTENCIA MODIFICADA PARA SUPABASE ---
+def persist_conversation(conversation_id: uuid.UUID, user_id: str, messages_data: list):
+    """
+    Guarda o actualiza una conversación completa en la tabla de Supabase.
+    Usa 'upsert' para insertar si es nueva o actualizar si ya existe.
+    """
     if not supabase:
-        print("Advertencia: El cliente de Supabase no está inicializado. No se guardará la conversación.")
-        return
+        return # No hacer nada si el cliente no se inicializó
 
     try:
-        # Prepara los datos de la conversación como una lista de diccionarios
-        conversation_log = []
-        for user_msg, assistant_msg in session_data:
-            conversation_log.append({"role": "user", "content": user_msg})
-            if assistant_msg:
-                conversation_log.append({"role": "assistant", "content": assistant_msg})
-
-        # Datos a insertar en la tabla 'conversations'
-        data_to_insert = {
-            'user_id': user_id,
-            'messages': conversation_log # El campo 'timestamp' y 'id' se generarán automáticamente por la BD
-        }
-
-        # Realiza la inserción en Supabase
-        # .insert() devuelve una tupla, el primer elemento es el resultado y el segundo es el count de registros afectados
-        response = supabase.table('conversations').insert(data_to_insert).execute()
-
-        # Supabase devuelve el resultado de la inserción en el atributo 'data' del objeto response.
-        # Verifica si la inserción fue exitosa
-        if response.data:
-            print(f"Conversación guardada en Supabase. ID de sesión: {response.data[0].get('id')}")
+        # El método 'upsert' es ideal: inserta una nueva fila si el 'id' no existe,
+        # o la actualiza si ya existe. Esto es perfecto para guardar el estado de
+        # una conversación en curso.
+        response = supabase.table(TABLE_NAME).upsert({
+            "id": str(conversation_id),
+            "timestamp": datetime.now().isoformat(), # Usamos formato ISO 8601 que Supabase entiende
+            "user_id": str(user_id), # Basado en tu columna 'user_id'
+            "messages": messages_data # Guardamos la lista de mensajes como JSONB
+        }).execute()
+        
+        # Opcional: Verificar si hubo un error en la respuesta de Supabase
+        if hasattr(response, 'error') and response.error:
+             print(f"Error al guardar en Supabase: {response.error}")
         else:
-            print(f"Error al guardar en Supabase: No se recibieron datos en la respuesta. {response.status_code} {response.data}")
+             print(f"Conversación {conversation_id} guardada/actualizada en Supabase.")
 
     except Exception as e:
-        print(f"Error al persistir los datos en Supabase: {e}")
-
-def respond(message, history: list[tuple[str, str]]):
-    # Puedes generar un user_id más sofisticado si lo necesitas,
-    # por ejemplo, usando la IP del usuario o un ID de sesión más único.
-    # Por ahora, mantendremos "anonymous_user" o puedes usar algo como "user_" + un timestamp
-    current_user_id = "anonymous_user"
+        print(f"Ocurrió una excepción al intentar guardar en Supabase: {e}")
+
+# --- FUNCIÓN PRINCIPAL DEL CHATBOT (MODIFICADA) ---
+def respond(message: str, history: list[tuple[str, str]], conversation_id: uuid.UUID):
+    """
+    Función principal que procesa la entrada del usuario, llama a la API de Groq y guarda la conversación.
+    """
+    # Si 'conversation_id' es None, significa que es el primer mensaje de una nueva conversación.
+    if conversation_id is None:
+        conversation_id = uuid.uuid4()
+        # Para la columna 'user_id', podemos usar el mismo ID de la conversación
+        # si no tienes un sistema de usuarios más complejo.
+        user_id = conversation_id
+    else:
+        user_id = conversation_id # Mantener el mismo user_id
 
+    # Construir el payload para la API de Groq
     messages = [{"role": "system", "content": SYSTEM_MESSAGE}]
     for user_msg, assistant_msg in history:
         if user_msg:
@@ -94,13 +96,16 @@ def respond(message, history: list[tuple[str, str]]):
         "stream": True
     }
 
-    response = requests.post(
-        GROQ_API_URL,
-        headers=headers,
-        json=payload,
-        stream=True
-    )
+    # Llamada a la API de Groq
+    try:
+        response = requests.post(GROQ_API_URL, headers=headers, json=payload, stream=True, timeout=10)
+        response.raise_for_status() # Lanza un error si la respuesta es 4xx o 5xx
+    except requests.exceptions.RequestException as e:
+        print(f"Error en la petición a la API de Groq: {e}")
+        yield "Lo siento, no pude conectarme con el servicio en este momento. Inténtalo de nuevo más tarde."
+        return
 
+    # Procesar la respuesta en streaming
     accumulated_response = ""
     for line in response.iter_lines():
         if line:
@@ -119,15 +124,27 @@ def respond(message, history: list[tuple[str, str]]):
                             yield accumulated_response
                 except json.JSONDecodeError:
                     continue
-
-    if not accumulated_response:
-        yield "Lo siento, ocurrió un error al procesar tu solicitud."
+    
+    # Después de obtener la respuesta completa, la guardamos en Supabase
+    if accumulated_response:
+        # Formateamos el historial completo para la columna 'messages' (JSONB)
+        # Esto es mucho más útil que un archivo de texto plano.
+        full_history_for_db = messages + [{"role": "assistant", "content": accumulated_response}]
+        
+        persist_conversation(conversation_id, user_id, full_history_for_db)
     else:
-        current_session = history + [(message, accumulated_response)]
-        persist_data(current_session, user_id=current_user_id)
+        yield "Lo siento, ocurrió un error al procesar tu solicitud."
+
+# --- INTERFAZ DE GRADIO (MODIFICADA) ---
+# Usamos gr.State para mantener el ID de la conversación a través de múltiples turnos
+# en la misma sesión del navegador.
+conversation_id_state = gr.State(value=None)
 
 demo = gr.ChatInterface(
     respond,
+    additional_inputs=[conversation_id_state], # Pasamos el estado como una entrada adicional
+    title="Tu Aliado Momentum",
+    description="Chatea conmigo para saber más sobre el programa Momentum.",
     examples=[["¡Bienvenido a Tu Aliado Momentum!"],
               ["¿En qué consiste el programa y para quién es?"],
               ["¿Qué beneficios obtengo y con qué empresas me conecto?"],