Update app.py

app.py

@@ -129,37 +129,36 @@ def WEBScout_translate():
 # Configure the API key
 palm.configure(api_key="AIzaSyDmHbZlWJJBlaW5BAJwfOu7HP-JAKFiCpY")
 
-@app.route('/chat', methods=['POST'])
-def generate():
-    data = request.get_json()
-    model = data.get('model', 'models/text-bison-001')
-    temperature = data.get('temperature', 1)
-    candidate_count = data.get('candidate_count', 1)
-    top_k = data.get('top_k', 40)
-    top_p = data.get('top_p', 0.95)
-    max_output_tokens = data.get('max_output_tokens', 1024)
-    stop_sequences = data.get('stop_sequences', [])
-    safety_settings = data.get('safety_settings', [
+defaults = {
+    'model': 'models/text-bison-001',
+    'temperature': 1,
+    'candidate_count': 1,
+    'top_k': 40,
+    'top_p': 0.95,
+    'max_output_tokens': 1024,
+    'stop_sequences': [],
+    'safety_settings': [
         {"category": "HARM_CATEGORY_DEROGATORY", "threshold": "BLOCK_NONE"},
         {"category": "HARM_CATEGORY_TOXICITY", "threshold": "BLOCK_NONE"},
         {"category": "HARM_CATEGORY_VIOLENCE", "threshold": "BLOCK_NONE"},
         {"category": "HARM_CATEGORY_SEXUAL", "threshold": "BLOCK_NONE"},
         {"category": "HARM_CATEGORY_MEDICAL", "threshold": "BLOCK_NONE"},
         {"category": "HARM_CATEGORY_DANGEROUS", "threshold": "BLOCK_NONE"},
-    ])
-
-    result = palm.generate(
-        model=model,
-        temperature=temperature,
-        candidate_count=candidate_count,
-        top_k=top_k,
-        top_p=top_p,
-        max_output_tokens=max_output_tokens,
-        stop_sequences=stop_sequences,
-        safety_settings=safety_settings
+    ]
+}
+
+@app.route('/chat', methods=['POST'])
+def generate():
+    data = request.get_json()
+    prompt = data.get('prompt', 'You are A helpfull AI that will follow users instructions.')
+
+    # Call the generate function
+    result = palm.generate_text(
+        **defaults,
+        prompt=prompt
     )
 
-    return jsonify(result)
+    return jsonify(result.result)
     
 @app.route('/gpt', methods=['POST'])
 def chat_gpt():