Update app.py

app.py

@@ -3,37 +3,16 @@ import requests
 import time 
 import os
 import json
-from bardapi import Bard
 
-bardKey = os.environ.get('_BARD_API_KEY')
+genai.configure(
+    api_key=os.environ['API_KEY'])
+model = genai.GenerativeModel(
+    model_name='gemini-pro')
+
 
 
 def bardChat(data):
-    # Create a session object using the requests library
-    session = requests.Session()
-    
-    # Set the headers for the session
-    session.headers = {
-                "Host": "bard.google.com",
-                "X-Same-Domain": "1",
-                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Safari/537.36",
-                "Content-Type": "application/x-www-form-urlencoded;charset=UTF-8",
-                "Origin": "https://bard.google.com",
-                "Referer": "https://bard.google.com/",
-            }
-    
-    # Set the "__Secure-1PSID" cookie with the Bard API key
-    session.cookies.set("__Secure-1PSID", bardKey) 
-    
-    # Create a Bard object with the session and a timeout of 30 seconds
-    bard = Bard(session=session, timeout=30)
-    answer = bard.get_answer(data)['content']
-    respo = {
-        "message": answer,
-        "action": "nothing",
-        "function": "nothing"
-    }
-    # print(answer)
+
     return json.dumps(respo)
     
 def responsenew(data):
@@ -47,7 +26,18 @@ gradio_interface = gr.Interface(
   outputs = "text" 
 )
 gradio_interface.launch()
-    
+
+
+
+
+
+
+
+
+
+
+
+
     # remind_val = ["create a reminder", "create reminder", "remind me"]
     # if remind_val in data:
     #     return "Reminder created!"