Update app.py

app.py

@@ -1,9 +1,15 @@
 import random
 import streamlit as st
-from transformers import pipeline
+import requests
+from dotenv import load_dotenv
+import os
 
-# Load the emotion analyzer model
-emotion_analyzer = pipeline("text-classification", model="distilbert-base-uncased")
+# Load environment variables from .env file
+load_dotenv()
+
+# Fetch the API key from the environment
+API_KEY = os.getenv("GEMINI_API_KEY")
+GEMINI_API_URL = "https://gemini-api-url.com"  # Replace with the actual URL for Gemini API
 
 # Define the questions for mood analysis
 questions = [
@@ -12,57 +18,27 @@ questions = [
     "Do you feel calm or overwhelmed right now?",
 ]
 
-# Define a suggestion database for different moods
-suggestion_database = {
-    "NEGATIVE": {
-        "suggestions": ["Try guided meditation", "Take a walk in nature", "Connect with a loved one"],
-        "articles": [
-            {"title": "Emotional Wellness Toolkit", "url": "https://www.nih.gov/health-information/emotional-wellness-toolkit"},
-            {"title": "Understanding Anxiety", "url": "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety"},
-        ],
-        "videos": [
-            {"title": "Mindfulness for Calm", "url": "https://youtu.be/m1vaUGtyo-A"},
-            {"title": "Relaxation Techniques", "url": "https://www.youtube.com/shorts/Tq49ajl7c8Q?feature=share"},
-        ],
-    },
-    "POSITIVE": {
-        "suggestions": ["Practice gratitude", "Engage in a hobby", "Celebrate your wins"],
-        "articles": [
-            {"title": "Benefits of Joy", "url": "https://www.health.harvard.edu/health-a-to-z"},
-            {"title": "Gratitude Practices", "url": "https://www.helpguide.org/mental-health/meditation/mindful-breathing-meditation"},
-        ],
-        "videos": [
-            {"title": "Boosting Happiness", "url": "https://youtu.be/MIc299Flibs"},
-            {"title": "Celebrating Wins", "url": "https://www.youtube.com/shorts/fwH8Ygb0K60?feature=share"},
-        ],
-    },
-    "NEUTRAL": {
-        "suggestions": ["Take a short break", "Engage in a relaxing activity", "Spend time outdoors"],
-        "articles": [
-            {"title": "Self-Care Practices", "url": "https://www.nih.gov/health-information/emotional-wellness-toolkit"},
-            {"title": "Stress Management", "url": "https://www.helpguide.org/mental-health/anxiety/tips-for-dealing-with-anxiety"},
-        ],
-        "videos": [
-            {"title": "Relaxation Exercises", "url": "https://youtu.be/Y8HIFRPU6pM"},
-            {"title": "Mindfulness Tips", "url": "https://youtu.be/-e-4Kx5px_I"},
-        ],
-    },
-}
-
-# Function to fetch relevant resources based on detected mood
-def get_relevant_resources(mood):
-    resources = suggestion_database.get(mood, {})
-    return resources.get("suggestions", []), resources.get("articles", []), resources.get("videos", [])
-
-# Function to suggest activities based on the mood
-def suggest_activity(mood):
-    suggestions, articles, videos = get_relevant_resources(mood)
-    return {
-        "suggestions": suggestions,
-        "articles": articles,
-        "videos": videos,
+# Function to call Gemini API and get recommendations
+def get_recommendations(user_responses):
+    # Create the payload with user responses
+    data = {
+        "responses": user_responses,
+    }
+
+    headers = {
+        "Authorization": f"Bearer {API_KEY}",
+        "Content-Type": "application/json"
     }
 
+    # Call the Gemini API
+    response = requests.post(GEMINI_API_URL, json=data, headers=headers)
+
+    if response.status_code == 200:
+        return response.json()  # Return the JSON response containing suggestions and mood analysis
+    else:
+        st.error(f"Error in calling Gemini API: {response.status_code}")
+        return {}
+
 # Streamlit app
 def main():
     st.title("Mood Analysis and Suggestions")
@@ -77,39 +53,32 @@ def main():
 
     # Analyze responses if all questions are answered
     if len(responses) == len(questions):
-        combined_text = " ".join(responses)
-        analysis_result = emotion_analyzer(combined_text)
-        detected_emotion = analysis_result[0]['label']
-
-        # Map detected emotion to a mood state
-        mood_mapping = {
-            "LABEL_0": "NEGATIVE",  # for negative emotions
-            "LABEL_1": "POSITIVE",  # for positive emotions
-            "LABEL_2": "NEUTRAL",   # for neutral emotions
-        }
-
-        # Map the detected emotion into a proper mood
-        mood = mood_mapping.get(detected_emotion, "NEUTRAL")
-
-        st.write(f"Detected Mood: {mood}")
-
-        # Fetch suggestions based on mood
-        resources = suggest_activity(mood)
-
-        # Display suggestions
-        st.write("### Suggestions")
-        for suggestion in resources["suggestions"]:
-            st.write(f"- {suggestion}")
-
-        # Display articles
-        st.write("### Articles")
-        for article in resources["articles"]:
-            st.write(f"- [{article['title']}]({article['url']})")
-
-        # Display videos
-        st.write("### Videos")
-        for video in resources["videos"]:
-            st.write(f"- [{video['title']}]({video['url']})")
+        # Send the responses to the Gemini API for analysis and suggestions
+        analysis_result = get_recommendations(responses)
+
+        if analysis_result:
+            # Extract mood and recommendations from the response
+            mood = analysis_result.get("mood", "NEUTRAL")
+            suggestions = analysis_result.get("suggestions", [])
+            articles = analysis_result.get("articles", [])
+            videos = analysis_result.get("videos", [])
+
+            st.write(f"Detected Mood: {mood}")
+
+            # Display suggestions
+            st.write("### Suggestions")
+            for suggestion in suggestions:
+                st.write(f"- {suggestion}")
+
+            # Display articles
+            st.write("### Articles")
+            for article in articles:
+                st.write(f"- [{article['title']}]({article['url']})")
+
+            # Display videos
+            st.write("### Videos")
+            for video in videos:
+                st.write(f"- [{video['title']}]({video['url']})")
     else:
         st.write("Please answer all 3 questions to receive suggestions.")