Update app.py

app.py

@@ -1,41 +1,86 @@
 import streamlit as st
-import json
+import openai
 from datetime import datetime
+import json
+from collections import defaultdict
+
+# Initialize OpenAI
+openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-# Initialize session state variables if they don't exist
+# Initialize session state variables
 if 'story' not in st.session_state:
     st.session_state.story = []
 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 '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."""
+    
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4-turbo-preview",
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": f"Story so far: {prompt}\n\nProvide response in JSON format."}
+            ],
+            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)
 
-# Function to simulate AI response (replace with actual GPT API call later)
-def get_ai_response(story_so_far, level):
-    # This is a placeholder - replace with actual GPT API integration
-    level_responses = {
-        'beginner': {
-            'next_sentence': "The little cat jumped over the fence.",
-            'feedback': "Good job! Try using more descriptive words next time.",
-            'vocabulary_suggestions': ['leap', 'hop', 'bounce']
-        },
-        'intermediate': {
-            'next_sentence': "The adventurous feline gracefully leaped over the wooden fence.",
-            'feedback': "Nice structure! Consider adding more complex adjectives.",
-            'vocabulary_suggestions': ['gracefully', 'elegantly', 'swiftly']
-        },
-        'advanced': {
-            'next_sentence': "With remarkable agility, the intrepid feline launched itself over the weathered fence.",
-            'feedback': "Excellent use of complex sentence structure!",
-            'vocabulary_suggestions': ['agility', 'intrepid', 'launched']
-        }
+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 level_responses[level]
+
+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 main():
     st.title("🎨 Interactive Story Adventure")
     
-    # Level selection only shown at start
+    # Level selection
     if not st.session_state.story_started:
         col1, col2 = st.columns([2, 1])
         with col1:
@@ -61,17 +106,26 @@ def main():
             for i, entry in enumerate(st.session_state.story):
                 if entry['author'] == 'AI':
                     st.info(entry['text'])
+                    if 'visual_description' in entry:
+                        st.caption(f"🎨 Scene: {entry['visual_description']}")
                 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:
                 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
                         st.session_state.story.append({
                             'author': 'User',
@@ -80,31 +134,61 @@ def main():
                         })
                         
                         # Get AI response
-                        ai_response = get_ai_response(st.session_state.story, st.session_state.current_level)
+                        ai_response = get_ai_response(f"{story_context}\n{user_input}", 
+                                                    st.session_state.current_level)
                         
                         # Add AI's sentence to story
                         st.session_state.story.append({
                             'author': 'AI',
                             'text': ai_response['next_sentence'],
+                            'question': ai_response['question'],
+                            'visual_description': ai_response['visual_description'],
                             'timestamp': datetime.now().isoformat()
                         })
                         
-                        # Show feedback
+                        # 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 Vocabulary Help"):
-                    ai_response = get_ai_response(st.session_state.story, st.session_state.current_level)
-                    st.write("### Suggested Words:")
-                    for word in ai_response['vocabulary_suggestions']:
-                        st.write(f"- {word}")
+                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:
-                        st.write("### Story Complete!")
+                        # 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",
@@ -112,10 +196,26 @@ def main():
                             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
-                        st.session_state.story = []
-                        st.session_state.story_started = False
-                        st.rerun()
+                        if st.button("Start New Story"):
+                            st.session_state.story = []
+                            st.session_state.story_started = False
+                            st.session_state.vocabulary_used = set()
+                            st.rerun()
 
 if __name__ == "__main__":
     main()