Update app.py

app.py

@@ -6,23 +6,28 @@ from slack_bolt import App
 import flask
 import requests
 import json
+from simple_salesforce import Salesforce
+
 app = flask.Flask(__name__, template_folder="static")
-def text_generate(prompt): 
-    url = "http://34.122.217.42/complete_batch"
-    data = {
-        "debug_ext_path": "foo", 
-        "context":prompt, 
-        "top_p": 0.9, 
-        "temp": 0.75}
+
+username=os.environ.get('SF_UserName')
+password=os.environ.get('SF_password')
+security_token=os.environ.get('security_token')
+domain=os.environ.get('domain')
+sf = Salesforce(username=username, password=password, security_token=security_token, domain=domain)
+
+    
+def search_knowledge(prompt):
+    String searchstring="Find {"+prompt+"}"
+    information = sf.search(searchstring)
+    
     try:
      response = requests.post(url, json=data)
      if response.status_code == 200:
         data_str=response.json()
         slack_msg="Supplied Prompt was :"+data["context"]+"\n\n"
         slack_msg = slack_msg+"Response receievd was : \n\n"
-        for i in range(4):
-            slack_msg = slack_msg+data_str["completions"][i]["completion"]+"\n\n"
-            slack_msg = slack_msg+data_str["completions"][i]["log_prob"]+"\n\n"
+        slack_msg = slack_msg+json.dumps(information)
      else:
         slack_msg="Failed to send JSON message to URL"
     except requests.exceptions.Timeout: