Update app.py

app.py

@@ -450,7 +450,19 @@ def clear_input():
 # Callback function for submit button
 def submit_story():
     if st.session_state.text_input.strip():
-        user_text = st.session_state.text_input
+        user_text = st.session_state.text_input.strip()
+        
+        # ตรวจสอบว่ามีเรื่องเริ่มต้นหรือยัง
+        if not st.session_state.story:
+            st.error("กรุณาเลือกธีมเรื่องราวก่อนเริ่มเขียน")
+            return
+
+        # เพิ่มประโยคของผู้ใช้
+        st.session_state.story.append({
+            "role": "You",
+            "content": user_text,
+            "is_corrected": False
+        })
         
         # เพิ่มคำศัพท์ที่ใช้
         words = set(user_text.lower().split())
@@ -1106,19 +1118,22 @@ def show_theme_selection():
             </div>
             """
             # ใช้ button แทน markdown สำหรับการคลิก
-            if st.button(
-                theme['name_th'],
-                key=f"theme_button_{theme['id']}",
-                help=theme['description_th'],
-                use_container_width=True
-            ):
-                # บันทึกธีมที่เลือกใน session state
+            if st.button(theme['name_th'], key=f"theme_button_{theme['id']}", help=theme['description_th']):
+                # บันทึกธีมที่เลือก
                 st.session_state.current_theme = theme['id']
                 
-                # สร้างจุดเริ่มต้นเรื่องแบบไดนามิก
+                # สร้างจุดเริ่มต้นเรื่อง
                 starter = generate_dynamic_story_starter(theme['id'], st.session_state.level)
                 st.session_state.theme_story_starter = starter
                 
+                # เพิ่มจุดเริ่มต้นเรื่องเป็นประโยคแรกใน story
+                st.session_state.story = [{
+                    "role": "AI",
+                    "content": starter['en'],
+                    "thai_content": starter['th'],
+                    "is_starter": True
+                }]
+                
                 # แสดง theme intro
                 st.markdown(f"""
                     <div style="background-color: {theme['background_color']}; 
@@ -1132,20 +1147,13 @@ def show_theme_selection():
                                   padding: 15px;
                                   border-radius: 8px;
                                   margin-top: 10px;">
-                            <p style="color: #666;">จุดเริ่มต้นเรื่องของคุณ:</p>
+                            <p style="color: #666;">AI เริ่มเรื่องให้คุณแล้ว คุณพร้อมที่จะเขียนต่อ:</p>
                             <p style="color: #1e88e5;">{starter['th']}</p>
                             <p style="color: #333;">{starter['en']}</p>
                         </div>
                     </div>
                 """, unsafe_allow_html=True)
-                
-                # เพิ่มคำแนะนำการเขียน
-                st.info("📝 เริ่มเขียนเรื่องราวของคุณได้เลย! คุณสามารถใช้จุดเริ่มต้นด้านบนเป็นแรงบันดาลใจ")
-                
-                # แสดงคำศัพท์แนะนำสำหรับธีมนี้
-                with st.expander("📚 คำศัพท์แนะนำสำหรับธีมนี้"):
-                    show_theme_vocabulary()
-                
+        
                 st.rerun()
 
 # เพิ่ม CSS สำหรับ theme cards