Update app.py

app.py

@@ -126,16 +126,31 @@ def submit_story():
             "is_corrected": False
         })
         
-        # รับ feedback และประโยคที่ถูกต้อง
-        feedback_data = provide_feedback(user_text, st.session_state.level)
-        st.session_state.feedback = feedback_data
-        
-        # Generate AI continuation
-        ai_response = generate_story_continuation(
-            feedback_data['corrected'] if feedback_data['has_errors'] else user_text, 
-            st.session_state.level
-        )
-        st.session_state.story.append({"role": "AI", "content": ai_response})
+        try:
+            # รับ feedback และประโยคที่ถูกต้อง
+            feedback_data = provide_feedback(user_text, st.session_state.level)
+            st.session_state.feedback = feedback_data
+            
+            # ถ้ามีข้อผิดพลาด แสดงคำแนะนำ
+            if feedback_data['has_errors']:
+                st.markdown(f"""
+                    <div style='background-color: #fff3e0; padding: 10px; border-radius: 5px; margin: 10px 0;'>
+                        <p style='color: #ff6d00;'>🎯 คำแนะนำ:</p>
+                        <p>{feedback_data['feedback']}</p>
+                    </div>
+                """, unsafe_allow_html=True)
+            
+            # Generate AI continuation using corrected text if there were errors
+            ai_response = generate_story_continuation(
+                feedback_data['corrected'] if feedback_data['has_errors'] else user_text, 
+                st.session_state.level
+            )
+            st.session_state.story.append({"role": "AI", "content": ai_response})
+            
+        except Exception as e:
+            st.error("เกิดข้อผิดพลาดในการวิเคราะห์ประโยค กรุณาลองใหม่อีกครั้ง")
+            # Log error for debugging
+            st.error(f"Debug - Error in submit_story: {str(e)}")
         
         # Clear input
         st.session_state.text_input = ""
@@ -356,65 +371,77 @@ def get_creative_prompt() -> Dict[str, str]:
             "thai": "แล้วอะไรจะเกิดขึ้นต่อ?"
         }
 
-def provide_feedback(text: str, level: str) -> str:
-    """Provide educational feedback in Thai with English examples."""
+def provide_feedback(text: str, level: str) -> Dict[str, str]:
+    """Provide feedback and corrected sentence."""
     try:
         response = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
                 {"role": "system", "content": f"""You are a Thai English teacher helping {level} students.
-                Provide feedback in Thai language that:
-                1. Points out ONE main grammar or vocabulary point
-                2. Explains the correct usage in simple Thai
-                3. Shows example sentences
-                4. Gives encouragement
-
-                Format example:
-                "🎯 คำว่า 'go' เมื่อพูดถึงเหตุการณ์ในอดีต ต้องเปลี่ยนเป็น 'went' นะคะ
+                Your task is to review the student's sentence and provide feedback.
                 
-                ตัวอย่าง:
-                - ถูก: Yesterday, I went to school.
-                - ผิด: Yesterday, I go to school.
-
-                เก่งมากๆ เลยค่ะ ที่พยายามเขียนเป็นภาษาอังกฤษ! ✨"
-                """},
-                {"role": "user", "content": f"Review this sentence and provide feedback in Thai: {text}"}
+                Return your response in this EXACT format only (must be valid JSON):
+                {{
+                    "feedback": "(ข้อเสนอแนะภาษาไทย)",
+                    "corrected": "(ประโยคภาษาอังกฤษที่ถูกต้อง)",
+                    "has_errors": true/false
+                }}
+                
+                Example 1 - with error:
+                {{
+                    "feedback": "คำว่า 'go' เมื่อพูดถึงเหตุการณ์ในอดีต ต้องเปลี่ยนเป็น 'went'",
+                    "corrected": "I went to school yesterday.",
+                    "has_errors": true
+                }}
+                
+                Example 2 - no error:
+                {{
+                    "feedback": "เขียนได้ถูกต้องแล้วค่ะ ประโยคสื่อความหมายได้ดี",
+                    "corrected": "The cat is sleeping.",
+                    "has_errors": false
+                }}"""},
+                {"role": "user", "content": f"Review this sentence and provide feedback in the specified JSON format: {text}"}
             ],
             max_tokens=200,
-            temperature=0.7
+            temperature=0.3
         )
-        feedback_data = json.loads(response.choices[0].message.content)
-        return feedback_data
+        
+        # Get the response text
+        response_text = response.choices[0].message.content.strip()
+        
+        try:
+            # Try to parse the 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):
+                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
+            }
+            
     except Exception as e:
-        st.error(f"Error generating feedback: {str(e)}")
+        st.error(f"Debug - Error type: {type(e).__name__}")
+        st.error(f"Debug - Error message: {str(e)}")
+        
+        # Return a safe fallback response
         return {
-            "feedback": "ขออภัย ไม่สามารถวิเคราะห์ประโยคได้",
+            "feedback": "⚠️ ขออภัย ระบบไม่สามารถวิเคราะห์ประโยคได้ในขณะนี้",
             "corrected": text,
             "has_errors": False
         }
 
-# Update the feedback display in the UI
-    if st.session_state.feedback:
-        st.markdown("""
-            <div class="thai-eng">
-                <div class="thai">📝 คำแนะนำจากคุณครู</div>
-                <div class="eng">Teacher's Feedback</div>
-            </div>
-        """, unsafe_allow_html=True)
-        
-        feedback_container = st.container()
-        with feedback_container:
-            st.markdown(f"""
-                <div style='background-color: #f0f2f6; 
-                           padding: 10px; 
-                           border-radius: 8px; 
-                           border-left: 4px solid #4CAF50;
-                           margin: 5px 0;
-                           font-family: "Sarabun", sans-serif;'>
-                    {st.session_state.feedback}
-                </div>
-            """, unsafe_allow_html=True)
-
 def update_achievements(text: str):
     """Update user achievements based on their writing."""
     words = set(text.lower().split())