Update main.py

main.py

@@ -1,15 +1,29 @@
 import requests
 import os
 
+from langchain_core.tools import Tool
+from langchain_google_community import GoogleSearchAPIWrapper
+
+from flask import Flask
+
+search = GoogleSearchAPIWrapper()
+
+tool = Tool(
+    name="google_search",
+    description="Search Google for recent results.",
+    func=search.run,
+)
+
 API_URL = "https://api-inference.huggingface.co/models/sentence-transformers/all-MiniLM-L6-v2"
 bearer = "Bearer " + os.getenv('TOKEN')
 headers = {"Authorization": bearer }
 
-from flask import Flask
 app = Flask(__name__)
  
 @app.route('/app')
 def command_app():   
+    res = tool.run("Obama's first name?")
+    
     payload ={
     	"inputs": {
     	"source_sentence": "That is a happy person",
@@ -20,8 +34,9 @@ def command_app():
     	]
     },
     }
+    
     response = requests.post(API_URL, headers=headers, json=payload)
- 	return response.json()
+ 	return response.json() + "-" + res
 	 
 @app.route('/')
 def command_server():