Update app_work.py

app_work.py

@@ -1,4 +1,4 @@
-# === 1. IMPORTS & SETUP ===
+# === 1. IMPORTS & SETUP === 
 import streamlit as st
 import json
 import datetime
@@ -1503,6 +1503,192 @@ def show_story():
                     </div>
                 """, unsafe_allow_html=True)
 
+def show_story_progress():
+    """Display story progress metrics"""
+    if st.session_state.story:
+        total_sentences = len(st.session_state.story)
+        st.markdown(f"""
+            <div style="
+                background-color: #f0f7ff;
+                padding: 15px;
+                border-radius: 10px;
+                margin: 10px 0;
+            ">
+                <div style="margin-bottom: 10px;">
+                    📊 ความยาวเรื่อง: {total_sentences} ประโยค
+                </div>
+                <div class="progress-bar">
+                    <div class="progress-bar-fill" style="width: {min(total_sentences * 10, 100)}%;"></div>
+                </div>
+                <div style="font-size: 0.9em; color: #666; margin-top: 5px;">
+                    เรื่องควรยาว 10-20 ประโยค เพื่อความสมบูรณ์
+                </div>
+            </div>
+        """, unsafe_allow_html=True)
+
+def show_story_ending_options():
+    """Display story ending options and guidance"""
+    if len(st.session_state.story) >= 10:  # แสดงตัวเลือกเมื่อเรื่องยาวพอ
+        st.markdown("### 🎭 ต้องการจบเรื่องหรือไม่?")
+        
+        # แสดงตัวเลือกวิธีจบเรื่อง
+        ending_type = st.radio(
+            "เลือกวิธีจบเรื่อง:",
+            options=[
+                "Happy Ending - จบแบบมีความสุข",
+                "Mysterious Ending - จบแบบทิ้งท้ายให้คิดต่อ",
+                "Lesson Learned - จบแบบได้ข้อคิด",
+                "Surprise Ending - จบแบบพลิกความคาดหมาย"
+            ],
+            index=0,
+            help="เลือกรูปแบบการจบเรื่องที่คุณต้องการ"
+        )
+        
+        if st.button("🎬 เริ่มจบเรื่อง", use_container_width=True):
+            st.session_state.ending_mode = True
+            st.session_state.ending_type = ending_type
+            st.session_state.sentences_to_end = 5
+            st.rerun()
+
+def handle_ending_mode(text: str):
+    """Handle story submission during ending mode"""
+    remaining = st.session_state.sentences_to_end
+    
+    # แสดงการแจ้งเตือนจำนวนประโยคที่เหลือ
+    st.info(f"🎯 เหลืออีก {remaining} ประโยคในการจบเรื่อง")
+    
+    # สร้าง prompt พิเศษสำหรับ AI ในโหมดจบเรื่อง
+    ending_prompts = {
+        "Happy Ending": "Work towards a positive and uplifting conclusion",
+        "Mysterious Ending": "Create an intriguing open-ended conclusion",
+        "Lesson Learned": "Incorporate a meaningful life lesson",
+        "Surprise Ending": "Build up to an unexpected twist"
+    }
+    
+    # ปรับ AI response ให้พยายามจบเรื่อง
+    modified_continuation = generate_ending_continuation(
+        text,
+        ending_type=st.session_state.ending_type,
+        remaining_sentences=remaining
+    )
+    
+    # อัพเดทจำนวนประโยคที่เหลือ
+    st.session_state.sentences_to_end -= 1
+    
+    # ตรวจสอบว่าถึงจบเรื่องหรือยัง
+    if st.session_state.sentences_to_end <= 0:
+        complete_story()
+
+def generate_ending_continuation(text: str, ending_type: str, remaining_sentences: int) -> str:
+    """Generate AI continuation focusing on story conclusion"""
+    try:
+        ending_prompts = {
+            "Happy Ending": """
+                Role: Story concluder aiming for a happy ending
+                Goal: Create a satisfying, positive conclusion
+                Rules:
+                - Build towards joy, success, or resolution
+                - Use uplifting and positive language
+                - Connect to previous story elements
+                """,
+            "Mysterious Ending": """
+                Role: Mystery writer creating intrigue
+                Goal: Leave readers thinking and wondering
+                Rules:
+                - Add subtle hints and clues
+                - Create atmospheric descriptions
+                - Leave some questions unanswered
+                """,
+            "Lesson Learned": """
+                Role: Moral story concluder
+                Goal: Incorporate meaningful life lessons
+                Rules:
+                - Connect actions to consequences
+                - Show character growth
+                - Express the moral naturally
+                """,
+            "Surprise Ending": """
+                Role: Plot twist creator
+                Goal: Deliver unexpected but satisfying conclusion
+                Rules:
+                - Plant subtle hints earlier
+                - Subvert expectations logically
+                - Maintain story coherence
+                """
+        }
+
+        response = client.chat.completions.create(
+            model="gpt-4",
+            messages=[
+                {
+                    "role": "system",
+                    "content": f"""
+                    {ending_prompts[ending_type]}
+                    
+                    Additional Rules:
+                    - You have {remaining_sentences} sentences to conclude
+                    - Each response should be max 2 sentences
+                    - Build towards the finale naturally
+                    - Connect to previous story elements
+                    """
+                },
+                {
+                    "role": "user",
+                    "content": f"Continue and work towards ending this story: {text}"
+                }
+            ],
+            max_tokens=100,
+            temperature=0.7
+        )
+        
+        return response.choices[0].message.content.strip()
+        
+    except Exception as e:
+        logging.error(f"Error generating ending: {str(e)}")
+        return "The story moved towards its conclusion..."
+
+def complete_story():
+    """Handle story completion and celebration"""
+    st.balloons()  # แสดงเอฟเฟคฉลอง
+    
+    # สร้าง Story Summary
+    story_summary = generate_story_summary(st.session_state.story)
+    
+    # แสดงหน้าจบเรื่อง
+    st.markdown(f"""
+        <div style="
+            background-color: #e8f5e9;
+            padding: 20px;
+            border-radius: 10px;
+            text-align: center;
+            margin: 20px 0;
+        ">
+            <h2>🎉 ยินดีด้วย! คุณเขียนเรื่องราวจบสมบูรณ์แล้ว</h2>
+            
+            <div style="margin: 20px 0;">
+                <h3>📝 สรุปเรื่องราว</h3>
+                <p>{story_summary}</p>
+            </div>
+            
+            <div style="margin: 20px 0;">
+                <h3>🏆 ความสำเร็จ</h3>
+                <p>จำนวนประโยค: {len(st.session_state.story)}</p>
+                <p>คำศัพท์ที่ใช้: {len(st.session_state.stats['vocabulary_used'])}</p>
+                <p>ความแม่นยำ: {st.session_state.stats['accuracy_rate']:.1f}%</p>
+            </div>
+        </div>
+    """, unsafe_allow_html=True)
+    
+    # แสดงตัวเลือกหลังจบเรื่อง
+    col1, col2 = st.columns(2)
+    with col1:
+        if st.button("💾 บันทึกเรื่องราว", use_container_width=True):
+            save_completed_story()
+    with col2:
+        if st.button("🔄 เริ่มเรื่องใหม่", use_container_width=True):
+            reset_story()
+            st.rerun()
+
 def show_feedback_section():
     """Display writing feedback section"""
     if not st.session_state.feedback:
@@ -1996,12 +2182,19 @@ def show_main_interface():
 
 # === 6. MAIN APPLICATION LOGIC ===
 def main():
-    """Main application entry point"""
     try:
         # Initialize states
         init_session_state()
         init_theme_state()
         
+        # Initialize ending system state if not exists
+        if 'ending_mode' not in st.session_state:
+            st.session_state.ending_mode = False
+        if 'sentences_to_end' not in st.session_state:
+            st.session_state.sentences_to_end = 5
+        if 'ending_type' not in st.session_state:
+            st.session_state.ending_type = None
+        
         # Add watermark
         st.markdown("""
             <div style='position: fixed; bottom: 10px; right: 10px; z-index: 1000; 
@@ -2028,11 +2221,37 @@ def main():
             if not st.session_state.current_theme:
                 show_theme_selection()
             else:
+                # Show story progress if story exists
+                if st.session_state.story:
+                    show_story_progress()
+                    
+                    # Show ending options if story is long enough and not in ending mode
+                    if (len(st.session_state.story) >= 10 and 
+                        not st.session_state.ending_mode):
+                        show_story_ending_options()
+                    
+                    # Show ending mode warning if active
+                    if st.session_state.ending_mode:
+                        st.warning(f"""
+                            🎭 กำลังอยู่ในโหมดจบเรื่อง - เห���ือ {st.session_state.sentences_to_end} ประโยค
+                            \nรูปแบบการจบ: {st.session_state.ending_type}
+                        """)
+                
+                # Show main interface
                 show_main_interface()
+                
+                # Check if story is complete
+                if (st.session_state.ending_mode and 
+                    st.session_state.sentences_to_end <= 0):
+                    complete_story()
 
         # Handle reset if needed
         if st.session_state.should_reset:
             reset_story()
+            # Reset ending mode states
+            st.session_state.ending_mode = False
+            st.session_state.sentences_to_end = 5
+            st.session_state.ending_type = None
             
         # Auto-save progress periodically
         if st.session_state.story: