Merge branch 'develop' into fix_websocket_disconnect

Author: massongit

README.md

@@ -153,6 +153,8 @@ pipenv run pre-commit install
     にゃーん ... 「よしよし」と返す。
     おみくじ ... おみくじを引いて返す。
     version ... バージョン情報を表示する。
+    chat [text] ... ChatGPTで文章生成する。
+    画像生成 [text] ... openAIのImage generationで画像生成する。
     ```
 
 ## バージョンアップによる変更点

commands.txt

@@ -19,3 +19,5 @@ text delete [int] ... 指定した番号[int]のパワーワードを削除す
 にゃーん ... 「よしよし」と返す。
 おみくじ ... おみくじを引いて返す。
 version ... バージョン情報を表示する。
+chat [text] ... ChatGPTで文章生成する。
+画像生成 [text] ... openAIのImage generationで画像生成する。

library/chat_gpt.py

@@ -13,3 +13,8 @@ def chat_gpt(message: str) -> str:
         ],
     )
     return result.get("choices")[0].message.content
+
+
+def image_create(message: str) -> str:
+    response = openai.Image.create(prompt=message, n=1, size="512x512")
+    return response["data"][0]["url"]

plugins/analyze.py

@@ -36,6 +36,9 @@ def analyze_message(message: str) -> Callable[[BaseClient], None]:
         "にゃーん": lambda m: hato.yoshiyoshi,
         "おみくじ": lambda m: hato.omikuji,
         "chat": lambda m: partial(hato.chat, message=m[len("chat") :].strip()),
+        "画像生成": lambda m: partial(
+            hato.image_generate, message=m[len("画像生成") :].strip()
+        ),
     }
 
     for key, method in conditions.items():

plugins/hato.py

@@ -18,7 +18,7 @@
 from git.exc import GitCommandNotFound, InvalidGitRepositoryError
 
 import slackbot_settings as conf
-from library.chat_gpt import chat_gpt
+from library.chat_gpt import chat_gpt, image_create
 from library.clientclass import BaseClient
 from library.earthquake import generate_map_img, get_quake_list
 from library.geo import get_geo_data
@@ -476,3 +476,22 @@ def chat(message: str):
     """
 
     return chat_gpt(message=message)
+
+
+@action("画像生成", with_client=True)
+def image_generate(client: BaseClient, message: str):
+    """
+    画像生成を行う
+    """
+
+    url = image_create(message=message)
+
+    """
+    urlから画像ファイルをダウンロードして、画像を返す
+    """
+    with NamedTemporaryFile() as generated_file:
+        generated_file.write(requests.get(url).content)
+        client.upload(
+            file=generated_file.name,
+            filename="image.png",
+        )