Update app.py

app.py

@@ -23,7 +23,7 @@ questions = [
 # Function to query the Gemini API
 def query_gemini_api(user_answers):
     # Correct Gemini API endpoint (based on Google's Generative AI documentation)
-    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key={GEMINI_API_KEY}"
+    url = f"https://generativelanguage.googleapis.com/v1beta2/models/text-bison-001:generateText?key={GEMINI_API_KEY}"
     
     headers = {'Content-Type': 'application/json'}
     
@@ -32,11 +32,15 @@ def query_gemini_api(user_answers):
     
     # Payload for the API
     payload = {
-        "prompt": {
-            "text": f"Analyze the following mood based on these inputs: {input_text}. Provide suggestions to improve the mood."
-        },
-        "temperature": 0.7,
-        "candidateCount": 1
+        "instances": [
+            {
+                "content": f"Analyze the following mood based on these inputs: {input_text}. Provide suggestions to improve the mood."
+            }
+        ],
+        "parameters": {
+            "temperature": 0.7,
+            "maxOutputTokens": 256
+        }
     }
 
     try:
@@ -48,9 +52,9 @@ def query_gemini_api(user_answers):
             result = response.json()
             
             # Extract the generated text from the response
-            candidates = result.get("candidates", [])
-            if candidates:
-                generated_text = candidates[0].get("output", "")
+            predictions = result.get("predictions", [])
+            if predictions:
+                generated_text = predictions[0].get("content", "")
                 return generated_text
             else:
                 return None