Update app.py

app.py

@@ -1,6 +1,10 @@
 import streamlit as st
 import datetime
 import time
+import openai
+
+# Set up OpenAI API key
+openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 # Task Management
 tasks = []
@@ -173,6 +177,45 @@ def resources():
         unsafe_allow_html=True,
     )
 
+# AI-based Personalization
+def get_chatgpt_response(prompt):
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant providing personalized recommendations for managing ADHD."},
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return response.choices[0].message['content'].strip()
+
+def personalization():
+    st.markdown(
+        f"""
+        <div class="main">
+            <h2>Personalization</h2>
+            <p>Get personalized recommendations for managing ADHD based on your responses.</p>
+        </div>
+        """,
+        unsafe_allow_html=True,
+    )
+
+    age = st.number_input("What is your age?", min_value=1, max_value=100, value=25, step=1)
+    occupation = st.text_input("What is your occupation?")
+    challenges = st.text_area("What are the main challenges you face with ADHD?")
+
+    if st.button("Get Personalized Recommendations"):
+        prompt = f"I am {age} years old, working as a {occupation}. The main challenges I face with ADHD are: {challenges}. Can you provide personalized recommendations for managing ADHD in my daily life and work?"
+        recommendations = get_chatgpt_response(prompt)
+        st.markdown(
+            f"""
+            <div class="main">
+                <h3>Personalized Recommendations:</h3>
+                <p>{recommendations}</p>
+            </div>
+            """,
+            unsafe_allow_html=True,
+        )
+
 def main():
     st.set_page_config(page_title="ADHD Toolkit", layout="wide")
 
@@ -196,7 +239,7 @@ def main():
         """,
         unsafe_allow_html=True,
     )
-    menu = ["Home", "Task Management", "Time Management", "Habit Tracking", "Workplace Strategies", "Resources"]
+    menu = ["Home", "Task Management", "Time Management", "Habit Tracking", "Workplace Strategies", "Resources", "Personalization"]
     choice = st.sidebar.selectbox("Select a section", menu)
 
     # Set up main content area
@@ -234,6 +277,8 @@ def main():
         workplace_strategies()
     elif choice == "Resources":
         resources()
+    elif choice == "Personalization":
+        personalization()
 
 if __name__ == "__main__":
     main()