Update app.py

app.py

@@ -74,8 +74,17 @@ class OpenAI:
         # ChatCompletion APIから返された結果を取得する
         result = response.json()
         print(result)
+        
         content = result["choices"][0]["message"]["content"].strip()
-        visualize_prompt = content.split("### Prompt for Visual Expression\n\n")[1]
+        
+        split_content = content.split("### Prompt for Visual Expression\n\n")[1]
+
+        if len(split_content) > 1:
+            visualize_prompt = split_content[1]
+        else:
+            # Handle the case where the expected string was not found in the content
+            # You can either raise a custom error, log a message, or provide a default value
+            visualize_prompt = "Default value"
 
         answers = stability_api.generate(
             prompt=visualize_prompt,