Update app.py

app.py

@@ -17,18 +17,21 @@ def extract_image_caption(image_data):
 def compose_story_from_caption(caption_detail):
     prompt_text = (
         "You are a talented and imaginative storyteller for children aged 3 to 10. "
-        "Using the details derived from the image below, craft a captivating tale that goes beyond merely describing the scene. "
-        "Let your creativity shine by introducing engaging characters, adventurous journeys, and delightful surprises. "
+        "Using the details derived from the image below, craft a captivating tale that includes three main characters, "
+        "an adventurous journey, and delightful surprises. "
         "Your story should be vivid, original, and between 90 and 100 words in length.\n\n"
         f"Image Details: {caption_detail}\n\nStory:"
     )
-    story_results = story_pipeline(prompt_text, num_return_sequences=1)
+    
+    # 生成故事，设置最大长度为150，以确保有足够空间生成90-100词的内容
+    story_results = story_pipeline(prompt_text, num_return_sequences=1, max_length=150)
     story_text = story_results[0]['generated_text']
     
+    # 提取故事文本
     if "Story:" in story_text:
         return story_text.split("Story:", 1)[1].strip()
     return story_text.strip()
-
+    
 def convert_text_to_audio(text_content, audio_path="output.mp3"):
     """
     将文本转换为音频文件。