Update app.py

app.py

@@ -2,32 +2,6 @@
 import streamlit as st
 from transformers import pipeline
 
-# 必须作为第一个Streamlit命令出现
-st.set_page_config(page_title="Magic Story Box", page_icon="🧚")
-
-# 修改CSS部分（其他代码保持不变）
-st.markdown("""
-<style>
-@import url('https://fonts.googleapis.com/css2?family=Comic+Neue:wght@700&display=swap');
-.header {
-    background-image: url('https://huggingface.co/spaces/Leo0129/classagm/resolve/main/background.jpg');
-    background-size: cover;
-    border-radius: 15px;
-    padding: 2rem;
-    text-align: center;
-    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-}
-.header h1 {
-    color: #FF6B6B;
-    font-family: 'Comic Neue', cursive;
-    font-size: 2.5rem;
-    text-shadow: 2px 2px #FFF;
-    margin-bottom: 0.5rem !important;  # 新增
-}
-
-</style>
-""", unsafe_allow_html=True)
-
 # function part
 # img2text
 def img2text(url):
@@ -49,6 +23,31 @@ def text2audio(story_text):
 
 
 def main():
+    st.set_page_config(page_title="Magic Story Box", page_icon="🧚")
+    
+    # 修改CSS部分（其他代码保持不变）
+    st.markdown("""
+    <style>
+    @import url('https://fonts.googleapis.com/css2?family=Comic+Neue:wght@700&display=swap');
+    .header {
+        background-image: url('https://huggingface.co/spaces/Leo0129/classagm/resolve/main/background.jpg');
+        background-size: cover;
+        border-radius: 15px;
+        padding: 2rem;
+        text-align: center;
+        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+    }
+    .header h1 {
+        color: #FF6B6B;
+        font-family: 'Comic Neue', cursive;
+        font-size: 2.5rem;
+        text-shadow: 2px 2px #FFF;
+        margin-bottom: 0.5rem !important;  # 新增
+    }
+    
+    </style>
+    """, unsafe_allow_html=True)  
+
     # 新版标题区
     st.markdown("""
     <div class="header">