Update app.py

app.py

@@ -2937,8 +2937,20 @@ def show_completion_options():
     try:
         st.balloons()
         
-        # Show completion message
-        st.markdown("""
+        # Show completion message with proper ending type display
+        ending_type_display = {
+            "Happy Ending": "จบแบบมีความสุข",
+            "Mysterious Ending": "จบแบบทิ้งท้ายให้คิดต่อ",
+            "Lesson Learned": "จบแบบได้ข้อคิด",
+            "Surprise Ending": "จบแบบพลิกความคาดหมาย"
+        }
+        
+        current_ending = ending_type_display.get(
+            st.session_state.ending_type,
+            st.session_state.ending_type
+        )
+        
+        st.markdown(f"""
             <div style="
                 background-color: #e8f5e9;
                 padding: 20px;
@@ -2951,7 +2963,7 @@ def show_completion_options():
                     🎉 ยินดีด้วย! คุณเขียนเรื่องราวจบสมบูรณ์แล้ว
                 </h2>
                 <p style="color: #1b5e20;">
-                    เรื่องราวของคุณจบลงด้วย: {st.session_state.ending_type}
+                    เรื่องราวของคุณจบลงด้วย: {current_ending}
                 </p>
             </div>
         """, unsafe_allow_html=True)
@@ -2967,11 +2979,14 @@ def show_completion_options():
                     use_container_width=True):
             with st.spinner("กำลังเรียบเรียงเรื่องราว..."):
                 try:
-                    stitched_story = stitch_story(
-                        st.session_state.story,
-                        st.session_state.current_theme,
-                        st.session_state.level
+                    stitched_story = generate_stitched_story(
+                        story=st.session_state.story,
+                        style="Classic Fairytale",  # Default style
+                        detail_level="ปานกลาง",    # Default detail level
+                        theme=st.session_state.current_theme,
+                        level=st.session_state.level
                     )
+                    st.session_state.stitched_story = stitched_story
                     show_stitched_story(stitched_story)
                 except Exception as e:
                     logging.error(f"Error in story stitching: {str(e)}")