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:
        print(f"Acción recibida: {accion}")  # Depuración
        print(f"Parámetros recibidos: {parametros}")  # Depuración
        
        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 as e:
                return "Error al decodificar JSON: " + str(e), "" #  Devolver dos valores

            auth_code = parametros.get("auth_code")
            if auth_code:
                print(f"Auth code recibido: {auth_code}") # Depuración
                try:
                    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)
                except Exception as e:
                    return f"Error en la autorización OAuth: {e}"
                # 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) #  No enviar parametros vacíos

        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:
            return f"Error en gmail_tool: {resultado['error']}", "" #  Muestra el error de gmail_tool
        else:
            return str(resultado), ""  # Respuesta por defecto


    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.