Update app.py

app.py

@@ -20,3 +20,30 @@ if start_button:
     st.session_state["level"] = level  # บันทึกระดับที่เลือกไว้ใน session
     st.session_state["story_started"] = True  # ใช้เพื่อเช็คว่าเริ่มเรื่องหรือยัง
     st.experimental_rerun()  # รีเฟรชหน้าเพื่อลงไปยังส่วนการแต่งเรื่อง
+
+
+# ตรวจสอบว่าการแต่งเรื่องเริ่มแล้วหรือยัง
+if st.session_state.get("story_started"):
+    st.header("JoyStory - Let's Create!")
+    
+    # แสดงเรื่องราวที่กำลังดำเนินอยู่
+    if "story_text" not in st.session_state:
+        st.session_state["story_text"] = "Once upon a time, in a magical forest..."
+    
+    st.write(st.session_state["story_text"])
+    
+    # กล่องให้เด็กแต่งประโยคของตัวเอง
+    user_input = st.text_input("Add your sentence:")
+    if st.button("Submit"):
+        # ประมวลผลเพื่อให้ AI เพิ่มประโยคต่อจากที่เด็กเขียน
+        response = openai.Completion.create(
+            model="text-davinci-003",
+            prompt=st.session_state["story_text"] + " " + user_input,
+            max_tokens=50
+        )
+        ai_text = response.choices[0].text.strip()
+        
+        # อัปเดตเรื่องราวที่กำลังดำเนินอยู่
+        st.session_state["story_text"] += " " + user_input + " " + ai_text
+        st.experimental_rerun()  # รีเฟรชหน้าจอเพื่อแสดงประโยคใหม่
+