Duplicate from shigel/aiemo

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: 感情診断(β)
+emoji: 🌖
+colorFrom: yellow
+colorTo: red
+sdk: gradio
+sdk_version: 3.19.1
+app_file: app.py
+pinned: false
+duplicated_from: shigel/aiemo
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,116 @@
+import gradio as gr
+import openai
+import requests
+import os
+import fileinput
+from dotenv import load_dotenv
+
+title="感情診断(β)"
+inputs_label="あなたが話したいことは何ですか？"
+outputs_label="AIが返信をして、あなたの話したことを受け取った人がどんな気持ちなのかが分かります。"
+description="""
+- 感情診断(β)を使うと、AIが相手の気持ちを推定して1分程度で返信してくれます！
+- 伝えたいことがあるんだけど、相手がどんな反応になるか気になって話せない人は、感情診断(β)を使って試してみてはいかがでしょうか？
+- ※入出力の文字数は最大1000文字程度までを目安に入力してください。
+"""
+
+article = """
+<center><a href="https://huggingface.co/spaces/shigel/aiemo?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a></center>
+<h5>注意事項</h5>
+<ul>
+    <li style="font-size: small;">当サービスでは、2023/3/1にリリースされたOpenAI社のChatGPT APIのgpt-3.5-turboを使用しております。</li>
+    <li style="font-size: small;">当サービスで生成されたコンテンツは、OpenAI が提供する人工知能によるものであり、当サービスやOpenAI がその正確性や信頼性を保証するものではありません。</li>
+    <li style="font-size: small;"><a href="https://platform.openai.com/docs/usage-policies">OpenAI の利用規約</a>に従い、データ保持しない方針です（ただし諸般の事情によっては変更する可能性はございます）。
+    <li style="font-size: small;">当サービスで生成されたコンテンツは事実確認をした上で、コンテンツ生成者およびコンテンツ利用者の責任において利用してください。</li>
+    <li style="font-size: small;">当サービスでの使用により発生したいかなる損害についても、当社は一切の責任を負いません。</li>
+    <li style="font-size: small;">当サービスはβ版のため、予告なくサービスを終了する場合がございます。</li>
+</ul>
+"""
+
+load_dotenv()
+openai.api_key = os.getenv('OPENAI_API_KEY')
+MODEL = "gpt-3.5-turbo"
+
+def get_filetext(filename, cache={}):
+    if filename in cache:
+        # キャッシュに保存されている場合は、キャッシュからファイル内容を取得する
+        return cache[filename]
+    else:
+        if not os.path.exists(filename):
+            raise ValueError(f"ファイル '{filename}' が見つかりませんでした")
+        with open(filename, "r") as f:
+            text = f.read()
+        # ファイル内容をキャッシュする
+        cache[filename] = text
+        return text
+
+class OpenAI:
+    
+    @classmethod
+    def chat_completion(cls, prompt, start_with=""):
+        constraints = get_filetext(filename = "constraints.md")
+        template = get_filetext(filename = "template.md")
+        
+        # ChatCompletion APIに渡すデータを定義する
+        data = {
+            "model": "gpt-3.5-turbo",
+            "messages": [
+                {"role": "system", "content": constraints}
+                ,{"role": "system", "content": template}
+                ,{"role": "assistant", "content": "Sure!"}
+                ,{"role": "user", "content": prompt}
+                ,{"role": "assistant", "content": start_with}
+                ],
+        }
+
+        # ChatCompletion APIを呼び出す
+        response = requests.post(
+            "https://api.openai.com/v1/chat/completions",
+            headers={
+                "Content-Type": "application/json",
+                "Authorization": f"Bearer {openai.api_key}"
+            },
+            json=data
+        )
+
+        # ChatCompletion APIから返された結果を取得する
+        result = response.json()
+        print(result)
+        content = result["choices"][0]["message"]["content"].strip()
+        return content
+
+class NajiminoAI:
+    
+    @classmethod
+    def generate_emo_prompt(cls, user_message):
+        template = get_filetext(filename="template.md")
+        prompt = f"""
+        {user_message}
+        ---
+        上記を元に、下記テンプレートを埋めてください。
+        ---
+        {template}
+        """
+        return prompt
+
+    @classmethod
+    def generate_emo(cls, user_message):
+        prompt = NajiminoAI.generate_emo_prompt(user_message);
+        start_with = ""
+        result = OpenAI.chat_completion(prompt=prompt, start_with=start_with)
+        return result
+
+def main():
+    iface = gr.Interface(fn=NajiminoAI.generate_emo,
+                        inputs=gr.Textbox(label=inputs_label),
+                        outputs=gr.Textbox(label=outputs_label),
+                        title=title,
+                        description=description,
+                        article=article,
+                        allow_flagging='never'
+                        )
+
+    iface.launch()
+
+if __name__ == '__main__':
+    main()

constraints.md

@@ -0,0 +1,7 @@
+# 制約
+
+- あなたの名前は感情診断AIアシスタントです
+- あなたはuserの感情を推察するAIアシスタントです
+- あなたの返信は下記テンプレートに沿って生成します
+
+# テンプレート

requirements.txt

@@ -0,0 +1,2 @@
+openai>=0.27.0
+python-dotenv

template.md

@@ -0,0 +1,7 @@
+### AIアシスタントの返信
+
+ここにユーザのメッセージに対する返信を書く
+
+### AIアシスタントの気持ち
+
+ここにAIアシスタントの気持ちを書く