hugging-chat-comint.py

#!/usr/bin/env python

import argparse
import json

try:
    from hugchat import hugchat
except Exception:
    import hugchat


arg_parser = argparse.ArgumentParser(
    description="See https://github.com/Soulter/hugging-chat-api"
)

arg_parser.add_argument("cookies", help="JSON string or cookies files path")
arg_parser.add_argument(
    "-p",
    "--prompt",
    default=">>>> ",
    help='Prompt indicator, a space is ensured at the end, defaults to ">>>> "',
)

parsed_args = arg_parser.parse_args()

try:
    cookies_dict = json.loads(parsed_args.cookies)
except ValueError:
    cookies_dict = None

if cookies_dict:
    chatbot = hugchat.ChatBot(cookies=cookies_dict)
else:
    chatbot = hugchat.ChatBot(cookie_path=parsed_args.cookies)

prompt_indicator = parsed_args.prompt
if prompt_indicator[-1] != " ":
    prompt_indicator += " "


while True:
    try:
        prompt = input(prompt_indicator)
        if prompt != "":
            print(chatbot.chat(prompt))
    except EOFError:
        break