import gradio as gr
#from hugchat import hugchat
#from hugchat.login import Login
import getpass  # Asegúrate de importar esta biblioteca para obtener la contraseña (en caso no hayas ejecutado las primeras celdas)

# Log in to huggingface and grant authorization to huggingchat
#email = input("Ingrese su correo electrónico: ")
#passwd = getpass.getpass("Ingrese su contraseña: ")

#sign = Login(email, passwd)
#cookies = sign.login()

# Save cookies to the local directory
cookie_path_dir = "./cookies_snapshot"
sign.saveCookiesToDir(cookie_path_dir)

# Create a ChatBot instance with cookies
chatbot = hugchat.ChatBot(cookies=cookies.get_dict())

def generate_response(prompt):
    response = chatbot.chat(prompt)
    return response

def hugchat_app(prompt):
    response = generate_response(prompt)

    # Guarda la conversación en un archivo .txt
    with open("conversacion.txt", "a") as file:
        file.write("Usuario: " + prompt + "\n")
        file.write("ChatBot: " + response + "\n\n")

    return response

iface = gr.Interface(fn=hugchat_app, inputs="text", outputs="text", title="🤗💬 ELISA I MODELO DE INTELIGENCIA ARTIFICIAL PROF: GERARDO", # Modificar como quieras
                     description="Esta aplicación es un chatbot impulsado por LLM creado utilizando HugChat. 💡",
                     article="Creado con 🖤 por                        AREA DE INFORMATICA COLEGIO VILLA ELISA IED",
                     )
iface.launch(share=True, debug=True) # Si el problema se soluciona cambiar share a True para compartir URL (probar en unos días)