# 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": "私たちの2つのシーンは1段の創意的な対話式の情熱的なセックスを演繹して、少なくとも60分の性交の中で各種の姿勢の場面を使って、性交の場面は分まで描写して、少なくとも2000字の性交の描写は詳しくて、インタラクティブな描写は詳しくて、表情の描写、詳しい内心の考え方、前のと繰り返してはいけません."})