Upload 2 files

database.py

@@ -1,531 +1,530 @@
-import sqlite3
-import pandas as pd
-from datetime import datetime, timedelta
-import os
-import bcrypt
-import crypto
-
-# Persistencia: usar /data si existe (Hugging Face Persistent Storage)
-_hf_data = "/data"
-if os.path.isdir(_hf_data):
-    DB_PATH = os.path.join(_hf_data, "proyelec_crm.db")
-else:
-    DB_PATH = os.getenv("DB_PATH", "proyelec_crm.db")
-
-MAX_INTENTOS   = 5        # Intentos fallidos antes de bloquear
-TIEMPO_BLOQUEO = 15      # Minutos de bloqueo
-
-def get_connection(db_path=DB_PATH):
-    conn = sqlite3.connect(db_path, check_same_thread=False)
-    if db_path != ":memory:":
-        conn.execute('PRAGMA journal_mode=WAL;')
-    return conn
-
-def init_db():
-    conn = get_connection()
-    c = conn.cursor()
-
-    # Usuarios
-    c.execute('''CREATE TABLE IF NOT EXISTS users (
-        username TEXT PRIMARY KEY,
-        password TEXT,
-        role TEXT,
-        gemini_key TEXT,
-        tavily_key TEXT,
-        email_user TEXT,
-        email_pass_enc TEXT
-    )''')
-
-    # Historial de análisis
-    c.execute('''CREATE TABLE IF NOT EXISTS history (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        username TEXT,
-        licitacion TEXT,
-        fecha TEXT,
-        items INTEGER
-    )''')
-
-    # Bandeja inteligente de correos
-    c.execute('''CREATE TABLE IF NOT EXISTS smart_inbox (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        licitacion TEXT,
-        remitente TEXT,
-        asunto TEXT,
-        fecha TEXT,
-        resumen TEXT,
-        renglones_relacionados TEXT,
-        cuerpo TEXT,
-        borrador_respuesta TEXT
-    )''')
-
-    # Cache de fichas técnicas (ahorra tokens Gemini)
-    c.execute('''CREATE TABLE IF NOT EXISTS fichas_cache (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        username TEXT,
-        licitacion TEXT,
-        codigo_renglon TEXT,
-        datasheet_md TEXT,
-        fecha TEXT,
-        UNIQUE(username, licitacion, codigo_renglon)
-    )''')
-
-    # Tabla de cotizaciones extraídas de correos (para uso futuro)
-    c.execute('''CREATE TABLE IF NOT EXISTS cotizaciones (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        licitacion TEXT,
-        renglon TEXT,
-        proveedor TEXT,
-        precio_unitario REAL,
-        moneda TEXT,
-        tiempo_entrega TEXT,
-        condiciones TEXT,
-        fecha TEXT,
-        email_asunto TEXT
-    )''')
-
-    # === MÚLTIPLES WORKSPACES POR USUARIO ===
-    c.execute('''CREATE TABLE IF NOT EXISTS workspaces (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        username TEXT NOT NULL,
-        licitacion TEXT NOT NULL,
-        data_json TEXT,
-        cg_json TEXT,
-        fecha_guardado TEXT,
-        UNIQUE(username, licitacion)
-    )''')
-
-    # app_state legacy (se mantiene para compatibilidad)
-    c.execute('''CREATE TABLE IF NOT EXISTS app_state (
-        username TEXT PRIMARY KEY,
-        last_licitacion TEXT,
-        last_data TEXT,
-        last_cg TEXT
-    )''')
-
-    # === MONITOR DE LICITACIONES ACP ===
-    c.execute('''CREATE TABLE IF NOT EXISTS seguimiento_licitaciones (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        numero_licitacion TEXT UNIQUE NOT NULL,
-        objeto TEXT,
-        fecha_asignacion TEXT,
-        fecha_envio_oferta TEXT,
-        monto_ofertado REAL,
-        moneda TEXT DEFAULT "USD",
-        estado TEXT DEFAULT "En Preparacion",
-        link_sli TEXT,
-        notas TEXT,
-        responsable TEXT,
-        fecha_registro TEXT
-    )''')
-    c.execute('''CREATE TABLE IF NOT EXISTS seguimiento_historial (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        licitacion_id INTEGER,
-        fecha TEXT,
-        estado_nuevo TEXT,
-        nota TEXT,
-        registrado_por TEXT,
-        FOREIGN KEY(licitacion_id) REFERENCES seguimiento_licitaciones(id)
-    )''')
-
-    # Protección contra fuerza bruta en login
-    c.execute('''CREATE TABLE IF NOT EXISTS login_attempts (
-        username TEXT PRIMARY KEY,
-        intentos INTEGER DEFAULT 0,
-        bloqueado_hasta TEXT
-    )''')
-
-    # --- Migrar workspace legacy a nueva tabla si existe ---
-    c.execute("SELECT username, last_licitacion, last_data, last_cg FROM app_state")
-    legacy_rows = c.fetchall()
-    for row in legacy_rows:
-        uname, lic, data, cg = row
-        if lic and data and cg:
-            c.execute("""INSERT OR IGNORE INTO workspaces (username, licitacion, data_json, cg_json, fecha_guardado)
-                         VALUES (?, ?, ?, ?, ?)""",
-                      (uname, lic, data, cg, datetime.now().strftime("%Y-%m-%d %H:%M:%S")))
-
-    # Usuario admin por defecto (contraseña: admin)
-    c.execute("SELECT * FROM users WHERE username='admin'")
-    if not c.fetchone():
-        salt = bcrypt.gensalt()
-        hashed_pw = bcrypt.hashpw(b"admin", salt).decode('utf-8')
-        c.execute("INSERT INTO users VALUES ('admin', ?, 'Gerencia', '', '', '', '')", (hashed_pw,))
-
-    conn.commit()
-    conn.close()
-
-# =============================================
-# WORKSPACES (MÚLTIPLES POR USUARIO)
-# =============================================
-
-def save_workspace(username, licitacion, data_json, cg_json):
-    """Guarda o actualiza el workspace de una licitación específica."""
-    conn = get_connection()
-    c = conn.cursor()
-    fecha = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    c.execute("""INSERT OR REPLACE INTO workspaces (username, licitacion, data_json, cg_json, fecha_guardado)
-                 VALUES (?, ?, ?, ?, ?)""",
-              (username, licitacion, data_json, cg_json, fecha))
-    conn.commit()
-    conn.close()
-
-def get_all_workspaces(username, all_users=False):
-    """Retorna todos los workspaces. Si all_users=True, retorna de todos los usuarios (para Gerencia)."""
-    conn = get_connection()
-    if all_users:
-        df = pd.read_sql_query(
-            """SELECT username, licitacion, fecha_guardado FROM workspaces
-               ORDER BY fecha_guardado DESC""", conn)
-    else:
-        df = pd.read_sql_query(
-            """SELECT username, licitacion, fecha_guardado FROM workspaces
-               WHERE username=? ORDER BY fecha_guardado DESC""",
-            conn, params=(username,))
-    conn.close()
-    return df
-
-def load_workspace(username, licitacion):
-    """Carga un workspace específico por licitación."""
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("SELECT data_json, cg_json FROM workspaces WHERE username=? AND licitacion=?",
-              (username, licitacion))
-    row = c.fetchone()
-    conn.close()
-    return row  # (data_json, cg_json) o None
-
-def delete_workspace(username, licitacion):
-    """Elimina un workspace guardado."""
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("DELETE FROM workspaces WHERE username=? AND licitacion=?", (username, licitacion))
-    conn.commit()
-    conn.close()
-
-# Legacy — mantener compatibilidad con código existente
-def save_workspace_state(username, licitacion, data_json, cg_json):
-    save_workspace(username, licitacion, data_json, cg_json)
-
-def load_workspace_state(username):
-    """Carga el workspace más reciente del usuario (compatibilidad legacy)."""
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("SELECT data_json, cg_json FROM workspaces WHERE username=? ORDER BY fecha_guardado DESC LIMIT 1",
-              (username,))
-    row = c.fetchone()
-    conn.close()
-    return row
-
-# =============================================
-# GESTIÓN DE USUARIOS (ADMIN)
-# =============================================
-
-def get_all_users():
-    """Retorna todos los usuarios del sistema para el panel de administración."""
-    conn = get_connection()
-    df = pd.read_sql_query(
-        "SELECT username as Usuario, role as Nivel, email_user as Correo, gemini_key as Clave_Gemini, tavily_key as Clave_Tavily FROM users ORDER BY role, Usuario", conn)
-    conn.close()
-    
-    # Enmascarar las llaves visualmente para seguridad (opcional, pero recomendado)
-    df['Clave_Gemini'] = df['Clave_Gemini'].apply(lambda x: f"{x[:12]}...{x[-4:]}" if x and len(x) > 15 else ("Sin configurar" if not x else x))
-    df['Clave_Tavily'] = df['Clave_Tavily'].apply(lambda x: f"{x[:12]}...{x[-4:]}" if x and len(x) > 15 else ("Sin configurar" if not x else x))
-    df['Correo'] = df['Correo'].apply(lambda x: x if x else "Sin configurar")
-    
-    return df
-
-def create_user(username, password_plain, role):
-    """Crea un nuevo usuario. Retorna True si exitoso, False si el username ya existe."""
-    conn = get_connection()
-    c = conn.cursor()
-    salt = bcrypt.gensalt()
-    hashed = bcrypt.hashpw(password_plain.encode(), salt).decode('utf-8')
-    try:
-        c.execute("INSERT INTO users (username, password, role, gemini_key, tavily_key, email_user, email_pass_enc) VALUES (?, ?, ?, '', '', '', '')",
-                  (username, hashed, role))
-        conn.commit()
-        conn.close()
-        return True
-    except sqlite3.IntegrityError:
-        conn.close()
-        return False
-
-def delete_user(username):
-    """Elimina un usuario. No permite eliminar al admin principal."""
-    if username == 'admin':
-        return False
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("DELETE FROM users WHERE username=?", (username,))
-    conn.commit()
-    conn.close()
-    return True
-
-def update_user_role(username, new_role):
-    """Cambia el rol de un usuario."""
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("UPDATE users SET role=? WHERE username=?", (new_role, username))
-    conn.commit()
-    conn.close()
-
-def reset_user_password(username, new_password_plain):
-    """Resetea la contraseña de un usuario."""
-    conn = get_connection()
-    c = conn.cursor()
-    salt = bcrypt.gensalt()
-    hashed = bcrypt.hashpw(new_password_plain.encode(), salt).decode('utf-8')
-    c.execute("UPDATE users SET password=? WHERE username=?", (hashed, username))
-    conn.commit()
-    conn.close()
-
-# =============================================
-# PROTECCIÓN CONTRA FUERZA BRUTA
-# =============================================
-
-def esta_bloqueado(username: str) -> tuple:
-    """Retorna (bloqueado: bool, segundos_restantes: int)."""
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("SELECT intentos, bloqueado_hasta FROM login_attempts WHERE username=?", (username,))
-    row = c.fetchone()
-    conn.close()
-    if not row or not row[1]:
-        return False, 0
-    hasta = datetime.fromisoformat(row[1])
-    restante = (hasta - datetime.now()).total_seconds()
-    if restante > 0:
-        return True, int(restante)
-    return False, 0
-
-def registrar_intento_fallido(username: str):
-    """Incrementa el contador de intentos fallidos; bloquea si supera el máximo."""
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("INSERT OR IGNORE INTO login_attempts (username, intentos) VALUES (?, 0)", (username,))
-    c.execute("UPDATE login_attempts SET intentos = intentos + 1 WHERE username=?", (username,))
-    c.execute("SELECT intentos FROM login_attempts WHERE username=?", (username,))
-    intentos = c.fetchone()[0]
-    if intentos >= MAX_INTENTOS:
-        hasta = (datetime.now() + timedelta(minutes=TIEMPO_BLOQUEO)).isoformat()
-        c.execute("UPDATE login_attempts SET bloqueado_hasta=? WHERE username=?", (hasta, username))
-    conn.commit()
-    conn.close()
-
-def resetear_intentos(username: str):
-    """Limpia los intentos fallidos tras un login exitoso."""
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("DELETE FROM login_attempts WHERE username=?", (username,))
-    conn.commit()
-    conn.close()
-
-# =============================================
-# FUNCIONES DE USUARIOS
-# =============================================
-
-def get_user(username, password_plain):
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("SELECT * FROM users WHERE LOWER(username)=LOWER(?)", (username,))
-    user = c.fetchone()
-    conn.close()
-    if user:
-        stored_hash = user[1]
-        try:
-            if bcrypt.checkpw(password_plain.encode(), stored_hash.encode('utf-8')):
-                # Descifrar gemini_key y tavily_key antes de devolver
-                user = list(user)
-                user[3] = crypto.decrypt_data(user[3]) if user[3] else ""
-                user[4] = crypto.decrypt_data(user[4]) if user[4] else ""
-                return tuple(user)
-        except ValueError:
-            pass
-    return None
-
-def update_user_profile(username, gemini, tavily, email, enc_pass):
-    """Guarda el perfil del usuario. gemini y tavily se cifran aquí antes de guardar."""
-    conn = get_connection()
-    c = conn.cursor()
-    enc_gemini = crypto.encrypt_data(gemini) if gemini else ""
-    enc_tavily = crypto.encrypt_data(tavily) if tavily else ""
-    c.execute("UPDATE users SET gemini_key=?, tavily_key=?, email_user=?, email_pass_enc=? WHERE username=?",
-              (enc_gemini, enc_tavily, email, enc_pass, username))
-    conn.commit()
-    conn.close()
-
-def save_history(username, licitacion, items_count):
-    conn = get_connection()
-    c = conn.cursor()
-    fecha_actual = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    c.execute("INSERT INTO history (username, licitacion, fecha, items) VALUES (?, ?, ?, ?)",
-              (username, licitacion, fecha_actual, items_count))
-    conn.commit()
-    conn.close()
-
-def get_user_history_df(username):
-    conn = get_connection()
-    df = pd.read_sql_query(
-        "SELECT licitacion as 'Nº Licitación', fecha as 'Fecha Proceso', items as 'Renglones' FROM history WHERE username=? ORDER BY id DESC",
-        conn, params=(username,))
-    conn.close()
-    return df
-
-def delete_history_entry(username, licitacion):
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("DELETE FROM history WHERE username=? AND licitacion=?", (username, licitacion))
-    conn.commit()
-    conn.close()
-
-def clear_all_history(username):
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("DELETE FROM history WHERE username=?", (username,))
-    conn.commit()
-    conn.close()
-
-def get_correos_licitacion_df(licitacion):
-    conn = get_connection()
-    try:
-        df = pd.read_sql_query(
-            "SELECT id, remitente, asunto, fecha, resumen, renglones_relacionados, cuerpo, borrador_respuesta FROM smart_inbox WHERE licitacion=? ORDER BY id DESC",
-            conn, params=(licitacion,))
-    except Exception:
-        df = pd.DataFrame()
-    conn.close()
-    return df
-
-def get_user_credentials(username):
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("SELECT email_user, email_pass_enc, gemini_key, tavily_key FROM users WHERE username=?", (username,))
-    row = c.fetchone()
-    conn.close()
-    if row:
-        # Descifrar gemini_key (índice 2) y tavily_key (índice 3) antes de devolver
-        return (row[0], row[1], crypto.decrypt_data(row[2]) if row[2] else "", crypto.decrypt_data(row[3]) if row[3] else "")
-    return None
-
-def check_email_exists(licitacion, asunto, remitente):
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("SELECT id FROM smart_inbox WHERE licitacion=? AND asunto=? AND remitente=?",
-              (licitacion, asunto, remitente))
-    exists = c.fetchone() is not None
-    conn.close()
-    return exists
-
-def insert_smart_inbox(licitacion, remitente, asunto, fecha, resumen, renglones, cuerpo, borrador=""):
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute(
-        "INSERT INTO smart_inbox (licitacion, remitente, asunto, fecha, resumen, renglones_relacionados, cuerpo, borrador_respuesta) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
-        (licitacion, remitente, asunto, fecha, resumen, renglones, cuerpo, borrador))
-    conn.commit()
-    conn.close()
-
-def insert_cotizacion(licitacion, renglon, proveedor, precio_unitario, moneda, tiempo_entrega, condiciones, fecha, email_asunto):
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute(
-        "INSERT INTO cotizaciones (licitacion, renglon, proveedor, precio_unitario, moneda, tiempo_entrega, condiciones, fecha, email_asunto) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)",
-        (licitacion, renglon, proveedor, precio_unitario, moneda, tiempo_entrega, condiciones, fecha, email_asunto))
-    conn.commit()
-    conn.close()
-
-def check_cotizacion_exists(licitacion, renglon, proveedor):
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("SELECT id FROM cotizaciones WHERE licitacion=? AND renglon=? AND proveedor=?",
-              (licitacion, renglon, proveedor))
-    exists = c.fetchone() is not None
-    conn.close()
-    return exists
-
-def get_ficha_cache(username, licitacion, codigo_renglon):
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("SELECT datasheet_md FROM fichas_cache WHERE username=? AND licitacion=? AND codigo_renglon=?",
-              (username, licitacion, codigo_renglon))
-    row = c.fetchone()
-    conn.close()
-    return row[0] if row else None
-
-def save_ficha_cache(username, licitacion, codigo_renglon, datasheet_md):
-    conn = get_connection()
-    c = conn.cursor()
-    fecha = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    c.execute(
-        "INSERT OR REPLACE INTO fichas_cache (username, licitacion, codigo_renglon, datasheet_md, fecha) VALUES (?, ?, ?, ?, ?)",
-        (username, licitacion, codigo_renglon, datasheet_md, fecha))
-    conn.commit()
-    conn.close()
-
-# =============================================
-# MONITOR DE LICITACIONES ACP
-# =============================================
-
-ESTADOS_ACP = [
-    "En Preparacion",
-    "Oferta Enviada al SLI",
-    "Cumple Tecnicamente",
-    "No Cumple Tecnicamente",
-    "En Evaluacion Economica",
-    "Adjudicada",
-    "No Adjudicada",
-    "Desierta",
-]
-
-def crear_seguimiento(numero_licitacion, objeto, fecha_asignacion, fecha_envio_oferta,
-                      monto_ofertado, moneda, link_sli, notas, responsable):
-    conn = get_connection()
-    c = conn.cursor()
-    fecha = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    try:
-        c.execute("""INSERT INTO seguimiento_licitaciones
-            (numero_licitacion, objeto, fecha_asignacion, fecha_envio_oferta,
-             monto_ofertado, moneda, estado, link_sli, notas, responsable, fecha_registro)
-            VALUES (?, ?, ?, ?, ?, ?, 'En Preparacion', ?, ?, ?, ?)""",
-            (numero_licitacion, objeto, fecha_asignacion, fecha_envio_oferta,
-             monto_ofertado, moneda, link_sli, notas, responsable, fecha))
-        conn.commit()
-        conn.close()
-        return True
-    except Exception:
-        conn.close()
-        return False
-
-def get_seguimientos():
-    conn = get_connection()
-    df = pd.read_sql_query(
-        "SELECT * FROM seguimiento_licitaciones ORDER BY fecha_registro DESC", conn)
-    conn.close()
-    return df
-
-def actualizar_estado(licitacion_id, nuevo_estado, nota, registrado_por):
-    conn = get_connection()
-    c = conn.cursor()
-    fecha = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    c.execute("UPDATE seguimiento_licitaciones SET estado=? WHERE id=?", (nuevo_estado, licitacion_id))
-    c.execute("""INSERT INTO seguimiento_historial (licitacion_id, fecha, estado_nuevo, nota, registrado_por)
-                 VALUES (?, ?, ?, ?, ?)""", (licitacion_id, fecha, nuevo_estado, nota, registrado_por))
-    conn.commit()
-    conn.close()
-
-def get_historial_seguimiento(licitacion_id):
-    conn = get_connection()
-    df = pd.read_sql_query(
-        "SELECT fecha, estado_nuevo, nota, registrado_por FROM seguimiento_historial WHERE licitacion_id=? ORDER BY id DESC",
-        conn, params=(licitacion_id,))
-    conn.close()
-    return df
-
-def eliminar_seguimiento(licitacion_id):
-    conn = get_connection()
-    c = conn.cursor()
-    c.execute("DELETE FROM seguimiento_historial WHERE licitacion_id=?", (licitacion_id,))
-    c.execute("DELETE FROM seguimiento_licitaciones WHERE id=?", (licitacion_id,))
-    conn.commit()
-    conn.close()
-
+import psycopg2
+import pandas as pd
+from datetime import datetime, timedelta
+import os
+import bcrypt
+import crypto
+import warnings
+
+# Suprimir advertencias de Pandas al usar psycopg2 directo
+warnings.filterwarnings("ignore", category=UserWarning, module="pandas")
+
+DATABASE_URL = os.getenv("DATABASE_URL")
+
+MAX_INTENTOS   = 5        # Intentos fallidos antes de bloquear
+TIEMPO_BLOQUEO = 15      # Minutos de bloqueo
+
+def get_connection():
+    if not DATABASE_URL:
+        raise ValueError("Falta DATABASE_URL en las variables de entorno")
+    conn = psycopg2.connect(DATABASE_URL)
+    return conn
+
+def init_db():
+    conn = get_connection()
+    c = conn.cursor()
+
+    # Usuarios
+    c.execute('''CREATE TABLE IF NOT EXISTS users (
+        username TEXT PRIMARY KEY,
+        password TEXT,
+        role TEXT,
+        gemini_key TEXT,
+        tavily_key TEXT,
+        email_user TEXT,
+        email_pass_enc TEXT
+    )''')
+
+    # Historial de análisis
+    c.execute('''CREATE TABLE IF NOT EXISTS history (
+        id SERIAL PRIMARY KEY,
+        username TEXT,
+        licitacion TEXT,
+        fecha TEXT,
+        items INTEGER
+    )''')
+
+    # Bandeja inteligente de correos
+    c.execute('''CREATE TABLE IF NOT EXISTS smart_inbox (
+        id SERIAL PRIMARY KEY,
+        licitacion TEXT,
+        remitente TEXT,
+        asunto TEXT,
+        fecha TEXT,
+        resumen TEXT,
+        renglones_relacionados TEXT,
+        cuerpo TEXT,
+        borrador_respuesta TEXT
+    )''')
+
+    # Cache de fichas técnicas (ahorra tokens Gemini)
+    c.execute('''CREATE TABLE IF NOT EXISTS fichas_cache (
+        id SERIAL PRIMARY KEY,
+        username TEXT,
+        licitacion TEXT,
+        codigo_renglon TEXT,
+        datasheet_md TEXT,
+        fecha TEXT,
+        UNIQUE(username, licitacion, codigo_renglon)
+    )''')
+
+    # Tabla de cotizaciones extraídas de correos (para uso futuro)
+    c.execute('''CREATE TABLE IF NOT EXISTS cotizaciones (
+        id SERIAL PRIMARY KEY,
+        licitacion TEXT,
+        renglon TEXT,
+        proveedor TEXT,
+        precio_unitario REAL,
+        moneda TEXT,
+        tiempo_entrega TEXT,
+        condiciones TEXT,
+        fecha TEXT,
+        email_asunto TEXT
+    )''')
+
+    # === MÚLTIPLES WORKSPACES POR USUARIO ===
+    c.execute('''CREATE TABLE IF NOT EXISTS workspaces (
+        id SERIAL PRIMARY KEY,
+        username TEXT NOT NULL,
+        licitacion TEXT NOT NULL,
+        data_json TEXT,
+        cg_json TEXT,
+        fecha_guardado TEXT,
+        UNIQUE(username, licitacion)
+    )''')
+
+    # app_state legacy (se mantiene para compatibilidad)
+    c.execute('''CREATE TABLE IF NOT EXISTS app_state (
+        username TEXT PRIMARY KEY,
+        last_licitacion TEXT,
+        last_data TEXT,
+        last_cg TEXT
+    )''')
+
+    # === MONITOR DE LICITACIONES ACP ===
+    c.execute('''CREATE TABLE IF NOT EXISTS seguimiento_licitaciones (
+        id SERIAL PRIMARY KEY,
+        numero_licitacion TEXT UNIQUE NOT NULL,
+        objeto TEXT,
+        fecha_asignacion TEXT,
+        fecha_envio_oferta TEXT,
+        monto_ofertado REAL,
+        moneda TEXT DEFAULT "USD",
+        estado TEXT DEFAULT "En Preparacion",
+        link_sli TEXT,
+        notas TEXT,
+        responsable TEXT,
+        fecha_registro TEXT
+    )''')
+    c.execute('''CREATE TABLE IF NOT EXISTS seguimiento_historial (
+        id SERIAL PRIMARY KEY,
+        licitacion_id INTEGER,
+        fecha TEXT,
+        estado_nuevo TEXT,
+        nota TEXT,
+        registrado_por TEXT,
+        FOREIGN KEY(licitacion_id) REFERENCES seguimiento_licitaciones(id)
+    )''')
+
+    # Protección contra fuerza bruta en login
+    c.execute('''CREATE TABLE IF NOT EXISTS login_attempts (
+        username TEXT PRIMARY KEY,
+        intentos INTEGER DEFAULT 0,
+        bloqueado_hasta TEXT
+    )''')
+
+    # --- Migrar workspace legacy a nueva tabla si existe ---
+    c.execute("SELECT username, last_licitacion, last_data, last_cg FROM app_state")
+    legacy_rows = c.fetchall()
+    for row in legacy_rows:
+        uname, lic, data, cg = row
+        if lic and data and cg:
+            c.execute("""INSERT INTO workspaces (username, licitacion, data_json, cg_json, fecha_guardado)
+                         VALUES (%s, %s, %s, %s, %s)""",
+                      (uname, lic, data, cg, datetime.now().strftime("%Y-%m-%d %H:%M:%S")))
+
+    # Usuario admin por defecto (contraseña: admin)
+    c.execute("SELECT * FROM users WHERE username='admin'")
+    if not c.fetchone():
+        salt = bcrypt.gensalt()
+        hashed_pw = bcrypt.hashpw(b"admin", salt).decode('utf-8')
+        c.execute("INSERT INTO users VALUES ('admin', ?, 'Gerencia', '', '', '', '')", (hashed_pw,))
+
+    conn.commit()
+    conn.close()
+
+# =============================================
+# WORKSPACES (MÚLTIPLES POR USUARIO)
+# =============================================
+
+def save_workspace(username, licitacion, data_json, cg_json):
+    """Guarda o actualiza el workspace de una licitación específica."""
+    conn = get_connection()
+    c = conn.cursor()
+    fecha = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    c.execute("""INSERT OR REPLACE INTO workspaces (username, licitacion, data_json, cg_json, fecha_guardado)
+                 VALUES (%s, %s, %s, %s, %s)""",
+              (username, licitacion, data_json, cg_json, fecha))
+    conn.commit()
+    conn.close()
+
+def get_all_workspaces(username, all_users=False):
+    """Retorna todos los workspaces. Si all_users=True, retorna de todos los usuarios (para Gerencia)."""
+    conn = get_connection()
+    if all_users:
+        df = pd.read_sql_query(
+            """SELECT username, licitacion, fecha_guardado FROM workspaces
+               ORDER BY fecha_guardado DESC""", conn)
+    else:
+        df = pd.read_sql_query(
+            """SELECT username, licitacion, fecha_guardado FROM workspaces
+               WHERE username=%s ORDER BY fecha_guardado DESC""",
+            conn, params=(username,))
+    conn.close()
+    return df
+
+def load_workspace(username, licitacion):
+    """Carga un workspace específico por licitación."""
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("SELECT data_json, cg_json FROM workspaces WHERE username=%s AND licitacion=%s",
+              (username, licitacion))
+    row = c.fetchone()
+    conn.close()
+    return row  # (data_json, cg_json) o None
+
+def delete_workspace(username, licitacion):
+    """Elimina un workspace guardado."""
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("DELETE FROM workspaces WHERE username=%s AND licitacion=%s", (username, licitacion))
+    conn.commit()
+    conn.close()
+
+# Legacy — mantener compatibilidad con código existente
+def save_workspace_state(username, licitacion, data_json, cg_json):
+    save_workspace(username, licitacion, data_json, cg_json)
+
+def load_workspace_state(username):
+    """Carga el workspace más reciente del usuario (compatibilidad legacy)."""
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("SELECT data_json, cg_json FROM workspaces WHERE username=%s ORDER BY fecha_guardado DESC LIMIT 1",
+              (username,))
+    row = c.fetchone()
+    conn.close()
+    return row
+
+# =============================================
+# GESTIÓN DE USUARIOS (ADMIN)
+# =============================================
+
+def get_all_users():
+    """Retorna todos los usuarios del sistema para el panel de administración."""
+    conn = get_connection()
+    df = pd.read_sql_query(
+        "SELECT username as Usuario, role as Nivel, email_user as Correo, gemini_key as Clave_Gemini, tavily_key as Clave_Tavily FROM users ORDER BY role, Usuario", conn)
+    conn.close()
+    
+    # Enmascarar las llaves visualmente para seguridad (opcional, pero recomendado)
+    df['Clave_Gemini'] = df['Clave_Gemini'].apply(lambda x: f"{x[:12]}...{x[-4:]}" if x and len(x) > 15 else ("Sin configurar" if not x else x))
+    df['Clave_Tavily'] = df['Clave_Tavily'].apply(lambda x: f"{x[:12]}...{x[-4:]}" if x and len(x) > 15 else ("Sin configurar" if not x else x))
+    df['Correo'] = df['Correo'].apply(lambda x: x if x else "Sin configurar")
+    
+    return df
+
+def create_user(username, password_plain, role):
+    """Crea un nuevo usuario. Retorna True si exitoso, False si el username ya existe."""
+    conn = get_connection()
+    c = conn.cursor()
+    salt = bcrypt.gensalt()
+    hashed = bcrypt.hashpw(password_plain.encode(), salt).decode('utf-8')
+    try:
+        c.execute("INSERT INTO users (username, password, role, gemini_key, tavily_key, email_user, email_pass_enc) VALUES (%s, %s, %s, '', '', '', '')",
+                  (username, hashed, role))
+        conn.commit()
+        conn.close()
+        return True
+    except psycopg2.IntegrityError:
+        conn.close()
+        return False
+
+def delete_user(username):
+    """Elimina un usuario. No permite eliminar al admin principal."""
+    if username == 'admin':
+        return False
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("DELETE FROM users WHERE username=%s", (username,))
+    conn.commit()
+    conn.close()
+    return True
+
+def update_user_role(username, new_role):
+    """Cambia el rol de un usuario."""
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("UPDATE users SET role=%s WHERE username=%s", (new_role, username))
+    conn.commit()
+    conn.close()
+
+def reset_user_password(username, new_password_plain):
+    """Resetea la contraseña de un usuario."""
+    conn = get_connection()
+    c = conn.cursor()
+    salt = bcrypt.gensalt()
+    hashed = bcrypt.hashpw(new_password_plain.encode(), salt).decode('utf-8')
+    c.execute("UPDATE users SET password=%s WHERE username=%s", (hashed, username))
+    conn.commit()
+    conn.close()
+
+# =============================================
+# PROTECCIÓN CONTRA FUERZA BRUTA
+# =============================================
+
+def esta_bloqueado(username: str) -> tuple:
+    """Retorna (bloqueado: bool, segundos_restantes: int)."""
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("SELECT intentos, bloqueado_hasta FROM login_attempts WHERE username=%s", (username,))
+    row = c.fetchone()
+    conn.close()
+    if not row or not row[1]:
+        return False, 0
+    hasta = datetime.fromisoformat(row[1])
+    restante = (hasta - datetime.now()).total_seconds()
+    if restante > 0:
+        return True, int(restante)
+    return False, 0
+
+def registrar_intento_fallido(username: str):
+    """Incrementa el contador de intentos fallidos; bloquea si supera el máximo."""
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("INSERT INTO login_attempts (username, intentos) VALUES (%s, 0) ON CONFLICT (username) DO NOTHING", (username,))
+    c.execute("UPDATE login_attempts SET intentos = intentos + 1 WHERE username=%s", (username,))
+    c.execute("SELECT intentos FROM login_attempts WHERE username=%s", (username,))
+    intentos = c.fetchone()[0]
+    if intentos >= MAX_INTENTOS:
+        hasta = (datetime.now() + timedelta(minutes=TIEMPO_BLOQUEO)).isoformat()
+        c.execute("UPDATE login_attempts SET bloqueado_hasta=? WHERE username=%s", (hasta, username))
+    conn.commit()
+    conn.close()
+
+def resetear_intentos(username: str):
+    """Limpia los intentos fallidos tras un login exitoso."""
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("DELETE FROM login_attempts WHERE username=%s", (username,))
+    conn.commit()
+    conn.close()
+
+# =============================================
+# FUNCIONES DE USUARIOS
+# =============================================
+
+def get_user(username, password_plain):
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("SELECT * FROM users WHERE LOWER(username)=LOWER(%s)", (username,))
+    user = c.fetchone()
+    conn.close()
+    if user:
+        stored_hash = user[1]
+        try:
+            if bcrypt.checkpw(password_plain.encode(), stored_hash.encode('utf-8')):
+                # Descifrar gemini_key y tavily_key antes de devolver
+                user = list(user)
+                user[3] = crypto.decrypt_data(user[3]) if user[3] else ""
+                user[4] = crypto.decrypt_data(user[4]) if user[4] else ""
+                return tuple(user)
+        except ValueError:
+            pass
+    return None
+
+def update_user_profile(username, gemini, tavily, email, enc_pass):
+    """Guarda el perfil del usuario. gemini y tavily se cifran aquí antes de guardar."""
+    conn = get_connection()
+    c = conn.cursor()
+    enc_gemini = crypto.encrypt_data(gemini) if gemini else ""
+    enc_tavily = crypto.encrypt_data(tavily) if tavily else ""
+    c.execute("UPDATE users SET gemini_key=?, tavily_key=?, email_user=?, email_pass_enc=? WHERE username=%s",
+              (enc_gemini, enc_tavily, email, enc_pass, username))
+    conn.commit()
+    conn.close()
+
+def save_history(username, licitacion, items_count):
+    conn = get_connection()
+    c = conn.cursor()
+    fecha_actual = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    c.execute("INSERT INTO history (username, licitacion, fecha, items) VALUES (%s, %s, %s, %s)",
+              (username, licitacion, fecha_actual, items_count))
+    conn.commit()
+    conn.close()
+
+def get_user_history_df(username):
+    conn = get_connection()
+    df = pd.read_sql_query(
+        "SELECT licitacion as 'Nº Licitación', fecha as 'Fecha Proceso', items as 'Renglones' FROM history WHERE username=%s ORDER BY id DESC",
+        conn, params=(username,))
+    conn.close()
+    return df
+
+def delete_history_entry(username, licitacion):
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("DELETE FROM history WHERE username=%s AND licitacion=%s", (username, licitacion))
+    conn.commit()
+    conn.close()
+
+def clear_all_history(username):
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("DELETE FROM history WHERE username=%s", (username,))
+    conn.commit()
+    conn.close()
+
+def get_correos_licitacion_df(licitacion):
+    conn = get_connection()
+    try:
+        df = pd.read_sql_query(
+            "SELECT id, remitente, asunto, fecha, resumen, renglones_relacionados, cuerpo, borrador_respuesta FROM smart_inbox WHERE licitacion=%s ORDER BY id DESC",
+            conn, params=(licitacion,))
+    except Exception:
+        df = pd.DataFrame()
+    conn.close()
+    return df
+
+def get_user_credentials(username):
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("SELECT email_user, email_pass_enc, gemini_key, tavily_key FROM users WHERE username=%s", (username,))
+    row = c.fetchone()
+    conn.close()
+    if row:
+        # Descifrar gemini_key (índice 2) y tavily_key (índice 3) antes de devolver
+        return (row[0], row[1], crypto.decrypt_data(row[2]) if row[2] else "", crypto.decrypt_data(row[3]) if row[3] else "")
+    return None
+
+def check_email_exists(licitacion, asunto, remitente):
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("SELECT id FROM smart_inbox WHERE licitacion=%s AND asunto=%s AND remitente=%s",
+              (licitacion, asunto, remitente))
+    exists = c.fetchone() is not None
+    conn.close()
+    return exists
+
+def insert_smart_inbox(licitacion, remitente, asunto, fecha, resumen, renglones, cuerpo, borrador=""):
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute(
+        "INSERT INTO smart_inbox (licitacion, remitente, asunto, fecha, resumen, renglones_relacionados, cuerpo, borrador_respuesta) VALUES (%s, %s, %s, %s, ?, ?, ?, ?)",
+        (licitacion, remitente, asunto, fecha, resumen, renglones, cuerpo, borrador))
+    conn.commit()
+    conn.close()
+
+def insert_cotizacion(licitacion, renglon, proveedor, precio_unitario, moneda, tiempo_entrega, condiciones, fecha, email_asunto):
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute(
+        "INSERT INTO cotizaciones (licitacion, renglon, proveedor, precio_unitario, moneda, tiempo_entrega, condiciones, fecha, email_asunto) VALUES (%s, %s, %s, %s, ?, ?, ?, ?, ?)",
+        (licitacion, renglon, proveedor, precio_unitario, moneda, tiempo_entrega, condiciones, fecha, email_asunto))
+    conn.commit()
+    conn.close()
+
+def check_cotizacion_exists(licitacion, renglon, proveedor):
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("SELECT id FROM cotizaciones WHERE licitacion=%s AND renglon=? AND proveedor=?",
+              (licitacion, renglon, proveedor))
+    exists = c.fetchone() is not None
+    conn.close()
+    return exists
+
+def get_ficha_cache(username, licitacion, codigo_renglon):
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("SELECT datasheet_md FROM fichas_cache WHERE username=%s AND licitacion=%s AND codigo_renglon=%s",
+              (username, licitacion, codigo_renglon))
+    row = c.fetchone()
+    conn.close()
+    return row[0] if row else None
+
+def save_ficha_cache(username, licitacion, codigo_renglon, datasheet_md):
+    conn = get_connection()
+    c = conn.cursor()
+    fecha = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    c.execute(
+        "INSERT OR REPLACE INTO fichas_cache (username, licitacion, codigo_renglon, datasheet_md, fecha) VALUES (%s, %s, %s, %s, %s)",
+        (username, licitacion, codigo_renglon, datasheet_md, fecha))
+    conn.commit()
+    conn.close()
+
+# =============================================
+# MONITOR DE LICITACIONES ACP
+# =============================================
+
+ESTADOS_ACP = [
+    "En Preparacion",
+    "Oferta Enviada al SLI",
+    "Cumple Tecnicamente",
+    "No Cumple Tecnicamente",
+    "En Evaluacion Economica",
+    "Adjudicada",
+    "No Adjudicada",
+    "Desierta",
+]
+
+def crear_seguimiento(numero_licitacion, objeto, fecha_asignacion, fecha_envio_oferta,
+                      monto_ofertado, moneda, link_sli, notas, responsable):
+    conn = get_connection()
+    c = conn.cursor()
+    fecha = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    try:
+        c.execute("""INSERT INTO seguimiento_licitaciones
+            (numero_licitacion, objeto, fecha_asignacion, fecha_envio_oferta,
+             monto_ofertado, moneda, estado, link_sli, notas, responsable, fecha_registro)
+            VALUES (%s, %s, %s, %s, ?, ?, 'En Preparacion', ?, ?, ?, ?)""",
+            (numero_licitacion, objeto, fecha_asignacion, fecha_envio_oferta,
+             monto_ofertado, moneda, link_sli, notas, responsable, fecha))
+        conn.commit()
+        conn.close()
+        return True
+    except Exception:
+        conn.close()
+        return False
+
+def get_seguimientos():
+    conn = get_connection()
+    df = pd.read_sql_query(
+        "SELECT * FROM seguimiento_licitaciones ORDER BY fecha_registro DESC", conn)
+    conn.close()
+    return df
+
+def actualizar_estado(licitacion_id, nuevo_estado, nota, registrado_por):
+    conn = get_connection()
+    c = conn.cursor()
+    fecha = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    c.execute("UPDATE seguimiento_licitaciones SET estado=? WHERE id=?", (nuevo_estado, licitacion_id))
+    c.execute("""INSERT INTO seguimiento_historial (licitacion_id, fecha, estado_nuevo, nota, registrado_por)
+                 VALUES (%s, %s, %s, %s, %s)""", (licitacion_id, fecha, nuevo_estado, nota, registrado_por))
+    conn.commit()
+    conn.close()
+
+def get_historial_seguimiento(licitacion_id):
+    conn = get_connection()
+    df = pd.read_sql_query(
+        "SELECT fecha, estado_nuevo, nota, registrado_por FROM seguimiento_historial WHERE licitacion_id=? ORDER BY id DESC",
+        conn, params=(licitacion_id,))
+    conn.close()
+    return df
+
+def eliminar_seguimiento(licitacion_id):
+    conn = get_connection()
+    c = conn.cursor()
+    c.execute("DELETE FROM seguimiento_historial WHERE licitacion_id=?", (licitacion_id,))
+    c.execute("DELETE FROM seguimiento_licitaciones WHERE id=?", (licitacion_id,))
+    conn.commit()
+    conn.close()
+

requirements.txt

@@ -1,16 +1,17 @@
-streamlit
-pandas
-plotly
-tavily-python
-google-generativeai
-cryptography
-requests
-fastapi
-uvicorn
-python-multipart
-python-dotenv
-openpyxl
-playwright
+streamlit
+pandas
+plotly
+tavily-python
+google-generativeai
+cryptography
+requests
+fastapi
+uvicorn
+python-multipart
+python-dotenv
+openpyxl
+playwright
 beautifulsoup4
 pypdf
-bcrypt
+bcrypt
+psycopg2-binary