update

.gitignore

@@ -1,2 +1,6 @@
-
-/.venv
+.env
+stories_data.json
+__pycache__/
+.pytest_cache/
+.venv/
+/venv

app.py

@@ -2,12 +2,97 @@ import streamlit as st
 import random
 import json
 import os
+from dotenv import load_dotenv
+import requests
+
+# 加载环境变量
+load_dotenv()
 
 # 定义表情列表
 EMOJI_LIST = ["😀", "😎", "🌞", "🌈", "🐶", "🏠", "🚀", "📚", "🎉", "🍕", "🎸", "🏆"]
 
 # 定义数据文件路径
 DATA_FILE = "stories_data.json"
+HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
+API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
+
+def query_huggingface(payload):
+    """调用Hugging Face API"""
+    headers = {
+        "Authorization": f"Bearer {HUGGINGFACE_API_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    
+    # 调整payload格式为Zephyr模型的要求
+    simplified_payload = {
+        "inputs": payload["inputs"],
+        "parameters": {
+            "max_new_tokens": 250,
+            "temperature": 0.7,
+            "top_p": 0.9,
+            "do_sample": True,
+            "return_full_text": False
+        }
+    }
+    
+    try:
+        st.write("Calling API...")
+        response = requests.post(API_URL, headers=headers, json=simplified_payload, timeout=60)
+        
+        if response.status_code != 200:
+            st.error(f"API call failed, status code: {response.status_code}")
+            st.write(f"Error message: {response.text}")
+            return None
+            
+        result = response.json()
+        st.write("API response:", result)
+        return result
+            
+    except Exception as e:
+        st.error(f"API request error: {str(e)}")
+        return None
+
+def generate_story_with_ai(emojis):
+    """Generate story using AI"""
+    emoji_text = ' '.join(emojis)
+    prompt = f"""Create a fun and engaging short story 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. Include a clear beginning, middle, and end
+4. Keep it concise (around 100-150 words)
+5. Make it creative and engaging
+
+Story beginning:
+Once upon a sunny day,"""
+    
+    try:
+        with st.spinner('Creating story...'):
+            response = query_huggingface({"inputs": prompt})
+            
+            if response and isinstance(response, list) and len(response) > 0:
+                # Get generated text
+                story = response[0].get('generated_text', '').strip()
+                
+                # Clean up story text
+                story = story.replace(prompt, '').strip()
+                
+                # Check if story is empty
+                if not story:
+                    st.error("Generated story is empty, please try again")
+                    return None
+                
+                # Format final story
+                final_story = f"Once upon a sunny day, {story}\n\n(Emojis used: {emoji_text})"
+                return final_story
+            
+            st.error("Failed to generate story, please try again")
+            return None
+            
+    except Exception as e:
+        st.error(f"Error generating story: {str(e)}")
+        return None
 
 def load_stories():
     """从文件加载故事数据"""
@@ -32,16 +117,6 @@ def save_stories_to_file(stories):
 if 'stories' not in st.session_state:
     st.session_state.stories = load_stories()
 
-def generate_simple_story(emojis):
-    templates = [
-        "从前有一天，{} 和 {} 一起去冒险。在路上他们遇到了 {}，并且学会了 {} 的重要性。",
-        "{} 和 {} 是最好的朋友。有一天他们决定去 {}，这是一次 {} 的经历！",
-        "在充满 {} 的世界里，{} 发现了 {} 的意义。这让大家开始了一场 {} 的庆祝！"
-    ]
-    story_template = random.choice(templates)
-    story_emojis = random.sample(emojis, 4) if len(emojis) >= 4 else emojis + random.sample(EMOJI_LIST, 4 - len(emojis))
-    return story_template.format(*story_emojis)
-
 def save_story(story):
     """保存新故事并更新文件"""
     st.session_state.stories.append({"story": story, "votes": 0})
@@ -55,43 +130,43 @@ def main():
     st.set_page_config(page_title="Emoji Story Generator", page_icon="📚")
     st.title("Emoji Story Generator")
     
-    # 添加表情选择器
-    selected_emojis = st.multiselect("选择表情符号", EMOJI_LIST)
+    # Add emoji selector
+    selected_emojis = st.multiselect("Choose emojis for your story", EMOJI_LIST)
     
     if selected_emojis:
-        st.write("你选择了:", " ".join(selected_emojis))
+        st.write("Selected emojis:", " ".join(selected_emojis))
         
-        # 添加生成故事按钮
-        if st.button("生成故事"):
-            story = generate_simple_story(selected_emojis)
-            save_story(story)
-            st.write("生成的故事:")
-            st.write(story)
-            st.success("故事已保存!")
+        # Add generate story button
+        if st.button("Generate Story"):
+            story = generate_story_with_ai(selected_emojis)
+            if story:  # Only save if story generation was successful
+                save_story(story)
+                st.write("Generated Story:")
+                st.write(story)
+                st.success("Story saved!")
             
-        # 显示已保存的故事
+        # Display saved stories
         if st.session_state.stories:
-            st.header("已生成的故事")
+            st.header("Generated Stories")
             
-            # 按照点赞数排序显示故事
+            # Sort stories by votes
             sorted_stories = sorted(st.session_state.stories, 
                                  key=lambda x: x['votes'], 
                                  reverse=True)
             
-            # 使用列布局来优化显示效果
+            # Use columns for layout
             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"点赞成功！")
+                        update_votes()
+                        st.success(f"Liked!")
                         st.experimental_rerun()
     else:
-        st.write("请至少选择一个表情符号。")
+        st.write("Please select at least one emoji.")
 
 if __name__ == "__main__":
     main()

requirements.txt

@@ -0,0 +1,3 @@
+streamlit==1.22.0
+python-dotenv==0.19.2
+requests==2.28.1

stories_data.json

@@ -18,5 +18,33 @@
   {
     "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
   }
 ]