Spaces:
Sleeping
Sleeping
| 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: | |
| 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." | |
| 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"] | |
| return gr.update(visible=True, value=f"Por favor, visita esta URL para autorizar la aplicación:\n{auth_url}\n\nLuego, copia el código de autorización y pégalo aquí como parámetro JSON: {{\"auth_code\": \"TU_CODIGO\"}}."), gr.update(visible=True) # Deja el campo para pegar el código visible | |
| # Formatear la respuesta de gmail_tool | |
| if accion == "leer_correos": | |
| messages = resultado.get("messages", []) | |
| formatted_messages = "" | |
| for message in messages: | |
| formatted_messages += f"ID: {message['id']}\nCuerpo: {message['body']}\n---\n" # Formateo básico. Adaptar según sea necesario. | |
| return formatted_messages | |
| elif accion == "enviar_correo": | |
| message_id = resultado.get("message_id") | |
| return f"Correo enviado con ID: {message_id}" | |
| elif accion == "verificar_almacenamiento": | |
| # Formatear la información de almacenamiento | |
| storage = resultado.get("storageQuota", {}) # Manejar el caso donde no hay storageQuota | |
| formatted_storage = f"Límite: {storage.get('limit', 'N/A')}\nUso: {storage.get('usage', 'N/A')}\nEn Drive: {storage.get('usageInDrive', 'N/A')}" | |
| return formatted_storage | |
| elif "error" in resultado: # Manejar errores de gmail_tool | |
| return f"Error: {resultado['error']}" | |
| return str(resultado) # Respuesta por defecto si no hay formato específico | |
| except Exception as e: | |
| return f"Error en gmail_interface: {e}" | |
| iface = gr.Interface( | |
| fn=gmail_interface, | |
| inputs=[ | |
| gr.Dropdown(["leer_correos", "enviar_correo", "verificar_almacenamiento"], label="Acción", info="Selecciona la acción que quieres realizar con Gmail."), | |
| gr.Textbox(label="Parámetros (JSON)", lines=3, info="Introduce los parámetros en formato JSON (ej. {\"maxResults\": 5})."), | |
| ], | |
| outputs=[gr.Textbox(label="Resultado"), gr.HTML(visible=False)], # Añade gr.HTML aunque no se use en este ejemplo | |
| title="Herramienta de Gmail", | |
| description="Herramienta para interactuar con Gmail.", | |
| allow_flagging="never", | |
| ) | |
| iface.queue().launch(share=True) # Agrega .queue().launch(share=True), importante para el flujo de OAuth. |