Update app.py

app.py

@@ -94,6 +94,8 @@ def init_session_state():
         st.session_state.story = []
     if 'feedback' not in st.session_state:
         st.session_state.feedback = None
+    if 'corrections' not in st.session_state:
+        st.session_state.corrections = {}  # เก็บประวัติการแก้ไข
     if 'level' not in st.session_state:
         st.session_state.level = 'Beginner'
     if 'unique_words' not in st.session_state:
@@ -114,20 +116,28 @@ def clear_input():
 # Callback function for submit button
 def submit_story():
     if st.session_state.text_input.strip():
-        # Add user's input to story
-        st.session_state.story.append({"role": "You", "content": st.session_state.text_input})
+        user_text = st.session_state.text_input
         
-        # Update achievements
-        update_achievements(st.session_state.text_input)
+        # เพิ่มประโยคของผู้ใช้
+        story_index = len(st.session_state.story)
+        st.session_state.story.append({
+            "role": "You", 
+            "content": user_text,
+            "is_corrected": False
+        })
         
-        # Generate feedback
-        st.session_state.feedback = provide_feedback(st.session_state.text_input, st.session_state.level)
+        # รับ 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(st.session_state.text_input, st.session_state.level)
+        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})
         
-        # Clear the input by setting it to empty
+        # Clear input
         st.session_state.text_input = ""
 
 def show_welcome_section():
@@ -250,6 +260,21 @@ def generate_story_continuation(user_input: str, level: str) -> str:
         st.error(f"Error generating story continuation: {str(e)}")
         return "I'm having trouble continuing the story. Please try again."
 
+# ฟังก์ชันสำหรับแก้ไขประโยค
+def apply_correction(story_index: int, corrected_text: str):
+    """Apply correction to a specific story entry."""
+    if 0 <= story_index < len(st.session_state.story):
+        original_text = st.session_state.story[story_index]['content']
+        # เก็บประวัติการแก้ไข
+        st.session_state.corrections[story_index] = {
+            'original': original_text,
+            'corrected': corrected_text,
+            'timestamp': datetime.datetime.now().isoformat()
+        }
+        # แก้ไขประโยคในเรื่อง
+        st.session_state.story[story_index]['content'] = corrected_text
+        st.session_state.story[story_index]['is_corrected'] = True
+
 def get_vocabulary_suggestions() -> List[str]:
     """Get contextual vocabulary suggestions with Thai translations."""
     try:
@@ -358,10 +383,15 @@ def provide_feedback(text: str, level: str) -> str:
             max_tokens=200,
             temperature=0.7
         )
-        return response.choices[0].message.content
+        feedback_data = json.loads(response.choices[0].message.content)
+        return feedback_data
     except Exception as e:
         st.error(f"Error generating feedback: {str(e)}")
-        return "🌟 พยายามได้ดีมากค่ะ/ครับ ลองเขียนต่อไปนะ!"
+        return {
+            "feedback": "ขออภัย ไม่สามารถวิเคราะห์ประโยคได้",
+            "corrected": text,
+            "has_errors": False
+        }
 
 # Update the feedback display in the UI
     if st.session_state.feedback:
@@ -462,13 +492,27 @@ with col1:
     """, unsafe_allow_html=True)
     
     story_display = st.container()
+
     with story_display:
         if not st.session_state.story:
             st.info("เริ่มต้นผจญภัยด้วยการเขียนประโยคแรกกันเลย! | Start your adventure by writing the first sentence!")
         else:
-            for entry in st.session_state.story:
+            for idx, entry in enumerate(st.session_state.story):
                 if entry['role'] == 'You':
-                    st.write("👤 You:", entry['content'])
+                    col_text, col_button = st.columns([4, 1])
+                    with col_text:
+                        prefix = "👤 You (แก้ไขแล้ว):" if entry.get('is_corrected') else "👤 You:"
+                        st.write(f"{prefix} {entry['content']}")
+                    
+                    # ถ้ามี feedback และมีข้อผิด แสดงปุ่��แก้ไข
+                    if (idx in st.session_state.corrections and 
+                        not entry.get('is_corrected') and 
+                        st.session_state.feedback and 
+                        st.session_state.feedback.get('has_errors')):
+                        with col_button:
+                            if st.button(f"✍️ แก้ไขประโยค", key=f"correct_{idx}"):
+                                apply_correction(idx, st.session_state.feedback['corrected'])
+                                st.rerun()
                 elif entry['role'] == 'AI':
                     st.write("🤖 AI:", entry['content'])