Update app.py

app.py

@@ -14,78 +14,73 @@ if 'current_level' not in st.session_state:
     st.session_state.current_level = 'beginner'
 if 'story_started' not in st.session_state:
     st.session_state.story_started = False
+if 'first_turn' not in st.session_state:
+    st.session_state.first_turn = None
+if 'current_turn' not in st.session_state:
+    st.session_state.current_turn = None
 if 'badges' not in st.session_state:
     st.session_state.badges = defaultdict(int)
 if 'vocabulary_used' not in st.session_state:
     st.session_state.vocabulary_used = set()
-if 'learning_summary' not in st.session_state:
-    st.session_state.learning_summary = []
 
-def get_ai_response(prompt, level):
-    """Get response from OpenAI API"""
-    system_prompt = f"""You are an English teacher helping a {level} level student write a story. 
-    Provide the following in JSON format:
-    1. next_sentence: Continue the story naturally
-    2. feedback: Constructive feedback about their writing
-    3. vocabulary_suggestions: 3 new words they could use, appropriate for their level
-    4. question: An open-ended question about the story to stimulate creativity
-    5. grammar_check: Any grammar corrections needed
-    6. visual_description: A simple description of the current scene
-    Keep responses appropriate for the student's level."""
-    
+def get_ai_story_continuation(story_context, level):
+    """Get story continuation from OpenAI API"""
     try:
         response = openai.ChatCompletion.create(
-            model="gpt-4o-mini",
+            model="gpt-4-mini",  # ใช้ model ที่ถูกต้อง
             messages=[
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": f"Story so far: {prompt}\n\nProvide response in JSON format."}
+                {"role": "system", "content": f"""You are helping a {level} level English student write a story. 
+                Provide a JSON response with:
+                1. continuation: A single sentence continuing the story (appropriate for {level} level)
+                2. creative_prompt: A short question to guide the student's next sentence
+                3. vocabulary_suggestions: 3 relevant words they might use next
+                Make the story engaging and suitable for their level."""},
+                {"role": "user", "content": f"Story context: {story_context}\n\nProvide the next part of the story and a guiding question."}
             ],
             temperature=0.7
         )
         return json.loads(response.choices[0].message.content)
     except Exception as e:
         st.error(f"Error with AI response: {e}")
-        return get_default_response(level)
-
-def get_default_response(level):
-    """Fallback response if API fails"""
-    return {
-        "next_sentence": "The story continues...",
-        "feedback": "Please try again.",
-        "vocabulary_suggestions": ["word1", "word2", "word3"],
-        "question": "What happens next?",
-        "grammar_check": "No issues found.",
-        "visual_description": "A simple scene"
-    }
+        return {
+            "continuation": "The story continues...",
+            "creative_prompt": "What happens next?",
+            "vocabulary_suggestions": ["word1", "word2", "word3"]
+        }
 
-def award_badge(category):
-    """Award badges based on achievements"""
-    st.session_state.badges[category] += 1
-    st.balloons()
-    st.success(f"🏆 New Badge Earned: {category}!")
-
-def generate_learning_summary():
-    """Generate end-of-story learning summary"""
-    unique_words = len(st.session_state.vocabulary_used)
-    story_length = len(st.session_state.story)
-    summary = {
-        "total_sentences": story_length,
-        "unique_vocabulary": unique_words,
-        "badges_earned": dict(st.session_state.badges),
-        "level": st.session_state.current_level,
-        "completion_date": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    }
-    return summary
+def start_story(level):
+    """Get the first sentence from AI"""
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": f"""You are starting a story for a {level} level English student.
+                Provide a JSON response with:
+                1. opening: An engaging opening sentence (appropriate for {level} level)
+                2. creative_prompt: A short question to guide the student's first sentence
+                3. vocabulary_suggestions: 3 words they might use in their response"""},
+                {"role": "user", "content": "Please start a new story."}
+            ],
+            temperature=0.7
+        )
+        return json.loads(response.choices[0].message.content)
+    except Exception as e:
+        st.error(f"Error with AI response: {e}")
+        return {
+            "opening": "Once upon a time...",
+            "creative_prompt": "Who is our main character?",
+            "vocabulary_suggestions": ["brave", "curious", "friendly"]
+        }
 
 def main():
     st.title("🎨 Interactive Story Adventure")
     
-    # Level selection
+    # Level selection and first turn choice
     if not st.session_state.story_started:
         col1, col2 = st.columns([2, 1])
         with col1:
             st.write("### Welcome to Story Adventure!")
-            st.write("Let's create an amazing story together. Choose your level and start writing!")
+            st.write("Let's create an amazing story together. Choose your level and who starts first!")
         
         level = st.selectbox(
             "Choose your English level:",
@@ -93,9 +88,30 @@ def main():
             key='level_select'
         )
         
+        first_turn = st.radio(
+            "Who should start the story?",
+            ['AI', 'Player'],
+            key='first_turn_select'
+        )
+        
         if st.button("Start Story!"):
             st.session_state.current_level = level
+            st.session_state.first_turn = first_turn
+            st.session_state.current_turn = first_turn
             st.session_state.story_started = True
+            
+            # If AI starts, get the first sentence
+            if first_turn == 'AI':
+                response = start_story(level)
+                st.session_state.story.append({
+                    'author': 'AI',
+                    'text': response['opening'],
+                    'prompt': response['creative_prompt'],
+                    'vocabulary': response['vocabulary_suggestions'],
+                    'timestamp': datetime.now().isoformat()
+                })
+                st.session_state.current_turn = 'Player'
+            
             st.rerun()
     
     # Main story interface
@@ -103,119 +119,66 @@ def main():
         # Story display area
         with st.container():
             st.write("### Our Story So Far:")
-            for i, entry in enumerate(st.session_state.story):
+            for entry in st.session_state.story:
                 if entry['author'] == 'AI':
                     st.info(entry['text'])
-                    if 'visual_description' in entry:
-                        st.caption(f"🎨 Scene: {entry['visual_description']}")
+                    st.write("🤔 " + entry['prompt'])
                 else:
                     st.success(entry['text'])
         
-        # Writing area
-        with st.container():
-            # Show AI's question if available
-            if st.session_state.story and 'question' in st.session_state.story[-1]:
-                st.write(f"🤔 Think about: {st.session_state.story[-1]['question']}")
-            
-            user_input = st.text_area("Add to the story:", key='story_input')
-            col1, col2, col3 = st.columns([1,1,1])
-            
-            with col1:
+        # Writing area - only show if it's player's turn
+        if st.session_state.current_turn == 'Player':
+            with st.container():
+                # Show vocabulary suggestions if available
+                if st.session_state.story and 'vocabulary' in st.session_state.story[-1]:
+                    st.write("💡 Suggested words you might use:")
+                    for word in st.session_state.story[-1]['vocabulary']:
+                        st.write(f"- {word}")
+                
+                user_input = st.text_area("Your turn! Add to the story:", key='story_input')
                 if st.button("Submit"):
                     if user_input:
-                        # Get story so far as context
-                        story_context = " ".join([entry['text'] for entry in st.session_state.story])
-                        
-                        # Add user's sentence to story
+                        # Add player's sentence
                         st.session_state.story.append({
-                            'author': 'User',
+                            'author': 'Player',
                             'text': user_input,
                             'timestamp': datetime.now().isoformat()
                         })
                         
-                        # Get AI response
-                        ai_response = get_ai_response(f"{story_context}\n{user_input}", 
-                                                    st.session_state.current_level)
+                        # Get AI's response
+                        story_context = " ".join([entry['text'] for entry in st.session_state.story])
+                        ai_response = get_ai_story_continuation(story_context, st.session_state.current_level)
                         
-                        # Add AI's sentence to story
+                        # Add AI's response
                         st.session_state.story.append({
                             'author': 'AI',
-                            'text': ai_response['next_sentence'],
-                            'question': ai_response['question'],
-                            'visual_description': ai_response['visual_description'],
+                            'text': ai_response['continuation'],
+                            'prompt': ai_response['creative_prompt'],
+                            'vocabulary': ai_response['vocabulary_suggestions'],
                             'timestamp': datetime.now().isoformat()
                         })
                         
-                        # Show feedback and grammar check
-                        st.write("### Feedback:")
-                        st.write(ai_response['feedback'])
-                        if ai_response['grammar_check'] != "No issues found.":
-                            st.warning(ai_response['grammar_check'])
-                        
-                        # Update vocabulary used
-                        words = set(user_input.lower().split())
-                        st.session_state.vocabulary_used.update(words)
-                        
-                        # Check for badges
-                        if len(st.session_state.story) >= 10:
-                            award_badge("Storyteller")
-                        if len(st.session_state.vocabulary_used) >= 20:
-                            award_badge("Vocabulary Master")
-                        
                         st.rerun()
-            
-            with col2:
-                if st.button("Get Help"):
-                    if st.session_state.story:
-                        ai_response = get_ai_response(" ".join([entry['text'] for entry in st.session_state.story]), 
-                                                    st.session_state.current_level)
-                        st.write("### Suggested Words:")
-                        for word in ai_response['vocabulary_suggestions']:
-                            st.write(f"- {word}")
-            
-            with col3:
-                if st.button("Finish Story"):
-                    if len(st.session_state.story) > 0:
-                        # Generate learning summary
-                        summary = generate_learning_summary()
-                        st.session_state.learning_summary.append(summary)
-                        
-                        st.write("### Story Complete! 🎉")
-                        st.write("#### Your Achievements:")
-                        st.write(f"- Total sentences written: {summary['total_sentences']}")
-                        st.write(f"- Unique words used: {summary['unique_vocabulary']}")
-                        st.write("- Badges earned:")
-                        for badge, count in summary['badges_earned'].items():
-                            st.write(f"  * {badge}: {count} 🏆")
-                        
-                        # Download options
-                        story_text = "\n".join([entry['text'] for entry in st.session_state.story])
-                        st.download_button(
-                            label="Download Story",
-                            data=story_text,
-                            file_name="my_story.txt",
-                            mime="text/plain"
-                        )
-                        
-                        # Download progress report for parents/teachers
-                        progress_report = {
-                            "student_level": st.session_state.current_level,
-                            "story_summary": summary,
-                            "learning_progress": st.session_state.learning_summary
-                        }
-                        st.download_button(
-                            label="Download Progress Report",
-                            data=json.dumps(progress_report, indent=2),
-                            file_name="progress_report.json",
-                            mime="application/json"
-                        )
-                        
-                        # Reset session state
-                        if st.button("Start New Story"):
-                            st.session_state.story = []
-                            st.session_state.story_started = False
-                            st.session_state.vocabulary_used = set()
-                            st.rerun()
+        
+        # Finish story button
+        if st.button("Finish Story"):
+            if len(st.session_state.story) > 0:
+                st.write("### Story Complete! 🎉")
+                story_text = "\n".join([entry['text'] for entry in st.session_state.story])
+                st.download_button(
+                    label="Download Story",
+                    data=story_text,
+                    file_name="my_story.txt",
+                    mime="text/plain"
+                )
+                
+                if st.button("Start New Story"):
+                    st.session_state.story = []
+                    st.session_state.story_started = False
+                    st.session_state.first_turn = None
+                    st.session_state.current_turn = None
+                    st.session_state.vocabulary_used = set()
+                    st.rerun()
 
 if __name__ == "__main__":
     main()