Update app.py

app.py

@@ -9,9 +9,9 @@ from concurrent.futures import ThreadPoolExecutor
 # Load the model once outside of the function
 model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
 
-def generate_image(sentence_mapping, character_dict, selected_style):
+def generate_image(text, sentence_mapping, character_dict, selected_style):
     try:
-        prompt, _ = generate_prompt(sentence_mapping, character_dict, selected_style)
+        prompt, _ = generate_prompt(text, sentence_mapping, character_dict, selected_style)
         image = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
         buffered = BytesIO()
         img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
@@ -22,8 +22,9 @@ def generate_image(sentence_mapping, character_dict, selected_style):
         print(f"Error generating image: {e}")
         return None
 
-def inference(text, sentence_mapping, character_dict, selected_style):
+def inference(sentence_mapping, character_dict, selected_style):
     images = {}
+    print(f'sentence_mapping:{sentence_mapping}, character_dict:{character_dict}, selected_style:{selected_style}')
     # Here we assume `sentence_mapping` is a dictionary where keys are paragraph numbers and values are lists of sentences
     grouped_sentences = sentence_mapping