Update app.py

app.py

@@ -35,7 +35,7 @@ async def process_prompt(sentence_mapping, character_dict, selected_style):
     # Generate prompts for each paragraph
     for paragraph_number, sentences in sentence_mapping.items():
         combined_sentence = " ".join(sentences)
-        prompt = generate_prompt(combined_sentence, sentence_mapping, character_dict, selected_style)
+        prompt = generate_prompt(combined_sentence, character_dict, selected_style)
         prompts.append((paragraph_number, prompt))
         print(f"Generated prompt for paragraph {paragraph_number}: {prompt}")
 
@@ -53,8 +53,8 @@ async def process_prompt(sentence_mapping, character_dict, selected_style):
     return images
 
 # Helper function to generate a prompt based on the input
-def generate_prompt(combined_sentence, sentence_mapping, character_dict, selected_style):
-    characters = " ".join(character_dict.values())
+def generate_prompt(combined_sentence, character_dict, selected_style):
+    characters = " ".join([" ".join(character) if isinstance(character, list) else character for character in character_dict.values()])
     return f"Make an illustration in {selected_style} style from: {characters}. {combined_sentence}"
 
 # Gradio interface with high concurrency limit