Update utils.py

utils.py

@@ -47,19 +47,15 @@ def split_story_to_pages(story_text):
     return eval(story_data)
 
 def generate_image(prompt):
-    headers = {
-        "Authorization": f"Bearer {LEONARDO_API_KEY}",
-        "Content-Type": "application/json"
-    }
-    data = {
-        "prompt": prompt,
-        "num_images": 1,
-        "modelId": "REPLACE_WITH_MODEL_ID"
-    }
+    api_key = os.getenv("DEEPAI_API_KEY")
 
     response = requests.post(
-        "https://cloud.leonardo.ai/api/rest/v1/generations",
-        headers=headers,
-        json=data
+        "https://api.deepai.org/api/text2img",
+        data={'text': prompt},
+        headers={'api-key': api_key}
     )
-    return response.json()['data'][0]['url']
+
+    if response.status_code == 200:
+        return response.json().get('output_url')
+    else:
+        return "https://via.placeholder.com/512?text=Image+Error"