Update app.py

app.py

@@ -13,8 +13,9 @@ print("Model loaded successfully.")
 
 def generate_image(prompt, prompt_name):
     try:
-        print(f"Generating image for {prompt_name}")
+        print(f"Generating image for {prompt_name} with prompt: {prompt}")
         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:
             print(f"Image generated for {prompt_name}")
@@ -48,7 +49,7 @@ async def queue_api_calls(sentence_mapping, character_dict, selected_style):
         responses = await asyncio.gather(*tasks)
     
     images = {paragraph_number: response for (paragraph_number, _), response in zip(prompts, responses)}
-    print("Finished queuing API calls.")
+    print("Finished queuing API calls. Generated images: ", images)
     return images
 
 def process_prompt(sentence_mapping, character_dict, selected_style):
@@ -65,7 +66,7 @@ def process_prompt(sentence_mapping, character_dict, selected_style):
         print("Created new event loop.")
 
     cmpt_return = loop.run_until_complete(queue_api_calls(sentence_mapping, character_dict, selected_style))
-    print("Prompt processing complete.")
+    print("Prompt processing complete. Generated images: ", cmpt_return)
     return cmpt_return
 
 gradio_interface = gr.Interface(