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

def generate_response(prompt):
    response = hugchat(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:
         write("Usuario: " + prompt + "\n")
         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=False, debug=True) # Si el problema se soluciona cambiar share a True para compartir URL (probar en unos días)