Merge branch 'develop' into add_help_chatgpt

Author: Goryudyuma

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",
+        )