Update app.py

app.py

@@ -56,21 +56,48 @@ def generate_story_continuation(user_input: str, level: str) -> str:
         return "I'm having trouble continuing the story. Please try again."
 
 def get_vocabulary_suggestions() -> List[str]:
-    """Get vocabulary suggestions based on the story context."""
+    """Get contextual vocabulary suggestions based on the current story."""
     try:
+        # Get the last few story entries for context
+        recent_story = '\n'.join([entry['content'] for entry in st.session_state.story[-3:]] if st.session_state.story else "Story just started")
+        
         response = client.chat.completions.create(
-            model="gpt-4o-mini",
+            model="gpt-3.5-turbo",
             messages=[
-                {"role": "system", "content": f"You are a language tutor. Suggest 3 appropriate vocabulary words for {st.session_state.level} English learners."},
-                {"role": "user", "content": "Suggest 3 interesting words that could be used in the story. Format: word1, word2, word3"}
+                {"role": "system", "content": f"""You are a creative writing tutor for {st.session_state.level} English learners.
+                Based on the story context, suggest 5 varied words that could be used next:
+                - Include a mix of nouns, verbs, and adjectives
+                - Match the {st.session_state.level} level
+                - Words should fit the story theme
+                - No basic words like 'happy', 'sad', 'good'
+                Format: word1 (type), word2 (type), etc."""},
+                {"role": "user", "content": f"Story context:\n{recent_story}\n\nSuggest 5 relevant, interesting words:"}
             ],
-            max_tokens=50,
-            temperature=0.6
+            max_tokens=100,
+            temperature=0.8
         )
-        return response.choices[0].message.content.split(', ')
+        
+        # Parse the response to get words and their types
+        word_suggestions = response.choices[0].message.content.split(', ')
+        return word_suggestions
     except Exception as e:
         st.error(f"Error getting vocabulary suggestions: {str(e)}")
-        return ["exciting", "beautiful", "happy"]
+        return ["adventure (noun)", "whisper (verb)", "mysterious (adjective)"]
+
+# In the main UI section, update how vocabulary suggestions are displayed:
+    if st.button("Get Vocabulary Ideas"):
+        vocab_suggestions = get_vocabulary_suggestions()
+        st.markdown("#### 📚 Suggested Words")
+        for word in vocab_suggestions:
+            st.markdown(f"• *{word}*")
+
+# And update how feedback is displayed to be more concise:
+    if st.session_state.feedback:
+        st.markdown("""
+            <div style='background-color: #f0f2f6; padding: 8px; border-radius: 4px; margin-bottom: 10px;'>
+                📝 <i>{}</i>
+            </div>
+            """.format(st.session_state.feedback), unsafe_allow_html=True)
 
 def get_creative_prompt() -> str:
     """Generate a creative prompt to inspire the next part of the story."""
@@ -90,21 +117,25 @@ def get_creative_prompt() -> str:
         return "What happens next in your story?"
 
 def provide_feedback(text: str, level: str) -> str:
-    """Provide constructive feedback on the user's writing."""
+    """Provide concise, constructive feedback on the user's writing."""
     try:
         response = client.chat.completions.create(
-            model="gpt-4o-mini",
+            model="gpt-3.5-turbo",
             messages=[
-                {"role": "system", "content": f"You are an encouraging English teacher for {level} students. Provide one brief, positive piece of feedback and one gentle correction if needed."},
-                {"role": "user", "content": f"Provide encouraging feedback on: {text}"}
+                {"role": "system", "content": f"""You are an encouraging English teacher for {level} students. 
+                Provide very brief feedback (max 15 words) focusing on ONE specific aspect:
+                - Either praise a good point OR
+                - Suggest one small improvement
+                Keep it short and friendly."""},
+                {"role": "user", "content": f"Give brief feedback on: {text}"}
             ],
-            max_tokens=50,
+            max_tokens=30,
             temperature=0.6
         )
         return response.choices[0].message.content
     except Exception as e:
         st.error(f"Error generating feedback: {str(e)}")
-        return "Great effort! Keep writing!"
+        return "Good effort! Keep writing!"
 
 def update_achievements(text: str):
     """Update user achievements based on their writing."""