Update app.py

app.py

@@ -1,13 +1,33 @@
 import gradio as gr
 import comunicacion_gmail
 import json
+from urllib.parse import urlparse, parse_qs
 
 def gmail_interface(accion, parametros):
     """Función que llama a gmail_tool y formatea la respuesta."""
     try:
-        parametros = json.loads(parametros) # Convierte la cadena JSON a un diccionario
+        if parametros:  # Si hay parámetros, verifica si contiene el auth_code
+            try:
+                parametros = json.loads(parametros) # Convierte la cadena JSON a un diccionario
+            except json.JSONDecodeError:
+                return "Error: Parámetros JSON inválidos."
 
-        resultado = comunicacion_gmail.gmail_tool(accion, parametros)
+            auth_code = parametros.get("auth_code")
+            if auth_code:
+                # Completar el flujo OAuth con el auth_code
+                flow = InstalledAppFlow.from_client_secrets_file(
+                    'credentials.json', comunicacion_gmail.SCOPES) # Accede a SCOPES desde comunicacion_gmail
+                flow.fetch_token(code=auth_code)
+                creds = flow.credentials
+                with open('token.pickle', 'wb') as token:
+                    pickle.dump(creds, token)
+                # Vuelve a llamar a gmail_tool para ejecutar la acción original (sin el auth_code)
+                del parametros["auth_code"]  # Eliminar auth_code antes de la siguiente llamada
+                resultado = comunicacion_gmail.gmail_tool(accion, parametros)
+            else:
+                resultado = comunicacion_gmail.gmail_tool(accion, parametros)
+        else:  # Si no hay parámetros, llama a gmail_tool
+            resultado = comunicacion_gmail.gmail_tool(accion, parametros)
 
         if "requires_auth" in resultado and resultado["requires_auth"]:
             auth_url = resultado["auth_url"]