Update app.py

app.py

@@ -11,9 +11,37 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
-    story_text = pipe(text)[0]['generated_text']
-    return story_text
+    # 添加儿童故事专用prompt模板
+    prompt = f"""Generate a VERY SHORT fairy tale for children aged 3-10 based on: {text}
+Story must:
+1. Have animal/fairy characters
+2. Teach kindness or courage
+3. Use simple words
+4. Be 50-100 words
+Story:"""
+    
+    pipe = pipeline(
+        "text-generation", 
+        model="pranavpsv/genre-story-generator-v2",
+        # 优化生成参数
+        max_new_tokens=150,          # 严格控制输出长度
+        min_new_tokens=50,           # 确保最低字数
+        do_sample=True,
+        temperature=0.7,            # 平衡创意与连贯性
+        top_k=40,                   # 加速生成
+        top_p=0.9,
+        repetition_penalty=1.2,
+        num_return_sequences=1      # 减少计算量
+    )
+    
+    # 生成后处理
+    raw_story = pipe(prompt)[0]['generated_text']
+    
+    # 提取核心故事内容（过滤prompt重复）
+    story = raw_story.split("Story:")[-1].strip()
+    
+    # 精确截断至150字（中文按字符计算）
+    return ' '.join(story.split()[:150]) if len(story) > 150 else story
 
 # text2audio
 def text2audio(story_text):