Update app.py

app.py

@@ -45,20 +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 += "."
+    # 对生成的故事进行后处理，确保不超过 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"):