hostserver3 / main.py
abdullahalioo's picture
Update main.py
74a7b28 verified
raw
history blame
1.05 kB
from hugchat import hugchat
from hugchat.login import Login
import os
# Credentials
EMAIL = os.getenv("EMAIL")
PASSWD = os.getenv("PASSWD")
# Path to store cookies
cookie_path_dir = "./cookies/" # Make sure this directory exists
# Log in and get cookies
sign = Login(EMAIL, PASSWD)
cookies = sign.login(cookie_dir_path=cookie_path_dir, save_cookies=True)
# Create ChatBot with cookies
chatbot = hugchat.ChatBot(cookies=cookies.get_dict())
# πŸ” Create a new conversation with a custom assistant ID
ASSISTANT_ID = "66017fca58d60bd7d5c5c26c" # Replace with your actual assistant ID
chatbot.new_conversation(assistant=ASSISTANT_ID, switch_to=True)
# πŸ” Stream response character-by-character as it generates
print("Assistant:", end=" ", flush=True)
for token in chatbot.chat("Hello, how can I help you today?", stream=True):
print(token, end="", flush=True)
# Optionally: web search example
# response = chatbot.chat("How many models stored in huggingface?", web_search=True)
# print("\n\nWeb Search Result:", response.wait_until_done())