Update appback.py3

appback.py3

@@ -12,6 +12,7 @@ from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
 from datetime import datetime 
 import random
+from sounds import AudioManager, get_sound_commands
 
 # === 2. CONFIGURATIONS ===
 # Theme Configuration
@@ -393,6 +394,11 @@ achievements_list = {
 # Initial CSS Setup
 st.markdown("""
     <style>
+    /* Hide audio elements */
+    .hide-tag {
+        display: none !important;
+    }
+    
     /* Base Styles */
     @import url('https://fonts.googleapis.com/css2?family=Sarabun:wght@400;600&display=swap');
     
@@ -566,15 +572,26 @@ def init_theme_state():
         st.session_state.theme_story_starter = None
 
 def reset_story():
-    """Reset story and related state variables"""
+    """Reset story and related state variables completely"""
     try:
-        # Reset story-related states
-        st.session_state.story = []
-        st.session_state.feedback = None
-        st.session_state.theme_story_starter = None
+        # Reset all story-related states
+        new_states = {
+            'story': [],
+            'feedback': None,
+            'theme_story_starter': None,
+            'text_input': "",
+            'current_theme': None,
+            'ending_mode': False,
+            'sentences_to_end': 0,
+            'ending_type': None,
+            'story_completed': False,
+            'stitched_story': None,
+            'clear_input': True,
+            'last_submission': None
+        }
         
         # Reset points
-        st.session_state.points = {
+        new_points = {
             'total': 0,
             'perfect_sentences': 0,
             'corrections_made': 0,
@@ -583,7 +600,7 @@ def reset_story():
         }
         
         # Reset stats
-        st.session_state.stats = {
+        new_stats = {
             'total_sentences': 0,
             'correct_first_try': 0,
             'accuracy_rate': 0.0,
@@ -594,12 +611,27 @@ def reset_story():
             'session_duration': 0
         }
         
-        # Reset progress
+        # Reset all session states
+        for key, value in new_states.items():
+            st.session_state[key] = value
+        st.session_state.points = new_points
+        st.session_state.stats = new_stats
+        
+        # Reset achievements and progress
+        st.session_state.achievements = []
         st.session_state.current_milestone = 0
         st.session_state.next_milestone = 5
         
-        # Reset flag
-        st.session_state.should_reset = False
+        # Reset UI elements
+        if 'story_input_area' in st.session_state:
+            del st.session_state.story_input_area
+        
+        # Force new input field creation
+        st.session_state.clear_count = st.session_state.get('clear_count', 0) + 1
+        
+        # Reset theme selection
+        st.session_state.theme_button_counter = 0
+        st.session_state.theme_selection_id = datetime.now().strftime('%Y%m%d%H%M%S')
         
         # Log reset
         logging.info("Story state reset successfully")
@@ -706,6 +738,46 @@ def update_session_stats():
         return False
 
 # === 4. UTILITY FUNCTIONS ===
+
+def initialize_audio():
+    """Initialize audio system"""
+    if 'audio_manager' not in st.session_state:
+        audio_manager = AudioManager()
+        st.session_state.audio_manager = audio_manager
+        st.session_state.sound_commands = get_sound_commands()
+        
+        # Add audio elements to the page using st.markdown with hide-tag class
+        st.markdown("""
+            <style>
+            .hide-tag {
+                display: none;
+            }
+            </style>
+        """, unsafe_allow_html=True)
+        
+        st.markdown(
+            f"""
+            <div class="hide-tag">
+                {audio_manager.get_audio_html()}
+            </div>
+            """,
+            unsafe_allow_html=True
+        )
+
+def show_audio_controls():
+    """Show audio control in sidebar"""
+    st.sidebar.markdown("### 🔊 ตั้งค่าเสียง")
+    
+    # Background music toggle
+    if st.sidebar.checkbox("เปิดเพลงประกอบ", value=True, key='bgm_enabled'):
+        st.markdown(f"<script>{st.session_state.sound_commands['bgm_play']}</script>", unsafe_allow_html=True)
+    else:
+        st.markdown(f"<script>{st.session_state.sound_commands['bgm_pause']}</script>", unsafe_allow_html=True)
+    
+    # Volume control
+    volume = st.sidebar.slider("ระดับเสียง", 0, 100, 50, key='volume') / 100
+    st.markdown(f"<script>setVolume({volume});</script>", unsafe_allow_html=True)
+    
 def generate_story_continuation(user_input: str, level: str) -> str:
     """Generate AI story continuation using ChatGPT"""
     
@@ -940,22 +1012,26 @@ def update_achievements():
             "🌟 นักเขียนไร้ที่ติ" not in current_achievements):
             current_achievements.append("🌟 นักเขียนไร้ที่ติ")
             st.success("🎉 ได้รับความสำเร็จใหม่: นักเขียนไร้ที่ติ!")
+            st.markdown(f"<script>{st.session_state.sound_commands['achievement']}</script>", unsafe_allow_html=True)
         
         if (len(st.session_state.stats['vocabulary_used']) >= 50 and 
             "📚 ราชาคำศัพท์" not in current_achievements):
             current_achievements.append("📚 ราชาคำศัพท์")
             st.success("🎉 ได้รับความสำเร็จใหม่: ราชาคำศัพท์!")
+            st.markdown(f"<script>{st.session_state.sound_commands['achievement']}</script>", unsafe_allow_html=True)
         
         if (len(st.session_state.story) >= 10 and 
             "📖 นักแต่งนิทาน" not in current_achievements):
             current_achievements.append("📖 นักแต่งนิทาน")
             st.success("🎉 ได้รับความสำเร็จใหม่: นักแต่งนิทาน!")
+            st.markdown(f"<script>{st.session_state.sound_commands['achievement']}</script>", unsafe_allow_html=True)
         
         if (st.session_state.stats['total_sentences'] >= 10 and 
             st.session_state.stats['accuracy_rate'] >= 80 and 
             "👑 ราชาความแม่นยำ" not in current_achievements):
             current_achievements.append("👑 ราชาความแม่นยำ")
             st.success("🎉 ได้รับความสำเร็จใหม่: ราชาความแม่นยำ!")
+            st.markdown(f"<script>{st.session_state.sound_commands['achievement']}</script>", unsafe_allow_html=True)
         
         # บันทึกความสำเร็จ
         st.session_state.achievements = current_achievements
@@ -1369,6 +1445,10 @@ def handle_theme_selection(theme: dict):
     """Handle theme selection and initialization"""
     try:
         with st.spinner("กำลังเตรียมเรื่องราว..."):
+            # Reset story completely first
+            reset_story()
+            
+            # Then set new theme
             st.session_state.current_theme = theme['id']
             starter = generate_dynamic_story_starter(
                 theme['id'],
@@ -1450,6 +1530,12 @@ def show_story():
     story_display = st.container()
     
     with story_display:
+        # Clear previous story if resetting
+        if st.session_state.get('should_reset'):
+            reset_story()
+            st.rerun()
+            return
+            
         if not st.session_state.story:
             st.info("เลือกธีมเรื่องราวที่ต้องการเพื่อเริ่มต้นการผจญภัย!")
             return
@@ -2312,45 +2398,52 @@ def generate_ending_continuation(text: str, ending_type: str, remaining_sentence
 
 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>
+    try:
+        # เล่นเสียงจบเรื่อง
+        st.markdown(f"<script>{st.session_state.sound_commands['complete']}</script>", unsafe_allow_html=True)
+        st.balloons()
+
+        # Generate 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>
-        </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()
+        """, 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()
+
+    except Exception as e:
+        logging.error(f"Error in complete_story: {str(e)}")
+        st.error("เกิดข้อผิดพลาดในการแสดงผลการจบเรื่อง")
 
 def show_feedback_section():
     """Display writing feedback section"""
@@ -2700,26 +2793,20 @@ def show_story_input():
     
     # Submit and Clear buttons
     col1, col2, col3 = st.columns([3, 1, 1])
-    
+
     with col1:
-        submit_button = st.button(
-            "📝 ส่งคำตอบ | Submit",
-            use_container_width=True,
-            disabled=not text_input.strip()  # Disable if empty
-        )
-        
-        if submit_button:
-            if text_input.strip():
-                try:
-                    # Save current submission
-                    st.session_state.last_submission = text_input.strip()
-                    with st.spinner("กำลังวิเคราะห์ประโยค..."):
-                        handle_story_submission(text_input.strip())
-                except Exception as e:
-                    logging.error(f"Error submitting story: {str(e)}")
-                    st.error("เกิดข้อผิดพลาดในการส่งคำตอบ กรุณาลองใหม่อีกครั้ง")
-            else:
+        if st.button("📝 ส่งคำตอบ | Submit", use_container_width=True):
+            if not text_input.strip():
                 st.warning("กรุณาเขียนข้อความก่อนส่ง")
+                return
+            try:
+                # เล่นเสียงตอนกดปุ่ม Submit
+                st.markdown(f"<script>{st.session_state.sound_commands['submit']}</script>", unsafe_allow_html=True)
+                with st.spinner("กำลังวิเคราะห์ประโยค..."):
+                    handle_story_submission(text_input.strip())
+            except Exception as e:
+                logging.error(f"Error submitting story: {str(e)}")
+                st.error("เกิดข้อผิดพลาดในการส่งคำตอบ กรุณาลองใหม่อีกครั้ง")
     
     with col2:
         # Clear button with unique key
@@ -3437,12 +3524,51 @@ def create_story_pdf():
         logging.error(f"Error creating PDF: {str(e)}")
         raise
 
+def check_audio_system():
+    """Check if audio system is working properly"""
+    try:
+        if 'audio_manager' in st.session_state:
+            # ตรวจสอบว่า audio elements ถูกโหลดสำเร็จ
+            js_check = """
+                <script>
+                function checkAudioElements() {
+                    const elements = ['bgm', 'sound_submit', 'sound_achievement', 'sound_complete'];
+                    for (let id of elements) {
+                        if (!document.getElementById(id)) {
+                            console.error(`Audio element ${id} not found`);
+                            return false;
+                        }
+                    }
+                    return true;
+                }
+                window.audioSystemWorking = checkAudioElements();
+                </script>
+            """
+            st.markdown(js_check, unsafe_allow_html=True)
+            return True
+    except Exception as e:
+        logging.error(f"Error checking audio system: {str(e)}")
+        return False
+
 # === 6. MAIN APPLICATION LOGIC ===
 def main():
     try:
         # Initialize states
         init_session_state()
         init_theme_state()
+
+        # Check if reset is needed
+        if st.session_state.get('should_reset'):
+            reset_story()
+            st.rerun()
+            return
+
+        # Initialize audio system
+        initialize_audio()
+
+        # Check audio system
+        if not check_audio_system():
+            st.warning("ระบบเสียงอาจทำงานไม่สมบูรณ์ แต่คุณยังสามารถใช้งานแอพได้")
         
         # Initialize ending system state if not exists
         if 'ending_mode' not in st.session_state:
@@ -3470,7 +3596,8 @@ def main():
         # Sidebar
         with st.sidebar:
             show_sidebar()
-
+            show_audio_controls()
+            
         # Session Status Check
         check_session_status()