Update app.py

app.py

@@ -4,7 +4,7 @@ from PIL import Image
 from diffusers import AutoPipelineForText2Image
 import gradio as gr
 import base64
-
+from generate_prompts import generate_prompt
 
 CONCURRENCY_LIMIT = 10
 
@@ -54,13 +54,10 @@ def inference(sentence_mapping, character_dict, selected_style):
         print(f"Received character_dict: {character_dict}, type: {type(character_dict)}")
         print(f"Received selected_style: {selected_style}, type: {type(selected_style)}")
 
-        if sentence_mapping is None or character_dict is None or selected_style is None:
-            return {"error": "One or more inputs are None"}
-
         images = {}
         for paragraph_number, sentences in sentence_mapping.items():
             combined_sentence = " ".join(sentences)
-            prompt = f"Make an illustration in {selected_style} style from: {combined_sentence}"
+            prompt = generate_prompt(combined_sentence,character_dict, selected_style)
             print(f"Generated prompt for paragraph {paragraph_number}: {prompt}")
             img_str, error = generate_image(prompt)
             if error: