Update app.py

app.py

@@ -14,21 +14,52 @@ def text2story(text):
     pipe = pipeline("text-generation", 
                     model="pranavpsv/genre-story-generator-v2",
                     max_new_tokens=100,
-                    truncation=True)     
-    story_text = pipe(text, 
-                      do_sample=True,     
-                      temperature=0.9,    
-                      top_k=50,           
+                    truncation=True)    
+    prompt_for_children = f"""Generate a fairy tale for children aged 3-10 based on: {text}.
+    Requirements:
+    1. Simple and interesting plot
+    2. Include magical elements or talking animals
+    3. Educational message
+    4. Use simple vocabulary (under 500 Lexile)
+    5. Short sentences (max 15 words)
+    6. Add onomatopoeia like 'Boom!', 'Whoosh!' etc.
+    
+    Story: Once upon a time,"""
+    
+    story_text = pipe(prompt_for_children,
+                      text, 
+                      max_new_tokens=200,  # 增加故事长度
+                      temperature=0.7,     # 降低随机性
+                      top_p=0.95,          # 使用核采样   
+                      repetition_penalty=1.2,
+                      do_sample=True,
                       num_return_sequences=1)[0]['generated_text']
     last_punctuation = max(story_text.rfind("."), story_text.rfind("!"), story_text.rfind("?"))
     if last_punctuation != -1:
         story_text = story_text[:last_punctuation+1]
     return story_text
 
+    # Post-processing optimization
+    story_text = story_text.replace(prompt, "").strip()
+    for stop_word in ["\nThe end", "The end", "End of story"]:
+        if story_text.endswith(stop_word):
+            story_text = story_text[:-len(stop_word)]
+    last_punct = max(story_text.rfind("."), story_text.rfind("!"), story_text.rfind("?"))
+    return story_text[:last_punct+1] if last_punct != -1 else story_text
+
 # text2audio
 def text2audio(story_text):
-    pipe = pipeline("text-to-audio", model="Matthijs/mms-tts-eng")
-    audio_data = pipe(story_text)
+    pipe = pipeline("text-to-audio", model="facebook/mms-tts-eng",
+                   config = {"speaker": "en_female"})
+    audio_data = pipe(
+        story_text,
+        generate_kwargs={
+            "tempo": 1.1,    # 稍快的语速（1.0为基准）
+            "pitch": 4,       # 提高音调（0-10范围）
+            "energy": 1.2,   # 增强表现力
+            "vocal_tract_length": 1.15  # 更明亮的音色
+        }
+    )
     return audio_data