Update app.py

app.py

@@ -178,20 +178,22 @@ def get_creations():
 
 def generate_image_hf(scene_idea, story):
     prompt = f"An artistic illustration of: {scene_idea}, dreamy and vivid"
-    for model in [
+    models = [
         "runwayml/stable-diffusion-v1-5",
         "stabilityai/stable-diffusion-2-1",
         "CompVis/stable-diffusion-v1-4"
-    ]:
-        for model in model:
-            try:
-                logging.info(f"trying model: {model}")
-                res = requests.post(
+    ]
+
+    for model in models:
+        try:
+            logging.info(f"Trying model: {model}")
+            res = requests.post(
                 f"https://api-inference.huggingface.co/models/{model}",
                 headers={"Authorization": f"Bearer {HUGGINGFACE_API_KEY}"},
                 json={"inputs": prompt},
-                timeout=15
+                timeout=15  # 15 seconds max
             )
+
             logging.info(f"Status: {res.status_code}")
             if res.status_code == 200:
                 import base64
@@ -205,10 +207,7 @@ def generate_image_hf(scene_idea, story):
             continue
 
     return generate_svg_placeholder(scene_idea, story)
-
-def generate_svg_placeholder(scene_idea, story):
-    return "<svg>Your beautiful fallback SVG here</svg>"
-
+        
 # DB setup
 from models import User, Creation
 with app.app_context():