Update app.py

app.py

@@ -22,8 +22,8 @@ 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/v1beta2/models/text-bison-001:generateText?key={GEMINI_API_KEY}"
+    # Correct Gemini API endpoint
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/text-bison-001:generateText?key={GEMINI_API_KEY}"
     
     headers = {'Content-Type': 'application/json'}
     
@@ -32,15 +32,13 @@ def query_gemini_api(user_answers):
     
     # Payload for the API
     payload = {
-        "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
-        }
+        "prompt": {
+            "text": f"Analyze the following mood based on these inputs: {input_text}. Provide suggestions to improve the mood."
+        },
+        "temperature": 0.7,
+        "maxOutputTokens": 256,
+        "topP": 0.8,
+        "topK": 40
     }
 
     try:
@@ -52,12 +50,8 @@ def query_gemini_api(user_answers):
             result = response.json()
             
             # Extract the generated text from the response
-            predictions = result.get("predictions", [])
-            if predictions:
-                generated_text = predictions[0].get("content", "")
-                return generated_text
-            else:
-                return None
+            generated_text = result.get("candidates", [{}])[0].get("output", "")
+            return generated_text
         else:
             st.error(f"API Error {response.status_code}: {response.text}")
             return None