fix: app.py and requirements.txt

app.py

@@ -4,10 +4,15 @@ import requests
 import os
 import fileinput
 from dotenv import load_dotenv
+import io
+from PIL import Image
+from stability_sdk import client
+import stability_sdk.interfaces.gooseai.generation.generation_pb2 as generation
 
 title="ochyAI recipe generator"
 inputs_label="どんな料理か教えてくれれば，新しいレシピを考えます"
 outputs_label="ochyAIが返信をします"
+visual_outputs_label="料理のイメージ"
 description="""
 - ※入出力の文字数は最大1000文字程度までを目安に入力してください。解答に120秒くらいかかります．
 """
@@ -17,6 +22,11 @@ article = """
 
 load_dotenv()
 openai.api_key = os.getenv('OPENAI_API_KEY')
+os.environ['STABILITY_HOST'] = 'grpc.stability.ai:443'
+stability_api = client.StabilityInference(
+    key=os.getenv('STABILITY_KEY'), 
+    verbose=True,
+)
 MODEL = "gpt-4"
 
 def get_filetext(filename, cache={}):
@@ -65,8 +75,20 @@ class OpenAI:
         result = response.json()
         print(result)
         content = result["choices"][0]["message"]["content"].strip()
-        return content
+        visualize_prompt = content.split("### Prompt for Visual Expression\n\n")[1]
 
+        answers = stability_api.generate(
+            prompt=visualize_prompt,
+        )
+
+        for resp in answers:
+            for artifact in resp.artifacts:
+                if artifact.finish_reason == generation.FILTER:
+                    print("NSFW")
+                if artifact.type == generation.ARTIFACT_IMAGE:
+                    img = Image.open(io.BytesIO(artifact.binary))
+                    return [content, img]
+        
 class NajiminoAI:
     
     @classmethod
@@ -91,7 +113,7 @@ class NajiminoAI:
 def main():
     iface = gr.Interface(fn=NajiminoAI.generate_emo,
                         inputs=gr.Textbox(label=inputs_label),
-                        outputs=gr.Textbox(label=outputs_label),
+                        outputs=[gr.Textbox(label=inputs_label), gr.Image(label=visual_outputs_label)],
                         title=title,
                         description=description,
                         article=article,

requirements.txt

@@ -1,2 +1,3 @@
 openai>=0.27.0
-python-dotenv
+python-dotenv
+stability-sdk