Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import openai | |
| import requests | |
| import os | |
| import fileinput | |
| from dotenv import load_dotenv | |
| title="あらすじのアイデア提案AI" | |
| inputs_label=""" | |
| - 物語のあらすじのアイデアの元になるキーワードを入力してください。 | |
| - 登場人物とその設定、物語の舞台、出来事など | |
| """ | |
| outputs_label="あらすじの提案" | |
| description=""" | |
| - あなたが入力したキーワードに応じて、AIが物語のあらすじのアイデアを提案します。 | |
| """ | |
| article = """ | |
| <!-- | |
| <h4>エンジニア募集中！</h4> | |
| 弊社ではAIを利用したサービスを一緒に開発してくれる仲間を募集しております！ | |
| <p> | |
| 当サービスに興味を持っていただいた方は、お問い合わせフォームよりご連絡ください！ | |
| <p> | |
| <ul> | |
| <li"><a href="https://www.najimino.co.jp/contact">お問い合わせフォーム</a></li> | |
| --> | |
| <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: | |
| 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: | |
| def generate_emo_prompt(cls, user_message): | |
| template = get_filetext(filename="template.md") | |
| prompt = f""" | |
| {user_message} | |
| --- | |
| 上記を元に、下記テンプレートを埋めてください。 | |
| --- | |
| {template} | |
| """ | |
| return prompt | |
| 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() |