Update app.py

app.py

@@ -1,137 +1,26 @@
 import gradio as gr
 from transformers import pipeline
+import openai  # Import if you are using OpenAI's API
 import random
 from datetime import datetime
 
 # Initialize sentiment analysis pipeline
 sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
 
-class JournalPrompts:
-    def __init__(self):
-        self.prompts = {
-            "POSITIVE": [
-                "What moments brought you the most joy today, and why?",
-                "How did you contribute to your own happiness today?",
-                "What accomplishment, big or small, are you proud of right now?",
-                "Who made a positive impact on your day? How can you pay it forward?",
-                "What unexpected positive surprise occurred today?",
-                "How did you show kindness to yourself or others today?",
-                "What made you laugh or smile today?",
-                "What personal strength helped you succeed today?",
-                "How did you make progress toward your goals today?",
-                "What are you feeling grateful for in this moment?"
-            ],
-            "NEGATIVE": [
-                "What is weighing on your mind, and what's one small step you could take to address it?",
-                "If you could tell someone exactly how you're feeling right now, what would you say?",
-                "What would make you feel even 1% better right now?",
-                "What lesson might be hidden in this challenging situation?",
-                "How have you overcome similar challenges in the past?",
-                "What support do you need right now, and who could provide it?",
-                "If your future self could send you a message of comfort, what would they say?",
-                "What aspects of this situation are within your control?",
-                "How can you show yourself compassion during this difficult time?",
-                "What would you tell a friend who was facing this same situation?"
-            ],
-            "NEUTRAL": [
-                "What's occupying your thoughts right now?",
-                "How would you describe your energy level today?",
-                "What would make today feel more meaningful?",
-                "What patterns have you noticed in your daily life lately?",
-                "What would you like to explore or learn more about?",
-                "How aligned are your actions with your values today?",
-                "What change would you like to see in your life six months from now?",
-                "What's something you've been putting off that you could tackle today?",
-                "How have your priorities shifted recently?",
-                "What boundaries do you need to set or maintain?"
-            ],
-            "GROWTH": [
-                "What skill would you like to develop further and why?",
-                "How have your beliefs or perspectives changed recently?",
-                "What habit would you like to build or break?",
-                "What's the most important lesson you've learned this week?",
-                "How are you different now compared to a year ago?",
-                "What feedback have you received recently that resonated with you?",
-                "What's one area of your life where you feel stuck? What's keeping you there?",
-                "What would you attempt if you knew you couldn't fail?",
-                "How do you want to be remembered?",
-                "What does success mean to you right now?"
-            ],
-            "WELLNESS": [
-                "How are you taking care of your physical health today?",
-                "What does your body need right now?",
-                "How well did you sleep last night? What affected your sleep?",
-                "What activities make you feel most energized?",
-                "How do you deal with stress? What works best for you?",
-                "What self-care practice would be most helpful today?",
-                "How connected do you feel to your body right now?",
-                "What healthy boundary do you need to set?",
-                "What does balance look like in your life?",
-                "How do you recharge when you're feeling depleted?"
-            ],
-            "REFLECTION": [
-                "What patterns have you noticed in your behavior lately?",
-                "How have your priorities shifted in the past year?",
-                "What advice would you give your younger self?",
-                "What are you ready to let go of?",
-                "What beliefs about yourself are holding you back?",
-                "How do you handle uncertainty?",
-                "What role does gratitude play in your life?",
-                "What legacy do you want to leave?",
-                "How do your values guide your decisions?",
-                "What questions are you sitting with right now?"
-            ]
-        }
-
-    def get_prompt(self, category):
-        if category not in self.prompts:
-            return "Invalid category"
-        return random.choice(self.prompts[category])
-
-    def get_mixed_prompts(self, sentiment):
-        prompts = []
-        prompts.append(f"Emotional: {self.get_prompt(sentiment)}")
-        prompts.append(f"Growth: {self.get_prompt('GROWTH')}")
-        random_category = random.choice(['WELLNESS', 'REFLECTION'])
-        prompts.append(f"{random_category}: {self.get_prompt(random_category)}")
-        return prompts
-
 class JournalCompanion:
     def __init__(self):
         self.entries = []
-        self.journal_prompts = JournalPrompts()
-        self.affirmations = {
-            "POSITIVE": [
-                "You're radiating positive energy! Keep embracing joy.",
-                "Your optimism is inspiring. You're on the right path.",
-                "You have so much to be proud of. Keep shining!",
-                "Your positive mindset creates beautiful opportunities."
-            ],
-            "NEGATIVE": [
-                "It's okay to feel this way. You're stronger than you know.",
-                "Every challenge helps you grow. You've got this.",
-                "Tomorrow brings new opportunities. Be gentle with yourself.",
-                "Your feelings are valid, and this too shall pass."
-            ],
-            "NEUTRAL": [
-                "You're exactly where you need to be right now.",
-                "Your journey is unique and valuable.",
-                "Take a moment to appreciate your progress.",
-                "Every moment is a chance for a fresh perspective."
-            ]
-        }
-
+    
     def analyze_entry(self, entry_text):
         if not entry_text.strip():
             return ("Please write something in your journal entry.", "", "", "")
 
         try:
-            # Attempt to perform sentiment analysis
+            # Perform sentiment analysis
             sentiment_result = sentiment_analyzer(entry_text)[0]
             sentiment = sentiment_result["label"].upper()
             sentiment_score = sentiment_result["score"]
         except Exception as e:
-            # Print the error to the console for debugging
             print("Error during sentiment analysis:", e)
             return (
                 "An error occurred during analysis. Please try again.",
@@ -147,13 +36,44 @@ class JournalCompanion:
             "sentiment_score": sentiment_score
         }
         self.entries.append(entry_data)
-        
-        prompts = self.journal_prompts.get_mixed_prompts(sentiment)
-        affirmation = random.choice(self.affirmations.get(sentiment, ["You're doing great! Keep reflecting and growing."]))
+
+        # Generate dynamic responses using a language model
+        prompts = self.generate_dynamic_prompts(sentiment)
+        affirmation = self.generate_dynamic_affirmation(sentiment)
         sentiment_percentage = f"{sentiment_score * 100:.1f}%"
         message = f"Entry analyzed! Sentiment: {sentiment} ({sentiment_percentage} confidence)"
         
-        return message, sentiment, "\n\n".join(prompts), affirmation
+        return message, sentiment, prompts, affirmation
+
+    def generate_dynamic_prompts(self, sentiment):
+        prompt_request = f"Generate three reflective journal prompts for a person feeling {sentiment.lower()}."
+        try:
+            response = openai.Completion.create(
+                engine="gpt-3.5-turbo",
+                prompt=prompt_request,
+                max_tokens=60,
+                n=1
+            )
+            prompts = response.choices[0].text.strip()
+        except Exception as e:
+            print("Error generating prompts:", e)
+            prompts = "Could not generate prompts at this time."
+        return prompts
+
+    def generate_dynamic_affirmation(self, sentiment):
+        affirmation_request = f"Generate an affirmation for someone who is feeling {sentiment.lower()}."
+        try:
+            response = openai.Completion.create(
+                engine="gpt-3.5-turbo",
+                prompt=affirmation_request,
+                max_tokens=20,
+                n=1
+            )
+            affirmation = response.choices[0].text.strip()
+        except Exception as e:
+            print("Error generating affirmation:", e)
+            affirmation = "Could not generate an affirmation at this time."
+        return affirmation
 
     def get_monthly_insights(self):
         if not self.entries: