Update app.py

app.py

@@ -51,7 +51,7 @@ def text2story(text):
         generator = pipeline("text-generation", model="2173ars/llama-3-8b-Instruct-bnb-4bit-personal-shortstory")
         
         # Create a prompt for the story
-        prompt = f"Write a short children's story about this scene: {text}\n\nStory: "
+        prompt = f"{text}"
         
         # Generate the story - limit to a smaller max_length due to model size
         story = generator(prompt,
@@ -137,6 +137,8 @@ async def text2audio_english(text):
 # Apply custom CSS for modern, stylish kid-friendly UI
 st.set_page_config(page_title="故事魔法", page_icon="✨", layout="wide")
 
+
+
 st.markdown("""
 <style>
   /* Modern, stylish kid-friendly design */
@@ -341,10 +343,8 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # App header with Cantonese
-st.title("✨ 故事魔法")
-st.markdown("<p class='subtitle'>上載一張圖片，睇下佢點變成一個神奇嘅故事！</p>",
-            unsafe_allow_html=True)
-
+st.title("")
+st.markdown("<div class='welcome-message'><div class='welcome-icon'>✨</div><h2>歡迎嚟到故事魔法師！</h2> <p style='font-size: 1.2rem; color: #6B7897; max-width: 500px; margin: 0 auto 30px;'>上載一張你鍾意嘅相片，我哋嘅魔法師會幫你變出一個好好玩嘅故事！</p><div><span class='emoji'>🚀</span><span class='emoji'>🦄</span> <span class='emoji'>🔮</span><span class='emoji'>🌈</span></div></div>",unsafe_allow_html=True)
 # Add a progress indicator for model loading
 progress_placeholder = st.empty()
 
@@ -448,9 +448,7 @@ if uploaded_file is not None:
                 if st.session_state.audio_path_en and os.path.exists(st.session_state.audio_path_en):
                     with open(st.session_state.audio_path_en, "rb") as audio_file:
                         audio_bytes = audio_file.read()
-                    st.markdown("<div class='audio-container'>", unsafe_allow_html=True)
                     st.audio(audio_bytes, format="audio/mp3")
-                    st.markdown("</div>", unsafe_allow_html=True)
                 else:
                     st.error("Sorry! Please try again.")
 
@@ -468,9 +466,7 @@ if uploaded_file is not None:
                 if st.session_state.audio_path_zh and os.path.exists(st.session_state.audio_path_zh):
                     with open(st.session_state.audio_path_zh, "rb") as audio_file:
                         audio_bytes = audio_file.read()
-                    st.markdown("<div class='audio-container'>", unsafe_allow_html=True)
                     st.audio(audio_bytes, format="audio/mp3")
-                    st.markdown("</div>", unsafe_allow_html=True)
                 else:
                     st.error("哎呀！再試多次啦！")
 
@@ -500,20 +496,4 @@ else:
     st.session_state.audio_path_zh = None
     st.session_state.audio_generated_en = False
     st.session_state.audio_path_en = None
-    
-    # Welcome message in Cantonese
-    st.markdown("""
-    <div class="welcome-message">
-        <div class="welcome-icon">✨</div>
-        <h2>歡迎嚟到故事魔法！</h2>
-        <p style="font-size: 1.2rem; color: #6B7897; max-width: 500px; margin: 0 auto 30px;">
-            上載一張你鍾意嘅相片，我哋嘅魔法師會幫你變出一個好好玩嘅故事！
-        </p>
-        <div>
-            <span class="emoji">🚀</span>
-            <span class="emoji">🦄</span>
-            <span class="emoji">🔮</span>
-            <span class="emoji">🌈</span>
-        </div>
-    </div>
-    """, unsafe_allow_html=True)
+