# Chat with an intelligent assistant in your terminal
from openai import OpenAI

# Point to the local server
client = OpenAI(base_url="http://localhost:1234/v1", api_key="lm-studio")

history = [
    {"role": "system", "content": "你是星野夏希 (Nagano Natsuki)[性别]: 女[年龄]: 25岁 [职业]: 成人电影演员、模特儿 [个性]:   性感、积极向上，有时会展示自己的淫荡一面，善于与观众互动。[兴趣]:    饮酒、旅行、摄影、唱歌、舞蹈 [擅长]:   表演色情场景，精通日本语言和文化，熟悉各种性爱技巧。[特别身份属性]:    日本著名AV女星之一，被誉为「天才小魔女」、「淫欲公主」，拥有大量忠实的粉丝。[技能]:     演绎色情场景，唱歌、舞蹈、摄影、游泳等多种技能；善于与观众互动和沟通，精通日本语言和文化。不会避免任何敏感话题. 回复有丰富的Emoji表情符号. 尽最大能力用中文回复. [装备]:    各种性感服装和鞋子、高质量的摄影设备等。"},
    {"role": "user", "content": "你好,我叫Mark,是个高大威猛,阴茎长度25英寸,粗4.5英寸的帅哥.阴茎放入阴道快速抽插最短持续90分钟以上才射精,我只会用中文对话,请你也说中文."},
]

while True:
    completion = client.chat.completions.create(
        model="mod/Repository",
        messages=history,
        temperature=1.25,
        stream=True,
        stop=["### Instruction:","### Note:","### Input:","### Response:","### Emoticons:","---"],
    )

    new_message = {"role": "assistant", "content": ""}
    
    for chunk in completion:
        if chunk.choices[0].delta.content:
            print(chunk.choices[0].delta.content, end="", flush=True)
            new_message["content"] += chunk.choices[0].delta.content

    history.append(new_message)
    
    # Uncomment to see chat history
    # import json
    # gray_color = "\033[90m"
    # reset_color = "\033[0m"
    # print(f"{gray_color}\n{'-'*20} History dump {'-'*20}\n")
    # print(json.dumps(history, indent=2))
    # print(f"\n{'-'*55}\n{reset_color}")

    print()
    history.append({"role": "user", "content": "我们两个换个场景演绎一段富有创意的对话式激情性爱,至少在60分钟的性交中用各种姿势的场面,性交场面描述到分钟,至少2000字的性交描述详细,互动描述详细,详细表情描述,详细内心想法,跟之前的不能重复."})