Update app.py

app.py

@@ -1445,7 +1445,7 @@ def show_story():
         for idx, entry in enumerate(st.session_state.story):
             if entry['role'] == 'AI':
                 if entry.get('is_starter'):
-                    # Story Starter
+                    # Story Starter (คงเดิม)
                     st.markdown(f"""
                         <div style="
                             background-color: #f0f7ff; 
@@ -1466,8 +1466,50 @@ def show_story():
                             </div>
                         </div>
                     """, unsafe_allow_html=True)
+                elif entry.get('is_final'):  # เพิ่มเงื่อนไขสำหรับประโยคจบ
+                    # Final Ending Display
+                    st.markdown(f"""
+                        <div style="
+                            background-color: #e8f5e9;
+                            padding: 25px;
+                            border-radius: 10px;
+                            margin: 15px 0;
+                            border: 2px solid #4caf50;
+                            text-align: center;
+                            box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+                        ">
+                            <div style="font-size: 1.3em; color: #2e7d32; margin-bottom: 15px;">
+                                🎭 จบเรื่อง | The End
+                            </div>
+                            <div style="color: #333; font-size: 1.1em; font-style: italic;">
+                                {entry['content']}
+                            </div>
+                            <div style="margin-top: 15px; padding-top: 15px; border-top: 1px solid #a5d6a7;">
+                                <span style="color: #666;">
+                                    ✨ รูปแบบการจบ: {st.session_state.ending_type}
+                                </span>
+                            </div>
+                        </div>
+                    """, unsafe_allow_html=True)
                 else:
-                    # AI Response
+                    # Regular AI Response with ending mode indicator
+                    ending_badge = ""
+                    if st.session_state.get('ending_mode'):
+                        remaining = st.session_state.sentences_to_end
+                        ending_badge = f"""
+                            <div style="
+                                display: inline-block;
+                                background-color: #fff3e0;
+                                padding: 3px 8px;
+                                border-radius: 12px;
+                                font-size: 0.8em;
+                                color: #f57c00;
+                                margin-left: 10px;
+                            ">
+                                🎭 Ending: {remaining} sentences left
+                            </div>
+                        """
+                    
                     st.markdown(f"""
                         <div style="
                             background-color: #f8f9fa;
@@ -1476,18 +1518,36 @@ def show_story():
                             margin: 8px 0;
                             border-left: 4px solid #4caf50;
                         ">
-                            <div style="color: #2e7d32;">
-                                🤖 AI: {entry['content']}
+                            <div style="color: #2e7d32; display: flex; align-items: center;">
+                                <span>🤖 AI: {entry['content']}</span>
+                                {ending_badge}
                             </div>
                         </div>
                     """, unsafe_allow_html=True)
             
             elif entry['role'] == 'You':
-                # User Entry
+                # User Entry with ending mode indicator
                 status_icon = "✅" if entry.get('is_correct') else "✍️"
                 bg_color = "#e8f5e9" if entry.get('is_correct') else "#fff"
                 border_color = "#4caf50" if entry.get('is_correct') else "#1e88e5"
                 
+                ending_badge = ""
+                if st.session_state.get('ending_mode'):
+                    remaining = st.session_state.sentences_to_end
+                    ending_badge = f"""
+                        <div style="
+                            display: inline-block;
+                            background-color: #fff3e0;
+                            padding: 3px 8px;
+                            border-radius: 12px;
+                            font-size: 0.8em;
+                            color: #f57c00;
+                            margin-left: 10px;
+                        ">
+                            🎭 Ending: {remaining} sentences left
+                        </div>
+                    """
+                
                 st.markdown(f"""
                     <div style="
                         background-color: {bg_color};
@@ -1496,12 +1556,33 @@ def show_story():
                         margin: 8px 0;
                         border-left: 4px solid {border_color};
                     ">
-                        <div style="color: #333;">
-                            👤 You: {status_icon} {entry['content']}
+                        <div style="color: #333; display: flex; align-items: center;">
+                            <span>👤 You: {status_icon} {entry['content']}</span>
+                            {ending_badge}
                         </div>
                         {f'<div style="font-size: 0.9em; color: #666; margin-top: 5px;">{entry.get("feedback", "")}</div>' if entry.get('feedback') else ''}
                     </div>
                 """, unsafe_allow_html=True)
+        
+        # Show completion message if story is done
+        if st.session_state.get('story_completed'):
+            st.markdown("""
+                <div style="
+                    background-color: #f3e5f5;
+                    padding: 20px;
+                    border-radius: 10px;
+                    margin: 20px 0;
+                    text-align: center;
+                    border: 2px dashed #9c27b0;
+                ">
+                    <h2 style="color: #9c27b0; margin-bottom: 10px;">
+                        🎉 ยินดีด้วย! คุณเขียนเรื่องราวจบสมบูรณ์แล้ว
+                    </h2>
+                    <p style="color: #666;">
+                        คุณสามารถบันทึกเรื่องราวหรือเริ่มเรื่องใหม่ได้
+                    </p>
+                </div>
+            """, unsafe_allow_html=True)
 
 def show_story_progress():
     """Display story progress metrics"""
@@ -1552,32 +1633,146 @@ def show_story_ending_options():
 
 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()
+    try:
+        remaining = st.session_state.sentences_to_end
+        
+        # Show remaining sentences warning
+        st.warning(f"""
+            🎭 Ending Mode Active:
+            - Remaining sentences: {remaining}
+            - Ending Type: {st.session_state.ending_type}
+            
+            Try to work towards completing your story!
+        """)
+        
+        # Get specialized ending feedback
+        feedback_data = provide_feedback(text, st.session_state.level)
+        st.session_state.feedback = feedback_data
+        is_correct = not feedback_data.get('has_errors', False)
+        
+        # Add user's sentence
+        st.session_state.story.append({
+            "role": "You",
+            "content": text,
+            "is_corrected": False,
+            "is_correct": is_correct,
+            "timestamp": datetime.now().isoformat()
+        })
+        
+        # Update stats and achievements
+        words = set(text.lower().split())
+        st.session_state.stats['vocabulary_used'].update(words)
+        update_points(is_correct)
+        update_achievements()
+        
+        # Generate AI's ending continuation
+        if remaining > 1:  # Still have sentences remaining
+            ai_response = generate_ending_continuation(
+                text,
+                ending_type=st.session_state.ending_type,
+                remaining_sentences=remaining - 1
+            )
+            
+            st.session_state.story.append({
+                "role": "AI",
+                "content": ai_response,
+                "timestamp": datetime.now().isoformat()
+            })
+            
+            # Decrease remaining sentences count
+            st.session_state.sentences_to_end = remaining - 2  # -2 because both user and AI added sentences
+            
+        else:  # Last sentence - complete the story
+            final_response = generate_final_ending(
+                st.session_state.story,
+                st.session_state.ending_type
+            )
+            
+            st.session_state.story.append({
+                "role": "AI",
+                "content": final_response,
+                "timestamp": datetime.now().isoformat(),
+                "is_final": True
+            })
+            
+            # Mark story as complete
+            st.session_state.story_completed = True
+            complete_story()
+        
+        # Update session stats
+        update_session_stats()
+        
+        # Clear input and rerun
+        st.session_state.story_input_area = ""
+        st.rerun()
+        
+    except Exception as e:
+        logging.error(f"Error in ending mode: {str(e)}")
+        st.error("เกิดข้อผิดพลาดในโหมดจบเรื่อง กรุณาลองใหม่อีกครั้ง")
+
+def generate_final_ending(story: List[dict], ending_type: str) -> str:
+    """Generate the final ending sentence based on the story and chosen ending type"""
+    try:
+        # Create a comprehensive prompt for the final ending
+        story_summary = " ".join([entry['content'] for entry in story[-5:]])  # Last 5 sentences
+        
+        ending_prompts = {
+            "Happy Ending": """
+                Create a final happy ending sentence that:
+                - Brings joy and satisfaction
+                - Resolves the main story elements
+                - Ends on a positive note
+                Maximum 2 sentences.
+            """,
+            "Mysterious Ending": """
+                Create a final mysterious ending sentence that:
+                - Leaves an intriguing question
+                - Creates a sense of wonder
+                - Maintains some mystery
+                Maximum 2 sentences.
+            """,
+            "Lesson Learned": """
+                Create a final sentence that:
+                - Shows what was learned
+                - Provides a moral lesson
+                - Connects to the story's events
+                Maximum 2 sentences.
+            """,
+            "Surprise Ending": """
+                Create a final twist ending that:
+                - Provides an unexpected but logical conclusion
+                - Connects to previous story elements
+                - Creates a satisfying surprise
+                Maximum 2 sentences.
+            """
+        }
+
+        response = client.chat.completions.create(
+            model="gpt-4",
+            messages=[
+                {
+                    "role": "system",
+                    "content": f"""
+                    You are a master storyteller creating the final ending.
+                    {ending_prompts[ending_type]}
+                    This must be the absolute final sentence(s) of the story.
+                    Make it conclusive and satisfying.
+                    """
+                },
+                {
+                    "role": "user",
+                    "content": f"Story context:\n{story_summary}\n\nCreate the final ending:"
+                }
+            ],
+            max_tokens=100,
+            temperature=0.7
+        )
+        
+        return response.choices[0].message.content.strip()
+        
+    except Exception as e:
+        logging.error(f"Error generating final ending: {str(e)}")
+        return "And so, the story came to an end."
 
 def generate_ending_continuation(text: str, ending_type: str, remaining_sentences: int) -> str:
     """Generate AI continuation focusing on story conclusion"""
@@ -2162,9 +2357,14 @@ def show_main_interface():
         """, unsafe_allow_html=True)
         
         show_story()
-        
-        if st.session_state.story:
+
+        # Only show input if story is not completed
+        if st.session_state.story and not st.session_state.get('story_completed'):
             show_story_input()
+        
+        # Show completion options if story is done
+        if st.session_state.get('story_completed'):
+            show_completion_options()
     
     with col2:
         # Feedback section