Update app.py

app.py

@@ -2,34 +2,17 @@ import gradio
 import requests
 import os
 import json
-from hugchat import hugchat
 from bardapi import Bard
 
 auth = os.environ.get('Authorization')
+bardKey = os.environ.get('_BARD_API_KEY')
 API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-xl"
 headers = {"Authorization": auth}
 
-#Replace XXXX with the values you get from __Secure-1PSID 
-os.environ['_BARD_API_KEY']="YQjBGmAPwuH18OslktGRMs3XDGJ8DFlZf4ruCDZe3WOG7Ewk8tiZTnsduSHSs7ErGiIsDA."
 
 
 
 
-def huggChat(data):
-    # start a new huggingchat connection
-    chatbot = hugchat.ChatBot(cookie_path="cookies.json")
-    
-    # start a new conversation
-    id = chatbot.new_conversation()
-    chatbot.change_conversation(id)
-    
-    
-    # print the response
-    
-    result = chatbot.chat(data)
-    print(result)
-    return 'Hello Sir'
-
 def bardChat(data):
     # Create a session object using the requests library
     session = requests.Session()
@@ -45,12 +28,16 @@ def bardChat(data):
             }
     
     # Set the "__Secure-1PSID" cookie with the Bard API key
-    session.cookies.set("__Secure-1PSID", os.getenv("_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']
+    print(answer)
+    
+    return json.dumps({'message':answer,'action':'null'})
+    
     
-    print(bard.get_answer(data)['content'])
 
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
@@ -62,7 +49,7 @@ def query(payload):
 
 def responsenew(data):
     return bardChat(data)
-    #return huggChat(data)
+   
     #return query({"inputs": "The answer to the universe is",})