Update app.py

app.py

@@ -13,6 +13,7 @@ async def load_model():
     model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
     print("Model loaded successfully.")
 
+# Run the model loading
 asyncio.run(load_model())
 
 async def generate_image(prompt, prompt_name):
@@ -47,7 +48,7 @@ async def queue_api_calls(sentence_mapping, character_dict, selected_style):
     for paragraph_number, sentences in sentence_mapping.items():
         combined_sentence = " ".join(sentences)
         print(f"combined_sentence for paragraph {paragraph_number}: {combined_sentence}")
-        prompt = generate_prompt(combined_sentence, character_dict, selected_style)  # Correct prompt generation
+        prompt = generate_prompt(combined_sentence, sentence_mapping, character_dict, selected_style)  # Correct prompt generation
         prompts.append((paragraph_number, prompt))
         print(f"Generated prompt for paragraph {paragraph_number}: {prompt}")