Update app.py

app.py

@@ -561,7 +561,7 @@ with col1:
 
 with col2:
     # Feedback Display
-    if st.session_state.feedback:
+    if st.session_state.feedback and st.session_state.feedback.get('has_errors', False):
         st.markdown("""
             <div class="thai-eng">
                 <div class="thai">📝 คำแนะนำจากครู</div>
@@ -572,55 +572,53 @@ with col2:
         # แสดง feedback ในรูปแบบที่อ่านง่าย
         feedback_container = st.container()
         with feedback_container:
-            if st.session_state.feedback.get('has_errors', False):
-                st.markdown(f"""
-                    <div style='background-color: #f0f2f6; 
-                           padding: 15px; 
-                           border-radius: 8px; 
-                           border-left: 4px solid #FF9800;
-                           margin: 5px 0;
-                           font-family: "Sarabun", sans-serif;'>
-                        <p style='color: #1e88e5; margin-bottom: 10px;'>
-                            {st.session_state.feedback['feedback']}
-                        </p>
-                        <p style='color: #666; font-size: 0.9em;'>
-                            ประโยคที่ถูกต้อง:<br/>
-                            <span style='color: #4CAF50; font-weight: bold;'>
-                                {st.session_state.feedback['corrected']}
-                            </span>
-                        </p>
-                        <button class='correction-button' onclick='submit_correction()'>
-                            ✍️ แก้ไขประโยคให้ถูกต้อง
-                        </button>
-                    </div>
-                """, unsafe_allow_html=True)
-                
-                # ปุ่มแก้ไขประโยค
-                if st.button("✍️ แก้ไขประโยคให้ถูกต้อง", key="correct_last"):
-                    last_user_entry_idx = next(
-                        (i for i, entry in reversed(list(enumerate(st.session_state.story))) 
-                         if entry['role'] == 'You'),
-                        None
+            st.markdown(f"""
+                <div style='background-color: #f0f2f6; 
+                       padding: 15px; 
+                       border-radius: 8px; 
+                       border-left: 4px solid #FF9800;
+                       margin: 5px 0;
+                       font-family: "Sarabun", sans-serif;'>
+                    <p style='color: #1e88e5; margin-bottom: 10px;'>
+                        {st.session_state.feedback['feedback']}
+                    </p>
+                    <p style='color: #666; font-size: 0.9em;'>
+                        ประโยคที่ถูกต้อง:<br/>
+                        <span style='color: #4CAF50; font-weight: bold;'>
+                            {st.session_state.feedback['corrected']}
+                        </span>
+                    </p>
+                </div>
+            """, unsafe_allow_html=True)
+            
+            # แสดงปุ่มแก้ไขเพียงปุ่มเดียว
+            if st.button("✍️ แก้ไขประโยคให้ถูกต้อง", key="correct_button"):
+                last_user_entry_idx = next(
+                    (i for i, entry in reversed(list(enumerate(st.session_state.story))) 
+                     if entry['role'] == 'You'),
+                    None
+                )
+                if last_user_entry_idx is not None:
+                    apply_correction(
+                        last_user_entry_idx, 
+                        st.session_state.feedback['corrected']
                     )
-                    if last_user_entry_idx is not None:
-                        apply_correction(
-                            last_user_entry_idx, 
-                            st.session_state.feedback['corrected']
-                        )
-                        st.rerun()
-            else:
-                st.markdown(f"""
-                    <div style='background-color: #f0f2f6; 
-                           padding: 15px; 
-                           border-radius: 8px; 
-                           border-left: 4px solid #4CAF50;
-                           margin: 5px 0;
-                           font-family: "Sarabun", sans-serif;'>
-                        <p style='color: #1e88e5;'>
-                            ✨ {st.session_state.feedback['feedback']}
-                        </p>
-                    </div>
-                """, unsafe_allow_html=True)
+                    st.rerun()
+        
+    elif st.session_state.feedback:
+        # กรณีไม่มีข้อผิดพลาด
+        st.markdown(f"""
+            <div style='background-color: #f0f2f6; 
+                   padding: 15px; 
+                   border-radius: 8px; 
+                   border-left: 4px solid #4CAF50;
+                   margin: 5px 0;
+                   font-family: "Sarabun", sans-serif;'>
+                <p style='color: #1e88e5;'>
+                    ✨ {st.session_state.feedback['feedback']}
+                </p>
+            </div>
+        """, unsafe_allow_html=True)
     
     # Help and Suggestions Box
     st.markdown("""