|
import gradio as gr |
|
import openai |
|
import requests |
|
import os |
|
import fileinput |
|
from dotenv import load_dotenv |
|
|
|
title="支援プラン作成アシスタントβ-GPT-4 v2.0" |
|
inputs_label="支援対象の情報を教えてください" |
|
outputs_label="AIの支援プラン" |
|
description=""" |
|
あなたが入力した情報を参照して、AIがアセスメントと支援プランを提案します。 |
|
""" |
|
|
|
|
|
article = """ |
|
<h5>リリースノート<h5> |
|
<ul> |
|
<li style="font-size: small;">2023-05-01-GPT-4 v1.1</li> |
|
<li style="font-size: small;">2023-05-01-GPT-4 v2.0</li> |
|
</ul> |
|
|
|
<h5>注意事項</h5> |
|
<ul> |
|
<li style="font-size: small;">当サービスでは、2023/3/14-GPT-4にリリースされたOpenAI社のChatGPT APIのgpt-4を使用しております。</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-4" |
|
|
|
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") |
|
|
|
|
|
data = { |
|
"model": "gpt-4", |
|
"messages": [ |
|
{"role": "system", "content": constraints} |
|
,{"role": "system", "content": template} |
|
,{"role": "assistant", "content": "Sure!"} |
|
,{"role": "user", "content": prompt} |
|
,{"role": "assistant", "content": start_with} |
|
], |
|
} |
|
|
|
|
|
response = requests.post( |
|
"https://api.openai.com/v1/chat/completions", |
|
headers={ |
|
"Content-Type": "application/json", |
|
"Authorization": f"Bearer {openai.api_key}" |
|
}, |
|
json=data |
|
) |
|
|
|
|
|
result = response.json() |
|
print(result) |
|
content = result["choices"][0]["message"]["content"].strip() |
|
return content |
|
|
|
class MasasanAI: |
|
|
|
@classmethod |
|
def generate_plan_prompt(cls, user_message): |
|
template = get_filetext(filename="template.md") |
|
prompt = f""" |
|
{user_message} |
|
--- |
|
上記を元に、下記テンプレートを埋めてください。 |
|
--- |
|
{template} |
|
""" |
|
return prompt |
|
|
|
@classmethod |
|
def generate_plan(cls, user_message): |
|
prompt = MasasanAI.generate_plan_prompt(user_message); |
|
start_with = "" |
|
result = OpenAI.chat_completion(prompt=prompt, start_with=start_with) |
|
return result |
|
|
|
def main(): |
|
iface = gr.Interface(fn=MasasanAI.generate_plan, |
|
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() |