Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 from diffusers import AutoPipelineForText2Image
-from io import BytesIO
 from generate_propmts import generate_prompt
 from concurrent.futures import ThreadPoolExecutor, as_completed
-import PIL
+from PIL import Image
+import traceback
 
 # Load the model once outside of the function
 model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
@@ -21,6 +21,7 @@ def generate_image(prompt):
             
     except Exception as e:
         print(f"Error generating image: {e}")
+        traceback.print_exc()
         return None
 
 def inference(sentence_mapping, character_dict, selected_style):
@@ -45,6 +46,7 @@ def inference(sentence_mapping, character_dict, selected_style):
                     images.append(image)
             except Exception as e:
                 print(f"Error processing prompt: {e}")
+                traceback.print_exc()
 
     return images