update

app.py

@@ -88,27 +88,28 @@ def query_huggingface(payload):
 def generate_story_with_ai(emojis):
     """Generate story using AI"""
     emoji_text = ' '.join(emojis)
-    prompt = f"""Create a short story using these emojis: {emoji_text}
+    prompt = f"""Create a short story (100-150 words) using these emojis: {emoji_text}
 
 Instructions:
-1. Create a story that naturally incorporates all the given emojis
-2. The story should be fun and suitable for all ages
-3. Keep the story between 100-150 words
-4. The story must have a complete ending
-5. Follow this structure:
-   - Set up the scene and characters
-   - Create a small event or conflict
-   - Resolve the situation with a satisfying ending
-6. Make sure the last sentence provides closure to the story
-7. Do not use any section labels or markers in the story
-8. Write the story as one continuous narrative without sections
-9. Do not include phrases like 'Story continuation:' or 'Story ending:'
+1. Write a coherent story that naturally incorporates the given emojis
+2. The story must be suitable for all ages and have a clear structure:
+   - Beginning: Introduce the main character and setting
+   - Middle: Present a small challenge or interesting situation
+   - End: Resolve the situation with a satisfying conclusion
+3. Important rules:
+   - Write as one continuous narrative without any section markers
+   - Do not use labels like 'Story event:' or 'Story resolution:'
+   - Ensure the story has a proper ending (no cliffhangers)
+   - Keep sentences complete (no trailing thoughts)
+   - Maintain a consistent tone throughout
+4. Example flow (do not copy this exactly):
+   "Character encounters situation → faces challenge → resolves it → learns or achieves something"
 
-Story beginning:
+Begin the story with:
 Once upon a sunny day,"""
     
     try:
-        with st.spinner('创建故事中...'):
+        with st.spinner('Creating story...'):
             response = query_huggingface({"inputs": prompt})
             
             if response and isinstance(response, list) and len(response) > 0:
@@ -116,47 +117,47 @@ Once upon a sunny day,"""
                 story = story.replace(prompt, '').strip()
                 
                 if not story:
-                    st.error("故事生成失败，请重试")
+                    st.error("Failed to generate story. Please try again.")
                     return None
                 
-                # 清理所有可能的章节标记
+                # 清理所有可能的章节标记和故事标签
                 markers_to_remove = [
-                    'Story middle:', 
-                    'Story end:', 
-                    'Story summary:', 
-                    'Story continuation:', 
-                    'Story ending:',
-                    'Story begins:',
-                    'Story continues:',
-                    'Story concludes:',
-                    'Beginning:',
-                    'Middle:',
-                    'End:',
-                    'Continuation:',
-                    'Ending:'
+                    'Story event:', 'Story resolution:', 'Story middle:',
+                    'Story end:', 'Story summary:', 'Story continuation:',
+                    'Story ending:', 'Story begins:', 'Story continues:',
+                    'Story concludes:', 'Beginning:', 'Middle:', 'End:',
+                    'Continuation:', 'Ending:', 'Event:', 'Resolution:',
+                    'Finally:', 'In conclusion:', 'The end:', 'Summary:',
+                    'Next:', 'Then:', 'After that:', 'Eventually:'
                 ]
                 
+                # 移除所有标记
                 for marker in markers_to_remove:
                     story = story.replace(marker, '')
                 
-                # 清理多余的空行，保持段落格式
+                # 清理多余的空行和空格
                 story = '\n'.join(line for line in story.split('\n') if line.strip())
+                story = ' '.join(story.split())
                 
-                # 检查故事是否完整结束
-                if story.endswith(('and the', 'and', 'but', 'or', '...')):
-                    story = story.rsplit(' ', 1)[0] + '.'
+                # 检查并修复不完整的结尾
+                incomplete_endings = ('and', 'but', 'or', 'so', 'while', 'as', 'then', 'when', '...')
+                while story.endswith(incomplete_endings) or story.rstrip()[-1] not in '.!?':
+                    story = story.rsplit(' ', 1)[0].rstrip()
+                    if not story:
+                        break
                 
-                # 清理多余的空格
-                story = ' '.join(story.split())
+                # 确保故事有适当的结尾标点
+                if story and story[-1] not in '.!?':
+                    story += '.'
                 
-                final_story = f"Once upon a sunny day, {story}\n\n(使用的表情: {emoji_text})"
+                final_story = f"Once upon a sunny day, {story}\n\n(Emojis used: {emoji_text})"
                 return final_story
             
-            st.error("故事生成失败，请重试")
+            st.error("Failed to generate story. Please try again.")
             return None
             
     except Exception as e:
-        st.error(f"生成故事时出错: {str(e)}")
+        st.error(f"Error generating story: {str(e)}")
         return None
 
 def load_stories():
@@ -166,7 +167,7 @@ def load_stories():
             with open(DATA_FILE, 'r', encoding='utf-8') as f:
                 return json.load(f)
         except Exception as e:
-            st.error(f"加载数据时出错: {str(e)}")
+            st.error(f"Error loading data: {str(e)}")
             return []
     return []
 
@@ -176,7 +177,7 @@ def save_stories_to_file(stories):
         with open(DATA_FILE, 'w', encoding='utf-8') as f:
             json.dump(stories, f, ensure_ascii=False, indent=2)
     except Exception as e:
-        st.error(f"保存数据时出错: {str(e)}")
+        st.error(f"Error saving data: {str(e)}")
 
 # 初始化或加载故事数据
 if 'stories' not in st.session_state:
@@ -200,12 +201,23 @@ def main():
         st.session_state.selected_emojis = []
     
     # 创建选项卡布局
-    tabs = st.tabs(list(EMOJI_CATEGORIES.keys()))
+    ENGLISH_CATEGORIES = {
+        "Faces & Emotions": EMOJI_CATEGORIES["表情与情绪"],
+        "Animals": EMOJI_CATEGORIES["动物"],
+        "Plants": EMOJI_CATEGORIES["植物"],
+        "Food": EMOJI_CATEGORIES["食物"],
+        "Activities & Sports": EMOJI_CATEGORIES["活动与运动"],
+        "Transportation": EMOJI_CATEGORIES["交通工具"],
+        "Places & Buildings": EMOJI_CATEGORIES["地点与建筑"],
+        "Objects & Symbols": EMOJI_CATEGORIES["物品与符号"]
+    }
+    
+    tabs = st.tabs(list(ENGLISH_CATEGORIES.keys()))
     
     # 在每个选项卡中显示对应类别的emoji
-    for tab, (category, emojis) in zip(tabs, EMOJI_CATEGORIES.items()):
+    for tab, (category, emojis) in zip(tabs, ENGLISH_CATEGORIES.items()):
         with tab:
-            st.write(f"选择{category}:")
+            st.write(f"Select {category}:")
             # 将emoji列表分成多列显示
             cols = st.columns(8)  # 每行8个emoji
             for i, emoji in enumerate(emojis):
@@ -214,36 +226,36 @@ def main():
                         if len(st.session_state.selected_emojis) < 5:  # 限制最多选择5个emoji
                             st.session_state.selected_emojis.append(emoji)
                         else:
-                            st.warning("最多只能选择5个表情符号！")
+                            st.warning("Maximum 5 emojis allowed!")
     
     # 显示已选择的emoji
     if st.session_state.selected_emojis:
         st.write("---")
-        st.write("已选择的表情符号:", " ".join(st.session_state.selected_emojis))
+        st.write("Selected emojis:", " ".join(st.session_state.selected_emojis))
         
         # 添加清除选择按钮
         col1, col2 = st.columns([1, 4])
         with col1:
-            if st.button("清除选择"):
+            if st.button("Clear Selection"):
                 st.session_state.selected_emojis = []
                 st.experimental_rerun()
         
         # 生成故事按钮
         with col2:
-            if st.button("生成故事"):
+            if st.button("Generate Story"):
                 story = generate_story_with_ai(st.session_state.selected_emojis)
                 if story:
                     save_story(story)
-                    st.write("生成的故事:")
+                    st.write("Generated Story:")
                     st.write(story)
-                    st.success("故事已保存！")
+                    st.success("Story saved!")
     else:
-        st.write("请选择至少一个表情符号。")
+        st.write("Please select at least one emoji.")
     
     # 显示已保存的故事
     if st.session_state.stories:
         st.markdown("---")
-        st.header("已生成的故事")
+        st.header("Generated Stories")
         
         # 按点赞数排序故事
         sorted_stories = sorted(st.session_state.stories, 
@@ -254,12 +266,12 @@ def main():
         for idx, story_data in enumerate(sorted_stories):
             col1, col2 = st.columns([4, 1])
             with col1:
-                st.write(f"{idx + 1}. {story_data['story']} (点赞数: {story_data['votes']})")
+                st.write(f"{idx + 1}. {story_data['story']} (Likes: {story_data['votes']})")
             with col2:
                 if st.button(f"👍", key=f"vote_{idx}"):
                     story_data['votes'] += 1
                     update_votes()
-                    st.success(f"点赞成功！")
+                    st.success("Liked!")
                     st.experimental_rerun()
 
 if __name__ == "__main__":

stories_data.json

@@ -1,74 +1,14 @@
 [
   {
-    "story": "在充满 📚 的世界里，🏠 发现了 🍕 的意义。这让大家开始了一场 🌞 的庆祝！",
-    "votes": 2
-  },
-  {
-    "story": "🏠 和 🌞 是最好的朋友。有一天他们决定去 🍕，这是一次 📚 的经历！",
-    "votes": 2
-  },
-  {
-    "story": "从前有一天，🏠 和 🍕 一起去冒险。在路上他们遇到了 🌞，并且学会了 📚 的重要性。",
-    "votes": 3
-  },
-  {
-    "story": "从前有一天，🌞 和 📚 一起去冒险。在路上他们遇到了 🍕，并且学会了 🏠 的重要性。",
-    "votes": 4
-  },
-  {
-    "story": "从前有一天，🍕 和 🏠 一起去冒险。在路上他们遇到了 🌞，并且学会了 📚 的重要性。",
-    "votes": 0
-  },
-  {
-    "story": "从前有一天，📚 和 😎 一起去冒险。在路上他们遇到了 🐶，并且学会了 🎉 的重要性。",
-    "votes": 0
-  },
-  {
-    "story": "在充满 😎 的世界里，📚 发现了 🐶 的意义。这让大家开始了一场 🍕 的庆祝！",
-    "votes": 0
-  },
-  {
-    "story": "从前有一天，😎 和 🐶 一起去冒险。在路上他们遇到了 🍕，并且学会了 📚 的重要性。",
-    "votes": 0
-  },
-  {
-    "story": "从前有一天，🐶 和 📚 一起去冒险。在路上他们遇到了 😎，并且学会了 🎉 的重要性。",
-    "votes": 0
-  },
-  {
-    "story": "😎 和 🎉 是最好的朋友。有一天他们决定去 📚，这是一次 🐶 的经历！",
-    "votes": 0
-  },
-  {
-    "story": "生 成 一 个 有 趣 的 故 事 ， 包 含 这 些 表 情 :\n\n(使用的表情: 🌞 🏠 📚 🍕)",
-    "votes": 1
-  },
-  {
-    "story": "Once upon a sunny day, Max the golden retriever woke up with a wagging tail and a big smile on his face. He loved nothing more than going on adventures with his best friend, Lily, who was always ready for a fun-filled day. Max padded over to his favorite spot on the couch, where Lily was already curled up with her nose buried in a book. Max nudged her playfully with his nose, and Lily giggled as she closed her book and joined Max for breakfast.\n\nStory middle:\nAfter breakfast, Max and Lily decided to take a walk in the park. Max was excited to run around and sniff all the interesting smells, but Lily had other plans. She spotted a small library tucked away in the corner of the park and begged Max to come with her. Max was hesitant at first, but Lily promised him a special treat if he agreed. Max reluctantly followed Lily to the library, where they spent the next hour browsing through the books. Max was surprised to find that some of the books even had pictures of dogs! Lily picked out a few of her favorites, and Max settled down for a snooze on the floor.\n\nStory end\n\n(Emojis used: 😎 🐶 📚 🍕)",
-    "votes": 3
-  },
-  {
-    "story": "Once upon a sunny day, a boy named Max decided to spend his afternoon reading a new book he had just bought. Max's cozy house was filled with the scent of freshly baked cookies, which made him feel even more excited to start his reading session.\n\nStory middle:\nMax settled into his favorite armchair with a glass of iced tea and his new book. As he delved into the pages, he heard a faint knock on the door. Max's mom appeared, carrying a tray of freshly baked cookies. \"Max, I've just made some cookies. Would you like some?\" she asked. Max smiled, feeling grateful for his loving mother. \"Sure, mom. That would be great!\"\n\nStory end:\nAs Max enjoyed his delicious cookies, he heard a loud rumble outside. Max's mom went to the window to check, and she gasped. \"Max, come here! Look at the sky!\" Max rushed to the window, and he saw a magnificent rainbow arching over the horizon. Max felt so lucky to have experienced such a beautiful moment, surrounded by the comfort of his cozy home and the love of his family.\n\nStory summary:\n\n(Emojis used: 😎 🏠 📚)",
-    "votes": 0
-  },
-  {
-    "story": "Once upon a sunny day, a man named Max decided to take his beloved dog, Buddy, for a walk in the nearby park. Max grabbed Buddy's leash and his favorite book, \"101 Dog Tricks,\" and set off.\n\nStory middle:\nAs they strolled through the park, Buddy spotted a group of dogs playing fetch with their owners. Excitedly, Buddy pulled on his leash, trying to get Max's attention. Max noticed Buddy's eagerness and decided to join the game. Max threw a ball, and Buddy eagerly retrieved it, wagging his tail with delight.\n\nAs they continued playing, Max noticed a group of children reading books nearby. Max had an idea and called Buddy over. Max and the children took turns reading aloud to Buddy, who listened intently.\n\nStory end:\nAs the sun began to set, Max and Buddy headed home, exhausted but happy. Max realized that sharing his love of learning and reading with others, including his furry friend, was a joyful experience. From then on, Max made it a point to incorporate reading and learning into his walks with Buddy.\n\n(Emojis used: 😎 🐶 📚)",
-    "votes": 0
-  },
-  {
-    "story": "Once upon a sunny day, a happy pup named Baxter was lounging in his cozy doghouse. Suddenly, he heard a loud rumbling sound coming from the sky! 🐶🌈\nCurious and excited, Baxter ran out of his doghouse to see what was happening. To his amazement, he saw a rainbow-colored spaceship 🚀 land in his backyard! Baxter barked excitedly, and the spaceship's door opened. Out stepped a group of friendly aliens, who welcomed Baxter aboard their spaceship.\nThe aliens took Baxter on a thrilling tour of the universe, showing him colorful planets, sparkling stars, and swirling galaxies. Baxter had never seen anything like it! After their adventure, the aliens returned Baxter safely to his backyard. Baxter was overjoyed and couldn't stop wagging his tail. From that day on, Baxter was the happiest pup in the world, knowing that he had experienced something truly extraordinary. 🐶🌈🚀\n\n(Emojis used: 🌈 🐶 🚀)",
-    "votes": 0
-  },
-  {
-    "story": "Once upon a sunny day, a little girl named Lily was walking in the park. She saw a group of children playing with a lion cub. The cub was playful and wagged its tail as soon as it saw Lily. Lily was thrilled and decided to join them.\nLily approached the group and smiled at the lion cub. The cub looked at her curiously and licked her hand. The children laughed and played with the cub, making funny faces with the 😃 and 😋 emojis. Lily was having the time of her life, and the cub seemed to enjoy her company too. Suddenly, the cub started running around the park, and Lily chased it, giggling and shouting with excitement.\nAfter a while, the cub got tired and lay down in the shade, yawning. Lily sat beside it, stroking its fur gently. The cub closed its eyes and fell asleep, snoring softly. Lily smiled and watched the cub sleep, feeling content and happy. The children gathered around Lily and the cub, taking pictures and making memories. The park was filled with laughter and joy, thanks to the playful lion cub and the\n\n(Emojis used: 😃 😋 🦁)",
+    "story": "Once upon a sunny day, a group of friends gathered at the park to play a game of baseball. There was Max, the captain, with a big grin on his face. He loved baseball and was determined to win the game. Then there was Lily, the star player, who could hit the ball out of the park. Alex, the goalkeeper, who could catch any ball thrown at him. And lastly, there was Sophie, who was new to the group but eager to learn the game. Story event: As the game started, Max threw the ball to Lily, who hit it hard. The ball flew high and fast, and Sophie tried to catch it, but it bounced off her glove. Alex, quick as a flash, dived and caught the ball, saving the team from losing a point. The crowd cheered as Alex got up, and Lily smiled, impressed by his skills. Story resolution: But then, disaster struck. As Alex was about to throw the ball to Max, Lily accidentally stepped on his foot, causing him to lose his balance. The ball slipped out of his hand, and it rolled towards the fence. Max and Lily ran after it, but Sophie, who had been watching\n\n(Emojis used: 😄 ⚾ 🏣)",
     "votes": 0
   },
   {
-    "story": "Once upon a sunny day, Emma was sitting on a bench in the park, eating a juicy orange (🍊) while watching a baseball game (⚾) being played on the field. Suddenly, she heard a loud noise and saw a plane (✈️) flying overhead. She wondered what could be happening. \nStory continuation:\nEmma's curiosity got the best of her, and she decided to follow the plane to see where it was going. She packed her backpack and set off on her adventure. She walked for hours until she finally reached the airport (✈️). There, she saw the same plane from the park, and to her surprise, she found out that it was going on a trip around the world! Emma was thrilled and knew she had to join the journey. She asked the pilot if she could come along, and to her delight, he agreed. \nStory ending:\nEmma spent the next few months traveling the world, visiting new places, and making new friends. She even got to play baseball (⚾) with some locals in a few of the countries she visited. When the trip was over, Emma returned home, feeling grateful.\n\n(使用的表情: 😄 🍊 ⚾ ✈️)",
+    "story": "Once upon a sunny day, a little boy named Max was playing with his toy animals in the park. Suddenly, a fierce lion (🦁) appeared out of nowhere and snatched his favorite teddy bear (🐯). Max chased after the lion, but he was too fast. Max's heart sank as he saw the lion disappear into the forest with his beloved toy. Max was sad and hungry (😟), so he decided to follow the lion's paw prints to find his toy and a source of food (🥩). As Max ventured into the woods, he stumbled upon a pack of mischievous wolves (🐺). They saw Max and immediately started chasing him. Max was frightened, but he remembered his father's advice: \"Be brave and keep running.\" Max ran as fast as he could and finally found the lion's den. The lion was sleeping, and Max found his toy right next to him. Max quietly picked up his toy and started to run back home, but the wolves caught up with him.\n\n(Emojis used: 😟 🐯 🦁 🐺 🥩)",
     "votes": 0
   },
   {
-    "story": "Once upon a sunny day, a group of friends gathered at the park to play a game of baseball. There was Max, the captain, with a big grin on his face. He loved baseball and was determined to win the game. Then there was Lily, the star player, who could hit the ball out of the park. Alex, the goalkeeper, who could catch any ball thrown at him. And lastly, there was Sophie, who was new to the group but eager to learn the game. Story event: As the game started, Max threw the ball to Lily, who hit it hard. The ball flew high and fast, and Sophie tried to catch it, but it bounced off her glove. Alex, quick as a flash, dived and caught the ball, saving the team from losing a point. The crowd cheered as Alex got up, and Lily smiled, impressed by his skills. Story resolution: But then, disaster struck. As Alex was about to throw the ball to Max, Lily accidentally stepped on his foot, causing him to lose his balance. The ball slipped out of his hand, and it rolled towards the fence. Max and Lily ran after it, but Sophie, who had been watching\n\n(使用的表情: 😄 ⚾ 🏣)",
+    "story": "Once upon a sunny day, a boy named Max was walking home from school with a smile on his face. He was excited to grab a hot dog from his favorite food truck, 🌭. As he approached the truck, he saw a group of seagulls, 🐧, pecking at the leftovers. In the middle, Max hesitated. He didn't want to disturb the birds, but he also didn't want to miss his favorite meal. He thought for a moment and came up with an idea. Max approached the truck and asked the vendor to give him a hot dog wrapped in a paper cone, just like the ice cream cones. The vendor laughed and obliged. Max walked back to the seagulls, holding his cone high. The birds didn't seem to mind, and Max enjoyed his meal with a grin, 😚. In the end, Max learned that sometimes, a little creativity goes a long way. He also realized that sharing is caring, and it's always better to spread kindness, even to the smallest creatures. The end.\n\n(Emojis used: 😚 🐧 🌭)",
     "votes": 0
   }
 ]