🔧 Refactor authentication to use global variables for Drive service and folders

app.py

@@ -20,6 +20,8 @@ SCOPES = ['https://www.googleapis.com/auth/drive.readonly']
 CHROMA_DIR = "chroma_db"
 
 chain = None
+carpetas_drive_global = {}
+servicio_drive_global = None
 
 # Autenticación manual para Hugging Face
 
@@ -27,7 +29,6 @@ def authenticate_manual(credentials_path, code):
     with open(credentials_path, "r") as f:
         client_config = json.load(f)
 
-    # Forzar uso de redirect_uri manual
     client_config["installed"]["redirect_uris"] = ["urn:ietf:wg:oauth:2.0:oob"]
     flow = InstalledAppFlow.from_client_config(client_config, SCOPES, redirect_uri="urn:ietf:wg:oauth:2.0:oob")
     auth_url, _ = flow.authorization_url(prompt='consent')
@@ -37,9 +38,9 @@ def authenticate_manual(credentials_path, code):
         creds = flow.credentials
         with open("token.json", 'w') as token:
             token.write(creds.to_json())
-        return build('drive', 'v3', credentials=creds), "✅ Autenticado correctamente."
+        return build('drive', 'v3', credentials=creds), "✅ Conectado correctamente.", ""
 
-    return None, auth_url
+    return None, "Copia y abre este enlace para autorizar:", f"[{auth_url}]({auth_url})"
 
 # Obtener carpetas de Drive
 
@@ -148,25 +149,24 @@ with gr.Blocks(title="Google Drive RAG") as demo:
         user_input = gr.Textbox(label="Escribe tu pregunta")
         enviar_btn = gr.Button("Enviar pregunta")
 
-    carpetas_drive = gr.State({})
-    servicio_drive = gr.State(None)
-
     def conectar_drive(cred_file, auth_code):
+        global carpetas_drive_global, servicio_drive_global
         try:
-            service, result = authenticate_manual(cred_file.name, auth_code.strip())
+            service, estado_text, link = authenticate_manual(cred_file.name, auth_code.strip())
             if not service:
-                return [], "Copia y abre este enlace para autorizar:", f"[{result}]({result})"
+                return gr.update(choices=[]), estado_text, link
+
             carpetas = listar_carpetas_drive(service)
-            carpetas_drive.value = carpetas
-            servicio_drive.value = service
-            return list(carpetas.keys()), "✅ Conectado correctamente.", ""
+            carpetas_drive_global = carpetas
+            servicio_drive_global = service
+            return gr.update(choices=list(carpetas.keys())), estado_text, ""
         except Exception as e:
-            return [], f"❌ Error: {e}", ""
+            return gr.update(choices=[]), f"❌ Error: {e}", ""
 
     def procesar_pdfs(nombre_carpeta, cred_file, tok_file, api_key):
         try:
-            service = servicio_drive.value
-            folder_id = carpetas_drive.value.get(nombre_carpeta)
+            service = servicio_drive_global
+            folder_id = carpetas_drive_global.get(nombre_carpeta)
             if not folder_id:
                 return "❌ Carpeta no encontrada."
             folder_path = get_pdfs_from_drive_by_id(service, folder_id)