Update app.py

app.py

@@ -25,7 +25,10 @@ def generate_image(prompt, prompt_name):
         output = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
         print(f"Model output for {prompt_name}: {output}")
 
-        if output and hasattr(output, 'images') and output.images:
+        if output is None:
+            raise ValueError(f"Model returned None for {prompt_name}")
+
+        if hasattr(output, 'images') and output.images:
             print(f"Image generated for {prompt_name}")
             image = output.images[0]
             buffered = BytesIO()
@@ -33,15 +36,13 @@ def generate_image(prompt, prompt_name):
             image_bytes = buffered.getvalue()
             return image_bytes
         else:
-            print(f"No images found or generated output is None for {prompt_name}")
-            return None
+            raise ValueError(f"No images found in model output for {prompt_name}")
     except Exception as e:
         print(f"An error occurred while generating image for {prompt_name}: {e}")
         return None
 
 async def queue_api_calls(sentence_mapping, character_dict, selected_style):
     print("Starting to queue API calls...")
-    print(f'sentence_mapping"{sentence_mapping}')
     prompts = []
     for paragraph_number, sentences in sentence_mapping.items():
         combined_sentence = " ".join(sentences)