Update app.py

app.py

@@ -6,12 +6,18 @@ import pandas as pd
 import re
 from datetime import datetime
 from huggingface_hub import InferenceClient
-from google.oauth2.credentials import Credentials
-from google_auth_oauthlib.flow import InstalledAppFlow
-from google.auth.transport.requests import Request
-from googleapiclient.discovery import build
-from googleapiclient.http import MediaFileUpload
-import pickle
+
+# Importar Google Drive solo si está disponible
+try:
+    from google.oauth2.credentials import Credentials
+    from google_auth_oauthlib.flow import InstalledAppFlow
+    from google.auth.transport.requests import Request
+    from googleapiclient.discovery import build
+    from googleapiclient.http import MediaFileUpload
+    import pickle
+    DRIVE_DISPONIBLE = True
+except ImportError:
+    DRIVE_DISPONIBLE = False
 
 # ============= EXTRAER TEXTO DEL PDF =============
 def extraer_texto_pdf(pdf_file):
@@ -28,7 +34,7 @@ def extraer_texto_pdf(pdf_file):
 def analizar_y_convertir_json(texto):
     """El LLM lee la factura, decide cómo estructurarla y devuelve JSON"""
     
-    token = os.getenv("aa")
+    token = os.getenv("HF_TOKEN")
     if not token:
         return None, None, "Error: Falta configurar HF_TOKEN en Settings → Secrets"
     
@@ -234,10 +240,78 @@ def json_a_csv(datos_json):
     
     return pd.DataFrame(filas)
 
-# ============= SUBIR A GOOGLE DRIVE =============
+# ============= AUTENTICAR GOOGLE DRIVE =============
+def autenticar_drive():
+    """Inicia el proceso de autenticación con Google Drive"""
+    
+    if not DRIVE_DISPONIBLE:
+        return "Error: Librerías de Google Drive no instaladas.\n\nAgrega al requirements.txt:\ngoogle-auth-oauthlib\ngoogle-auth-httplib2\ngoogle-api-python-client", False
+    
+    SCOPES = ['https://www.googleapis.com/auth/drive.file']
+    
+    try:
+        # Verificar si existe credentials.json
+        if not os.path.exists('credentials.json'):
+            return "Error: Falta el archivo credentials.json.\n\nPasos:\n1. Ve a https://console.cloud.google.com/\n2. Crea un proyecto\n3. Activa Google Drive API\n4. Crea credenciales OAuth 2.0\n5. Descarga credentials.json\n6. Súbelo a tu aplicación", False
+        
+        # Verificar si ya hay una sesión activa
+        if os.path.exists('token.pickle'):
+            with open('token.pickle', 'rb') as token:
+                creds = pickle.load(token)
+                if creds and creds.valid:
+                    return "Ya estás conectado a Google Drive", True
+        
+        # Iniciar flujo de autenticación
+        flow = InstalledAppFlow.from_client_secrets_file('credentials.json', SCOPES)
+        creds = flow.run_local_server(port=8080)
+        
+        # Guardar credenciales
+        with open('token.pickle', 'wb') as token:
+            pickle.dump(creds, token)
+        
+        return "Autenticación exitosa! Ahora puedes guardar archivos en Google Drive", True
+    
+    except Exception as e:
+        return f"Error en la autenticación: {str(e)}", False
+
+# ============= VERIFICAR ESTADO DE DRIVE =============
+def verificar_sesion_drive():
+    """Verifica si hay una sesión activa de Google Drive"""
+    
+    if not DRIVE_DISPONIBLE:
+        return "Librerías no instaladas", False
+    
+    if not os.path.exists('token.pickle'):
+        return "No conectado", False
+    
+    try:
+        with open('token.pickle', 'rb') as token:
+            creds = pickle.load(token)
+            if creds and creds.valid:
+                return "Conectado a Google Drive", True
+            else:
+                return "Sesión expirada", False
+    except:
+        return "Error al verificar sesión", False
+
+# ============= CERRAR SESIÓN DE DRIVE =============
+def cerrar_sesion_drive():
+    """Cierra la sesión de Google Drive"""
+    
+    try:
+        if os.path.exists('token.pickle'):
+            os.remove('token.pickle')
+            return "Sesión cerrada correctamente", False
+        else:
+            return "No había sesión activa", False
+    except Exception as e:
+        return f"Error al cerrar sesión: {str(e)}", False
 def subir_a_drive(archivo_csv):
     """Sube el archivo CSV a Google Drive"""
     
+    if not DRIVE_DISPONIBLE:
+        return "Error: Librerías de Google Drive no instaladas.\n\nAgrega al requirements.txt:\ngoogle-auth-oauthlib\ngoogle-auth-httplib2\ngoogle-api-python-client"
+    
     SCOPES = ['https://www.googleapis.com/auth/drive.file']
     creds = None
     
@@ -254,7 +328,7 @@ def subir_a_drive(archivo_csv):
             else:
                 # Verificar si existe credentials.json
                 if not os.path.exists('credentials.json'):
-                    return "Error: Falta el archivo credentials.json. Descárgalo desde Google Cloud Console."
+                    return "Error: Falta el archivo credentials.json.\n\nDescárgalo desde Google Cloud Console:\nhttps://console.cloud.google.com/"
                 
                 flow = InstalledAppFlow.from_client_secrets_file('credentials.json', SCOPES)
                 creds = flow.run_local_server(port=0)
@@ -382,15 +456,44 @@ with gr.Blocks(title="Extractor IA de Facturas") as demo:
                 type="filepath"
             )
             
+            gr.Markdown("")
+            gr.Markdown("---")
+            gr.Markdown("")
+            
+            # Sección de Google Drive
+            gr.Markdown("### Google Drive")
+            
+            with gr.Row():
+                btn_conectar_drive = gr.Button(
+                    "Conectar con Google Drive",
+                    size="sm",
+                    variant="secondary"
+                )
+                btn_cerrar_sesion = gr.Button(
+                    "Cerrar Sesión",
+                    size="sm"
+                )
+            
+            gr.Markdown("")
+            
+            drive_status_auth = gr.Textbox(
+                label="Estado de conexión",
+                value="No conectado",
+                interactive=False,
+                lines=2
+            )
+            
             gr.Markdown("")
             
             # Checkbox para Google Drive
             drive_checkbox = gr.Checkbox(
-                label="Guardar también en Google Drive",
+                label="Guardar en Google Drive",
                 value=False,
-                info="Requiere configuración de credenciales"
+                info="Primero debes conectar tu cuenta"
             )
             
+            gr.Markdown("")
+            gr.Markdown("---")
             gr.Markdown("")
             
             btn = gr.Button(
@@ -407,9 +510,9 @@ with gr.Blocks(title="Extractor IA de Facturas") as demo:
             
             gr.Markdown("")
             
-            # Estado de Google Drive
-            drive_status = gr.Textbox(
-                label="Estado de Google Drive",
+            # Estado de subida a Drive
+            drive_upload_status = gr.Textbox(
+                label="Estado de subida a Drive",
                 interactive=False,
                 lines=3
             )
@@ -463,14 +566,33 @@ with gr.Blocks(title="Extractor IA de Facturas") as demo:
     - *HF_TOKEN en Settings → Secrets (obligatorio)*
     - *credentials.json de Google Cloud para usar Drive (opcional)*
     
-    [Cómo obtener credentials.json](https://developers.google.com/drive/api/quickstart/python)
+    **Pasos para configurar Google Drive:**
+    1. Ve a [Google Cloud Console](https://console.cloud.google.com/)
+    2. Crea un proyecto y activa Google Drive API
+    3. Crea credenciales OAuth 2.0 (Tipo: Aplicación de escritorio)
+    4. Descarga el archivo credentials.json
+    5. Súbelo al directorio raíz de tu aplicación
+    6. Haz clic en "Conectar con Google Drive"
     """)
     
-    # Conectar botón con función
+    # Conectar botones
+    btn_conectar_drive.click(
+        fn=autenticar_drive,
+        inputs=[],
+        outputs=[drive_status_auth, drive_checkbox]
+    )
+    
+    btn_cerrar_sesion.click(
+        fn=cerrar_sesion_drive,
+        inputs=[],
+        outputs=[drive_status_auth, drive_checkbox]
+    )
+    
+    # Conectar botón principal
     btn.click(
         fn=procesar_factura,
         inputs=[pdf_input, drive_checkbox],
-        outputs=[texto_extraido, tabla_preview, csv_output, resumen_tecnico, info_util, drive_status]
+        outputs=[texto_extraido, tabla_preview, csv_output, resumen_tecnico, info_util, drive_upload_status]
     )
 
 if __name__ == "__main__":