Update app.py

app.py

@@ -1,9 +1,8 @@
 import os
 from io import BytesIO
 from PIL import Image
-from diffusers  import AutoPipelineForText2Image
+from diffusers import AutoPipelineForText2Image
 import gradio as gr
-from generate_prompts import generate_prompt
 import base64
 
 # Load the model once at the start
@@ -34,6 +33,7 @@ def generate_image(prompt):
             image.save(buffered, format="JPEG")
             image_bytes = buffered.getvalue()
             img_str = base64.b64encode(image_bytes).decode("utf-8")
+            print(f'img_str: {img_str}')
             return img_str, None
         else:
             print(f"No images found in model output")
@@ -54,7 +54,7 @@ def inference(sentence_mapping, character_dict, selected_style):
         images = {}
         for paragraph_number, sentences in sentence_mapping.items():
             combined_sentence = " ".join(sentences)
-            prompt = generate_prompt(combined_sentence, sentence_mapping, character_dict, selected_style)
+            prompt = f"Make an illustration in {selected_style} style from: {combined_sentence}"
             img_str, error = generate_image(prompt)
             if error:
                 images[paragraph_number] = f"Error: {error}"