Update app.py

app.py

@@ -1,84 +1,83 @@
-import random
-import streamlit as st
 import requests
-from dotenv import load_dotenv
-import os
-
-# 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
+import streamlit as st
 
-# Define the questions for mood analysis
+# Define the 3 questions for mood analysis
 questions = [
     "How are you feeling today in one word?",
     "What's currently on your mind?",
     "Do you feel calm or overwhelmed right now?",
 ]
 
-# Function to call Gemini API and get recommendations
-def get_recommendations(user_responses):
-    # Create the payload with user responses
-    data = {
-        "responses": user_responses,
+# Function to query the Gemini API
+def query_gemini_api(user_answers):
+    url = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key=AIzaSyBDo08q1vihxhI4nIHv0R6kDOPNT02NYG4'  # Replace with your actual API key
+    headers = {'Content-Type': 'application/json'}
+    
+    # Prepare the payload with user answers
+    input_text = " ".join(user_answers)  # Combining all answers into one input text
+    
+    payload = {
+        'contents': [
+            {
+                'parts': [
+                    {'text': input_text}
+                ]
+            }
+        ]
     }
 
-    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
+    try:
+        # Send the request to the Gemini API
+        response = requests.post(url, headers=headers, json=payload)
+
+        if response.status_code == 200:
+            result = response.json()
+            # Return the result from Gemini API (mood and recommendations)
+            return result
+        else:
+            print(f"Error: {response.status_code} - {response.text}")
+            return None
+    except requests.exceptions.RequestException as e:
+        print(f"An error occurred: {e}")
+        return None
+
+# Streamlit app for collecting answers
 def main():
     st.title("Mood Analysis and Suggestions")
+
     st.write("Answer the following 3 questions to help us understand your mood:")
 
-    # Collect responses
+    # Collect responses from the user
     responses = []
     for i, question in enumerate(questions):
         response = st.text_input(f"{i+1}. {question}")
         if response:
             responses.append(response)
 
-    # Analyze responses if all questions are answered
+    # If all 3 responses are collected, send them to Gemini for analysis
     if len(responses) == len(questions):
-        # 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("Processing your answers...")
+        
+        # Get mood and recommendations from Gemini API
+        result = query_gemini_api(responses)
+        
+        if result:
+            # Assuming the response contains mood and recommendations
+            mood = result.get("mood", "Unknown")
+            recommendations = result.get("recommendations", [])
+
+            # Display the detected mood
             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']})")
+            # Display the recommendations
+            if recommendations:
+                st.write("### Recommendations to Improve Your Mood:")
+                for recommendation in recommendations:
+                    st.write(f"- {recommendation}")
+            else:
+                st.write("No recommendations available.")
+        else:
+            st.write("Sorry, we couldn't process your answers. Please try again.")
     else:
         st.write("Please answer all 3 questions to receive suggestions.")