Update app.py

app.py

@@ -26,14 +26,14 @@ def generate_story(caption):
         "You are a highly imaginative children's story writer celebrated for your creativity and captivating narratives. "
         "Using the image details provided below, please craft an enchanting tale tailored for children aged 3 to 10. "
         "Rather than simply reiterating the image details, enhance your story with imaginative characters, quirky adventures, "
-        "and delightful surprises that ignite wonder in every young heart. Let your narrative flow naturally and kindle the magic of storytelling. "
-        "Please ensure that your story is engaging, coherent, and the words count should lower than 100 in length.\n\n"
+        "and delightful surprises that ignite wonder in every young heart. Let your narrative flow naturally and ensure that your story is complete, with a clear beginning, middle, and end. "
+        "Please ensure the total word count does not exceed 100 words, and do not leave the story incomplete.\n\n"
         f"Image Details: {caption}\n\nStory:"
     )
     
     result = story_generator(
         prompt, 
-        max_new_tokens=100,
+        max_new_tokens=150,
         num_return_sequences=1,
         do_sample=True,
         temperature=1.0
@@ -45,6 +45,20 @@ def generate_story(caption):
     else:
         story = full_text.strip()
     
+    # 对生成的故事进行后处理，确保不超过 100 个单词且结尾完整
+    words = story.split()
+    if len(words) > 100:
+        trimmed = " ".join(words[:100])
+        # 查找最后一个完整结束的标点位置，尽量在句子边界截断
+        last_period = trimmed.rfind(".")
+        last_exclam = trimmed.rfind("!")
+        last_question = trimmed.rfind("?")
+        end_index = max(last_period, last_exclam, last_question)
+        if end_index != -1:
+            trimmed = trimmed[:end_index+1]
+        story = trimmed
+    if story and story[-1] not in ".!?":
+        story += "."
     return story
 
 def text_to_speech(text, output_file="output.mp3"):