Update app.py

app.py

@@ -1,174 +1,215 @@
 import streamlit as st
-import openai
-import datetime
 import json
+import datetime
+from openai import OpenAI
+from typing import Dict, List, Set
 
-# Set your OpenAI API key
-openai.api_key = 'YOUR_OPENAI_API_KEY'
+# Initialize OpenAI client
+client = OpenAI()  # Make sure OPENAI_API_KEY is set in your environment variables
 
-# Set up the Streamlit page configuration
+# Set up Streamlit page configuration
 st.set_page_config(
     page_title="JoyStory - Interactive Story Adventure",
-    page_icon=":book:",
+    page_icon="📖",
     layout="wide",
     initial_sidebar_state="collapsed",
 )
 
 # Initialize session state variables
-if 'story' not in st.session_state:
-    st.session_state.story = []
-if 'achievements' not in st.session_state:
-    st.session_state.achievements = []
-if 'level' not in st.session_state:
-    st.session_state.level = 'Beginner'
-if 'unique_words' not in st.session_state:
-    st.session_state.unique_words = set()
-if 'total_words' not in st.session_state:
-    st.session_state.total_words = 0
-if 'badges' not in st.session_state:
-    st.session_state.badges = []
-
-# Function to generate AI response
-def generate_ai_response(user_input, level):
-    prompt = f"Continue the story in a way that is engaging and suitable for a {level.lower()} English learner.\n\nStory so far:\n{''.join(st.session_state.story)}\nUser input: {user_input}\nAI continuation:"
-    response = openai.Completion.create(
-        engine='gpt-4o-mini',
-        prompt=prompt,
-        max_tokens=100,
-        temperature=0.7,
-    )
-    ai_output = response.choices[0].text.strip()
-    return ai_output
-
-# Function to get vocabulary suggestions
-def get_vocabulary_suggestions():
-    suggestions = openai.Completion.create(
-        engine='gpt-4o-mini',
-        prompt=f"Provide 3 interesting words suitable for a {st.session_state.level.lower()} English learner to use in the next part of the story.",
-        max_tokens=20,
-        n=1,
-        stop=None,
-        temperature=0.5,
-    )
-    words = suggestions.choices[0].text.strip().split(', ')
-    return words
-
-# Function to get creative prompts
-def get_creative_prompt():
-    prompt = openai.Completion.create(
-        engine='gpt-4o-mini',
-        prompt="Ask an open-ended question to inspire the next part of the story.",
-        max_tokens=30,
-        n=1,
-        stop=None,
-        temperature=0.7,
-    )
-    question = prompt.choices[0].text.strip()
-    return question
-
-# Function to provide feedback
-def get_feedback(user_input):
-    feedback_prompt = f"Provide one positive, simple feedback on the following sentence for a {st.session_state.level.lower()} English learner:\n\n{user_input}"
-    feedback = openai.Completion.create(
-        engine='gpt-4o-mini',
-        prompt=feedback_prompt,
-        max_tokens=50,
-        temperature=0.5,
-    )
-    return feedback.choices[0].text.strip()
+def init_session_state():
+    if 'story' not in st.session_state:
+        st.session_state.story = []
+    if 'achievements' not in st.session_state:
+        st.session_state.achievements = []
+    if 'level' not in st.session_state:
+        st.session_state.level = 'Beginner'
+    if 'unique_words' not in st.session_state:
+        st.session_state.unique_words = set()
+    if 'total_words' not in st.session_state:
+        st.session_state.total_words = 0
+    if 'badges' not in st.session_state:
+        st.session_state.badges = []
 
-# Function to update achievements
-def update_achievements(user_input):
-    words = set(user_input.lower().split())
+init_session_state()
+
+def generate_story_continuation(user_input: str, level: str) -> str:
+    """Generate AI story continuation using ChatGPT."""
+    try:
+        story_context = '\n'.join([entry['content'] for entry in st.session_state.story])
+        
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": f"You are a creative storytelling assistant for {level} English learners. Keep the language simple and engaging, avoid complex vocabulary, and maintain story coherence."},
+                {"role": "user", "content": f"Story so far:\n{story_context}\nUser's addition:\n{user_input}\nContinue the story in an engaging way:"}
+            ],
+            max_tokens=150,
+            temperature=0.7
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        st.error(f"Error generating story continuation: {str(e)}")
+        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."""
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            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"}
+            ],
+            max_tokens=50,
+            temperature=0.6
+        )
+        return response.choices[0].message.content.split(', ')
+    except Exception as e:
+        st.error(f"Error getting vocabulary suggestions: {str(e)}")
+        return ["exciting", "beautiful", "happy"]
+
+def get_creative_prompt() -> str:
+    """Generate a creative prompt to inspire the next part of the story."""
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": "Generate an inspiring question to help continue the story."},
+                {"role": "user", "content": "Create an open-ended question to inspire the next part of the story."}
+            ],
+            max_tokens=50,
+            temperature=0.7
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        st.error(f"Error generating creative prompt: {str(e)}")
+        return "What happens next in your story?"
+
+def provide_feedback(text: str, level: str) -> str:
+    """Provide constructive feedback on the user's writing."""
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": f"You are an encouraging English teacher for {level} students. Provide one brief, positive piece of feedback."},
+                {"role": "user", "content": f"Provide encouraging feedback on: {text}"}
+            ],
+            max_tokens=50,
+            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!"
+
+def update_achievements(text: str):
+    """Update user achievements based on their writing."""
+    words = set(text.lower().split())
     st.session_state.unique_words.update(words)
     st.session_state.total_words += len(words)
-    # Example achievement: Use 50 unique words
-    if len(st.session_state.unique_words) >= 50 and '50 Unique Words' not in st.session_state.badges:
-        st.session_state.badges.append('50 Unique Words')
-        st.success("Achievement Unlocked: 50 Unique Words!")
-
-# Layout the UI
+    
+    # Check for achievements
+    achievements = {
+        '50 Words': len(st.session_state.unique_words) >= 50,
+        '100 Words': len(st.session_state.unique_words) >= 100,
+        'Story Master': len(st.session_state.story) >= 10,
+    }
+    
+    for badge, condition in achievements.items():
+        if condition and badge not in st.session_state.badges:
+            st.session_state.badges.append(badge)
+            st.success(f"🏆 Achievement Unlocked: {badge}!")
+
+# Main UI Layout
 st.title("📖 JoyStory - Interactive Story Adventure")
 
+# Sidebar for settings
+with st.sidebar:
+    st.header("Settings")
+    level = st.radio(
+        "Select your English level:",
+        ('Beginner', 'Intermediate', 'Advanced')
+    )
+    st.session_state.level = level
+    
+    if st.button("Start New Story"):
+        st.session_state.story = []
+        st.experimental_rerun()
+
+# Main content area
 col1, col2 = st.columns([3, 1])
 
 with col1:
     # Story Display Box
-    st.header("Story So Far")
-    story_display = st.empty()
-    if st.session_state.story:
-        story_display.markdown('\n\n'.join(st.session_state.story))
-    else:
-        st.info("Your story will appear here as you write it.")
-
+    st.header("Your Story")
+    story_display = st.container()
+    
+    with story_display:
+        if not st.session_state.story:
+            st.info("Start your adventure by writing the first sentence!")
+        else:
+            for entry in st.session_state.story:
+                st.write(f"{entry['role']}: {entry['content']}")
+    
     # User Input Box
-    st.header("Your Turn to Write")
+    st.header("Your Turn")
     user_input = st.text_area("Continue the story:", height=100)
+    
     if st.button("Submit"):
-        if user_input.strip() == "":
-            st.warning("Please enter some text to continue the story.")
-        else:
-            st.session_state.story.append(f"**You:** {user_input}")
+        if user_input.strip():
+            # Add user's input to story
+            st.session_state.story.append({"role": "You", "content": user_input})
+            
+            # Update achievements
             update_achievements(user_input)
-            feedback = get_feedback(user_input)
-            st.session_state.story.append(f"**Feedback:** {feedback}")
-
-            ai_response = generate_ai_response(user_input, st.session_state.level)
-            st.session_state.story.append(f"**AI:** {ai_response}")
-            story_display.markdown('\n\n'.join(st.session_state.story))
+            
+            # Get feedback
+            feedback = provide_feedback(user_input, st.session_state.level)
+            st.session_state.story.append({"role": "Feedback", "content": feedback})
+            
+            # Generate AI continuation
+            ai_response = generate_story_continuation(user_input, st.session_state.level)
+            st.session_state.story.append({"role": "AI", "content": ai_response})
+            
+            st.experimental_rerun()
+        else:
+            st.warning("Please write something to continue the story.")
 
 with col2:
     # Help and Suggestions Box
-    st.header("Need Some Help?")
-    if st.button("Get Vocabulary Suggestions"):
+    st.header("Writing Help")
+    if st.button("Get Vocabulary Ideas"):
         vocab_suggestions = get_vocabulary_suggestions()
-        st.write("Try using these words in your next sentence:")
+        st.write("Try using these words:")
         for word in vocab_suggestions:
-            st.write(f"- {word}")
-
+            st.write(f"• {word}")
+    
     if st.button("Get a Creative Prompt"):
         creative_prompt = get_creative_prompt()
         st.write(creative_prompt)
-
-    # Rewards and Progress Box
-    st.header("Achievements")
+    
+    # Achievements Box
+    st.header("Your Achievements")
     if st.session_state.badges:
         for badge in st.session_state.badges:
             st.success(f"🏆 {badge}")
     else:
-        st.write("No achievements yet. Keep writing to earn badges!")
-
-    # Story Saving
-    if st.button("Save Story"):
-        story_data = {
-            'level': st.session_state.level,
-            'date': datetime.datetime.now().isoformat(),
-            'story': st.session_state.story,
-            'achievements': st.session_state.badges,
-            'total_words': st.session_state.total_words,
-            'unique_words': list(st.session_state.unique_words),
-        }
-        story_json = json.dumps(story_data, indent=4)
-        st.download_button(
-            label="Download Story",
-            data=story_json,
-            file_name='joystory.json',
-            mime='application/json',
-        )
-
-# Sidebar for selecting language level
-with st.sidebar:
-    st.header("Settings")
-    level = st.radio(
-        "Select your English level:",
-        ('Beginner', 'Intermediate', 'Advanced'),
-        index=['Beginner', 'Intermediate', 'Advanced'].index(st.session_state.level)
-    )
-    st.session_state.level = level
-    if st.button("Start New Story"):
-        st.session_state.story = []
-        st.session_state.achievements = []
-        st.session_state.unique_words = set()
-        st.session_state.total_words = 0
-        st.session_state.badges = []
-        st.experimental_rerun()
+        st.write("Keep writing to earn badges!")
+    
+    # Save Story Button
+    if st.session_state.story:
+        if st.button("Save Story"):
+            story_data = {
+                'level': st.session_state.level,
+                'date': datetime.datetime.now().isoformat(),
+                'story': st.session_state.story,
+                'achievements': st.session_state.badges,
+                'total_words': st.session_state.total_words,
+                'unique_words': list(st.session_state.unique_words)
+            }
+            st.download_button(
+                label="Download Story",
+                data=json.dumps(story_data, indent=2),
+                file_name='joystory.json',
+                mime='application/json'
+            )