Update app.py

app.py

@@ -1,11 +1,9 @@
 import gradio as gr
-import csv
+import pandas as pd
 import re
 from datetime import datetime
 from hashlib import sha256
 import os
-from PIL import Image
-import numpy as np
 from huggingface_hub import HfApi, hf_hub_download
 
 # Configuración
@@ -23,17 +21,16 @@ PRESTACIONES = [
     "7. OTROS"
 ]
 
-# --- Nuevas funciones para persistencia ---
+# Funciones para CSV
 def cargar_csv():
     try:
         return pd.read_csv(hf_hub_download(
             repo_id=REPO_ID,
             filename=CSV_FILE,
             repo_type="space",
-            token=os.getenv("HF_TOKEN")
+            token=os.getenv("IMSS")
         ))
     except:
-        # Crear CSV inicial si no existe
         df = pd.DataFrame(columns=[
             "Fecha_Registro", "Entidad", "Nombre", 
             "CURP", "Prestacion", "Quincena"
@@ -43,7 +40,7 @@ def cargar_csv():
 
 def guardar_csv(df):
     df.to_csv(CSV_FILE, index=False)
-    api = HfApi(token=os.getenv("HF_TOKEN"))
+    api = HfApi(token=os.getenv("IMSS"))
     api.upload_file(
         path_or_fileobj=CSV_FILE,
         path_in_repo=CSV_FILE,
@@ -51,7 +48,6 @@ def guardar_csv(df):
         repo_type="space"
     )
 
-# --- Modificación de tu función original ---
 def guardar_registro(entidad, nombre, curp, prestacion, quincena):
     if not all([entidad, nombre, curp, prestacion, quincena]):
         return "⚠️ Todos los campos son obligatorios"
@@ -75,11 +71,72 @@ def guardar_registro(entidad, nombre, curp, prestacion, quincena):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-# --- Configuración de Gradio (similar a tu código original) ---
-with gr.Blocks() as app:
-    # ... (tu interfaz existente) ...
+def verificar_credenciales(username, password):
+    return username == "admin" and sha256(password.encode()).hexdigest() == PASSWORD_HASH
+
+# Interfaz
+with gr.Blocks(title="Sistema de Captura") as app:
+    logged_in = gr.State(False)
+
+    # Login
+with gr.Column(visible=True) as login_col:
+    try:
+        img = Image.open("d5bb4c4a-f9e5-4c70-a53c-23b7dc3872a1.jpg")
+        img.thumbnail((200, 200))
+        gr.Image(np.array(img), show_label=False)
+    except:
+        gr.Markdown("![Logo](https://via.placeholder.com/200x100?text=Logo+Sistema)")
+    
+    gr.Markdown("# Sistema de Captura de Incidencias")
+        username = gr.Textbox(label="Usuario")
+        password = gr.Textbox(label="Contraseña", type="password")
+        login_btn = gr.Button("Ingresar")
+        login_msg = gr.Textbox(label="Mensaje", interactive=False)
+
+    # Captura
+    with gr.Column(visible=False) as captura_col:
+        gr.Markdown("## Captura de Datos")
+        entidad = gr.Textbox(label="Entidad Federativa")
+        nombre = gr.Textbox(label="Nombre completo")
+        curp = gr.Textbox(label="CURP")
+        prestacion = gr.Dropdown(PRESTACIONES, label="Prestación afectada")
+        quincena = gr.Textbox(label="Quincena (AAAA-MM-DD)")
+        submit_btn = gr.Button("Guardar")
+        resultado = gr.Textbox(label="Resultado")
+        logout_btn = gr.Button("Cerrar sesión")
+
+    # Eventos (Dentro del bloque with gr.Blocks())
+    def autenticar(user, pwd):
+        if verificar_credenciales(user, pwd):
+            return {
+                login_col: gr.update(visible=False),
+                captura_col: gr.update(visible=True),
+                login_msg: ""
+            }
+        return {
+            login_msg: "⚠️ Credenciales incorrectas"
+        }
+
+    def cerrar_sesion():
+        return {
+            login_col: gr.update(visible=True),
+            captura_col: gr.update(visible=False),
+            login_msg: ""
+        }
+
+    login_btn.click(
+        autenticar,
+        inputs=[username, password],
+        outputs=[login_col, captura_col, login_msg]
+    )
+
+    logout_btn.click(
+        cerrar_sesion,
+        outputs=[login_col, captura_col, login_msg]
+    )
+
     submit_btn.click(
-        guardar_registro, 
+        guardar_registro,
         inputs=[entidad, nombre, curp, prestacion, quincena],
         outputs=resultado
     )