Update app.py

app.py

@@ -1,18 +1,16 @@
 import streamlit as st
-import plotly.express as px
+import openai
 import pandas as pd
+import plotly.express as px
 from datetime import datetime
-import json
-from pathlib import Path
+import os
 import random
 
+# --- GPT-4 API Configuration ---
+openai.api_key = os.getenv("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")
+
 # --- Page Configuration ---
-st.set_page_config(
-    page_title="Emma - Mental Health Companion",
-    page_icon="🌟",
-    layout="wide",
-    initial_sidebar_state="expanded"
-)
+st.set_page_config(page_title="Emma - Mental Health Companion", page_icon="🌟", layout="wide")
 
 # --- Session State Initialization ---
 if 'user_profile' not in st.session_state:
@@ -22,88 +20,48 @@ if 'chat_history' not in st.session_state:
 if 'current_page' not in st.session_state:
     st.session_state.current_page = 'home'
 
-# --- Utility Functions ---
-def save_user_profile(profile_data):
-    """Save user profile to session state and local storage"""
-    st.session_state.user_profile = profile_data
-    Path("data").mkdir(exist_ok=True)
-    with open('data/user_profile.json', 'w') as f:
-        json.dump(profile_data, f)
-
-def load_user_profile():
-    """Load user profile from local storage"""
+# --- Helper Functions ---
+def get_gpt4_response(prompt):
+    """Get response from GPT-4"""
     try:
-        with open('data/user_profile.json', 'r') as f:
-            return json.load(f)
-    except:
-        return None
+        response = openai.ChatCompletion.create(
+            model="gpt-4",
+            messages=[
+                {"role": "system", "content": "You are a helpful mental health companion named Emma."},
+                {"role": "user", "content": prompt}
+            ]
+        )
+        return response['choices'][0]['message']['content']
+    except Exception as e:
+        st.error("Error with GPT-4 API. Please check your API key or connection.")
+        return "I'm having trouble responding at the moment. Please try again later."
 
-# --- Navigation ---
-def navigation():
-    with st.sidebar:
-        st.title("Emma 🌟")
-        pages = {
-            "Home": "home",
-            "Chat Interface": "chat",
-            "Progress Dashboard": "progress",
-            "Support Tools": "tools",
-            "Settings": "settings"
-        }
-        
-        for page_name, page_id in pages.items():
-            if st.button(page_name):
-                st.session_state.current_page = page_id
+# --- Onboarding and User Profile ---
+def save_user_profile(profile_data):
+    """Save user profile to session state"""
+    st.session_state.user_profile = profile_data
 
-# --- Home Page ---
-def home_page():
+def onboarding_page():
     if st.session_state.user_profile is None:
         st.title("Welcome to Emma 🌟")
         st.write("Your personal mental health companion")
-        
+
         with st.form("onboarding_form"):
-            st.subheader("Let's personalize your experience")
             name = st.text_input("What should I call you?")
-            
-            st.subheader("What challenges would you like support with?")
             challenges = st.multiselect(
-                "Select all that apply:",
+                "What challenges would you like support with?",
                 ["Anxiety", "Depression", "Stress", "Sleep Issues", "Relationship Issues", "Work-Life Balance"]
             )
-            
-            st.subheader("Customize your therapeutic experience")
-            approach_style = st.slider(
-                "Preferred approach style:",
-                min_value=0,
-                max_value=10,
-                value=5,
-                help="0: More Reassuring, 10: More Solution-Oriented"
-            )
-            
-            focus_style = st.slider(
-                "Focus preference:",
-                min_value=0,
-                max_value=10,
-                value=5,
-                help="0: Holistic Well-being, 10: Targeted Issue Resolution"
-            )
-            
-            tone_style = st.slider(
-                "Preferred tone:",
-                min_value=0,
-                max_value=10,
-                value=5,
-                help="0: Lighthearted, 10: Serious"
-            )
-            
+            approach_style = st.slider("Preferred approach style (0: Reassuring, 10: Solution-Oriented):", 0, 10, 5)
+            tone_style = st.slider("Preferred tone (0: Lighthearted, 10: Serious):", 0, 10, 5)
+
             submitted = st.form_submit_button("Start My Journey")
-            
             if submitted:
                 profile_data = {
                     "name": name,
                     "challenges": challenges,
                     "preferences": {
                         "approach_style": approach_style,
-                        "focus_style": focus_style,
                         "tone_style": tone_style
                     },
                     "created_at": str(datetime.now())
@@ -113,49 +71,29 @@ def home_page():
                 st.rerun()
     else:
         st.title(f"Welcome back, {st.session_state.user_profile['name']}! 🌟")
-        st.write("How are you feeling today?")
-        
-        # Quick mood check-in
-        mood = st.select_slider(
-            "Rate your current mood:",
-            options=["😢", "😕", "😐", "🙂", "😊"],
-            value="😐"
-        )
-        
-        if st.button("Start Chat Session"):
+        if st.button("Continue to Chat"):
             st.session_state.current_page = "chat"
-            st.rerun()
 
 # --- Chat Interface ---
-def chat_interface():
+def chat_page():
     st.title("Chat with Emma 💭")
-    
-    # Display chat history
+
     for message in st.session_state.chat_history:
         with st.chat_message(message["role"]):
             st.write(message["content"])
-    
-    # Chat input
-    if prompt := st.chat_input("Type your message here..."):
-        # Add user message to chat history
+
+    prompt = st.chat_input("Type your message here...")
+    if prompt:
         st.session_state.chat_history.append({"role": "user", "content": prompt})
-        
-        # Simulate Emma's response (replace with actual AI integration)
-        responses = [
-            "I hear you. Can you tell me more about how that makes you feel?",
-            "That sounds challenging. Let's explore this together.",
-            "I understand this is difficult. What support do you need right now?",
-            "You're showing great courage in sharing this. How can I help?"
-        ]
-        emma_response = random.choice(responses)
+        emma_response = get_gpt4_response(prompt)
         st.session_state.chat_history.append({"role": "assistant", "content": emma_response})
-        st.rerun()
+        st.experimental_rerun()
 
 # --- Progress Dashboard ---
 def progress_dashboard():
     st.title("Your Progress Dashboard 📊")
     
-    # Sample mood data (replace with actual user data)
+    # Sample mood data
     dates = pd.date_range(start="2024-01-01", end="2024-01-07", freq="D")
     moods = [random.randint(1, 5) for _ in range(len(dates))]
     
@@ -163,130 +101,57 @@ def progress_dashboard():
         "Date": dates,
         "Mood": moods
     })
-    
-    # Mood tracking chart
+
     fig = px.line(df, x="Date", y="Mood", title="Your Mood Timeline")
     st.plotly_chart(fig)
-    
-    # Weekly insights
-    st.subheader("Weekly Insights")
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        st.metric("Average Mood", f"{sum(moods)/len(moods):.1f}")
-        st.metric("Sessions Completed", len(st.session_state.chat_history) // 2)
-    
-    with col2:
-        st.metric("Mood Trend", "↗️ Improving")
-        st.metric("Engagement", "Regular")
 
 # --- Support Tools ---
 def support_tools():
     st.title("Support Tools & Resources 🛠️")
-    
-    tab1, tab2, tab3 = st.tabs(["Meditation", "Breathing Exercises", "Coping Strategies"])
-    
-    with tab1:
-        st.subheader("Guided Meditation")
-        st.write("Choose a meditation to begin:")
-        meditation_type = st.selectbox(
-            "Select meditation type:",
-            ["Stress Relief", "Mindfulness", "Sleep", "Anxiety Relief"]
-        )
-        if st.button("Start Meditation"):
-            st.write("🎵 Meditation audio would play here")
-    
-    with tab2:
-        st.subheader("Breathing Exercises")
-        st.write("Try this simple breathing exercise:")
-        if st.button("Start Breathing Exercise"):
-            st.write("Breathe in... 2... 3... 4...")
-            st.write("Hold... 2... 3... 4...")
-            st.write("Breathe out... 2... 3... 4...")
-    
-    with tab3:
-        st.subheader("Coping Strategies")
-        strategies = [
-            "Practice mindfulness",
-            "Write in a journal",
-            "Talk to a friend",
-            "Exercise",
-            "Practice gratitude"
-        ]
-        st.write("Here are some strategies you can try:")
-        for strategy in strategies:
-            st.checkbox(strategy)
+    st.subheader("Guided Meditation")
+    st.write("Select a meditation type to begin.")
+    meditation_type = st.selectbox("Meditation Type", ["Stress Relief", "Mindfulness", "Sleep"])
+    if st.button("Play Meditation"):
+        st.write(f"Playing {meditation_type} meditation 🎵")
 
 # --- Settings Page ---
 def settings_page():
     st.title("Settings ⚙️")
-    
     if st.session_state.user_profile:
         st.subheader("Therapist Preferences")
-        
-        # Update preferences
-        new_approach = st.slider(
-            "Approach Style:",
-            min_value=0,
-            max_value=10,
-            value=st.session_state.user_profile["preferences"]["approach_style"]
-        )
-        
-        new_focus = st.slider(
-            "Focus Style:",
-            min_value=0,
-            max_value=10,
-            value=st.session_state.user_profile["preferences"]["focus_style"]
-        )
-        
-        new_tone = st.slider(
-            "Tone Style:",
-            min_value=0,
-            max_value=10,
-            value=st.session_state.user_profile["preferences"]["tone_style"]
-        )
-        
+        new_approach = st.slider("Approach Style", 0, 10, st.session_state.user_profile["preferences"]["approach_style"])
+        new_tone = st.slider("Tone Style", 0, 10, st.session_state.user_profile["preferences"]["tone_style"])
         if st.button("Update Preferences"):
             st.session_state.user_profile["preferences"].update({
                 "approach_style": new_approach,
-                "focus_style": new_focus,
                 "tone_style": new_tone
             })
-            save_user_profile(st.session_state.user_profile)
-            st.success("Preferences updated successfully!")
-        
-        # Privacy settings
-        st.subheader("Privacy Settings")
-        if st.button("Download My Data"):
-            st.download_button(
-                "Click to Download",
-                data=json.dumps(st.session_state.user_profile),
-                file_name="my_emma_data.json"
-            )
-        
-        if st.button("Delete My Data"):
-            if st.confirm("Are you sure? This action cannot be undone."):
-                st.session_state.user_profile = None
-                st.session_state.chat_history = []
-                Path("data/user_profile.json").unlink(missing_ok=True)
-                st.success("All data deleted successfully!")
-                st.rerun()
+            st.success("Preferences updated!")
+
+# --- Page Navigation ---
+def navigation():
+    with st.sidebar:
+        st.title("Emma 🌟")
+        if st.button("Home"):
+            st.session_state.current_page = 'home'
+        if st.button("Chat"):
+            st.session_state.current_page = 'chat'
+        if st.button("Progress Dashboard"):
+            st.session_state.current_page = 'progress'
+        if st.button("Support Tools"):
+            st.session_state.current_page = 'tools'
+        if st.button("Settings"):
+            st.session_state.current_page = 'settings'
 
 # --- Main App Logic ---
 def main():
     navigation()
-    
-    # Load user profile if exists
-    if st.session_state.user_profile is None:
-        profile = load_user_profile()
-        if profile:
-            st.session_state.user_profile = profile
-    
-    # Page routing
+
+    # Load pages based on navigation
     if st.session_state.current_page == "home":
-        home_page()
+        onboarding_page()
     elif st.session_state.current_page == "chat":
-        chat_interface()
+        chat_page()
     elif st.session_state.current_page == "progress":
         progress_dashboard()
     elif st.session_state.current_page == "tools":
@@ -295,4 +160,4 @@ def main():
         settings_page()
 
 if __name__ == "__main__":
-    main()
+    main()