Update app.py

app.py

@@ -1,4 +1,4 @@
-import google.generativeai as palm
+import google.generativeai as genai
 import gradio as gr
 import os
 import json
@@ -8,7 +8,11 @@ import uuid
 import re
 
 # Set your API key
-palm.configure(api_key=os.environ["PALM_KEY"])
+# Or use `os.getenv('GOOGLE_API_KEY')` to fetch an environment variable.
+GOOGLE_API_KEY=userdata.get('GEMINI_KEY')
+
+genai.configure(api_key=GOOGLE_API_KEY)
+model = genai.GenerativeModel('gemini-pro')
 
 # Select the PaLM 2 model
 # model = 'models/text-bison-001'
@@ -57,8 +61,8 @@ def responsenew(data):
         print(text)
     except json.JSONDecodeError as e:
         print("Invalid JSON:", e)
-    response = palm.chat( messages=text)
-    print(f"\n{response.last}")
+    response = model.generate_content( text)
+    print(f"\n{response.text}")
     if text is not None:
         if "remind me" in text:
             values = getValues(text)
@@ -231,7 +235,7 @@ def responsenew(data):
             }
         else:
             respo = {
-                "message": response.last,
+                "message": response.text,
                 "action": "nothing",
                 "function": "nothing",
             }