Update app.py

app.py

@@ -416,34 +416,26 @@ def provide_feedback(text: str, level: str) -> Dict[str, str]:
         response_text = response.choices[0].message.content.strip()
         
         try:
-            # Try to parse the JSON response
+            # แสดง debug info เพื่อตรวจสอบ response
+            st.write("Debug - Response received:", response_text)
+            
+            # Parse JSON response
             feedback_data = json.loads(response_text)
             
-            # Validate the required fields
-            required_fields = ['feedback', 'corrected', 'has_errors']
-            if not all(field in feedback_data for field in required_fields):
+            # Validate required fields
+            if all(key in feedback_data for key in ['feedback', 'corrected', 'has_errors']):
+                return feedback_data
+            else:
                 raise ValueError("Missing required fields in response")
-            
-            return feedback_data
-            
+                
         except json.JSONDecodeError as json_err:
-            # If JSON parsing fails, try to extract information manually
-            st.error(f"Debug - Response received: {response_text}")
-            
-            # Return a safe fallback response
-            return {
-                "feedback": "⚠️ คำแนะนำ: ระบบไม่สามารถวิเคราะห์ประโยคได้อย่างถูกต้อง กรุณาลองใหม่อีกครั้ง",
-                "corrected": text,
-                "has_errors": False
-            }
+            st.error(f"Debug - JSON Error: {str(json_err)}")
+            raise
             
     except Exception as e:
-        st.error(f"Debug - Error type: {type(e).__name__}")
-        st.error(f"Debug - Error message: {str(e)}")
-        
-        # Return a safe fallback response
+        st.error(f"Debug - Error: {str(e)}")
         return {
-            "feedback": "⚠️ ขออภัย ระบบไม่สามารถวิเคราะห์ประโยคได้ในขณะนี้",
+            "feedback": "⚠️ ระบบไม่สามารถวิเคราะห์ประโยคได้ กรุณาลองใหม่อีกครั้ง",
             "corrected": text,
             "has_errors": False
         }
@@ -560,8 +552,7 @@ with col1:
     )
 
 with col2:
-    # Feedback Display
-    if st.session_state.feedback and st.session_state.feedback.get('has_errors', False):
+    if st.session_state.feedback:
         st.markdown("""
             <div class="thai-eng">
                 <div class="thai">📝 คำแนะนำจากครู</div>
@@ -569,9 +560,8 @@ with col2:
             </div>
         """, unsafe_allow_html=True)
         
-        # แสดง feedback ในรูปแบบที่อ่านง่าย
-        feedback_container = st.container()
-        with feedback_container:
+        feedback_data = st.session_state.feedback
+        if isinstance(feedback_data, dict) and feedback_data.get('has_errors'):
             st.markdown(f"""
                 <div style='background-color: #f0f2f6; 
                        padding: 15px; 
@@ -580,18 +570,18 @@ with col2:
                        margin: 5px 0;
                        font-family: "Sarabun", sans-serif;'>
                     <p style='color: #1e88e5; margin-bottom: 10px;'>
-                        {st.session_state.feedback['feedback']}
+                        {feedback_data['feedback']}
                     </p>
                     <p style='color: #666; font-size: 0.9em;'>
                         ประโยคที่ถูกต้อง:<br/>
                         <span style='color: #4CAF50; font-weight: bold;'>
-                            {st.session_state.feedback['corrected']}
+                            {feedback_data['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))) 
@@ -601,24 +591,22 @@ with col2:
                 if last_user_entry_idx is not None:
                     apply_correction(
                         last_user_entry_idx, 
-                        st.session_state.feedback['corrected']
+                        feedback_data['corrected']
                     )
                     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)
+        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;'>
+                        {feedback_data.get('feedback', '✨ เขียนได้ถูกต้องแล้วค่ะ!')}
+                    </p>
+                </div>
+            """, unsafe_allow_html=True)
     
     # Help and Suggestions Box
     st.markdown("""