Update app.py

app.py

@@ -22,44 +22,44 @@ questions = [
 
 # Function to query the Gemini API
 def query_gemini_api(user_answers):
-    url = f'https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key={GEMINI_API_KEY}'
+    # Correct Gemini API endpoint (based on Google's Generative AI documentation)
+    url = f"https://generativelanguage.googleapis.com/v1beta2/models/text-bison-001:generateText?key={GEMINI_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
+    # Combine the user answers into a single input text
+    input_text = " ".join(user_answers)
     
+    # Payload for the API
     payload = {
-        'contents': [
-            {
-                'parts': [
-                    {'text': input_text}
-                ]
-            }
-        ]
+        "prompt": {
+            "text": f"Analyze the following mood based on these inputs: {input_text}. Provide suggestions to improve the mood."
+        },
+        "temperature": 0.7,
+        "candidateCount": 1
     }
 
     try:
-        # Send the request to the Gemini API
+        # Send the POST request
         response = requests.post(url, headers=headers, json=payload)
 
-        # Wait for the response and process it
+        # Check if the response is successful
         if response.status_code == 200:
             result = response.json()
-
-            # Check if the response contains valid mood and recommendations
-            mood = result.get("mood", None)
-            recommendations = result.get("recommendations", None)
-
-            if mood and recommendations:
-                return mood, recommendations
+            
+            # Extract the generated text from the response
+            candidates = result.get("candidates", [])
+            if candidates:
+                generated_text = candidates[0].get("output", "")
+                return generated_text
             else:
-                return None, None
+                return None
         else:
-            print(f"Error: {response.status_code} - {response.text}")
-            return None, None
+            st.error(f"API Error {response.status_code}: {response.text}")
+            return None
     except requests.exceptions.RequestException as e:
-        print(f"An error occurred: {e}")
-        return None, None
+        st.error(f"An error occurred: {e}")
+        return None
 
 # Streamlit app for collecting answers
 def main():
@@ -78,22 +78,17 @@ def main():
     if len(responses) == len(questions):
         st.write("Processing your answers...")
 
-        # Get mood and recommendations from Gemini API
-        mood, recommendations = query_gemini_api(responses)
-
-        if mood and recommendations:
-            # Display the detected mood
-            st.write(f"Detected Mood: {mood}")
+        # Query the Gemini API
+        generated_text = query_gemini_api(responses)
 
-            # Display the recommendations
-            st.write("### Recommendations to Improve Your Mood:")
-            for recommendation in recommendations:
-                st.write(f"- {recommendation}")
+        if generated_text:
+            # Display the generated mood analysis and recommendations
+            st.write("### Mood Analysis and Suggestions:")
+            st.write(generated_text)
         else:
-            # If no valid mood or recommendations are found, do nothing
-            pass
+            st.warning("Could not generate mood analysis. Please try again later.")
     else:
-        st.write("Please answer all 3 questions to receive suggestions.")
+        st.info("Please answer all 3 questions to receive suggestions.")
 
 if __name__ == "__main__":
     main()