import gradio as gr
from langchain.tools import Tool
from langchain_community.utilities import GoogleSearchAPIWrapper
import os

os.environ["GOOGLE_CSE_ID"] = "222a4a8ae035a4d1d"
os.environ["GOOGLE_API_KEY"] = "AIzaSyBxEUXINrQ313UZbgHEQXZSvTNADgOv9K0"

def get_search(query:str="", k:int=1):
    search = GoogleSearchAPIWrapper(k=k)
    def search_results(query):
        return search.results(query, k)
    tool = Tool(
        name="Google Search Snippets",
        description="Search Google for recent results.",
        func=search_results,
    )
    ref_text = tool.run(query)
    return ref_text 

def search(query:str):
  search_result = get_search(query,1)[0]
  title = search_result['title']
  link = search_result['link']
  return_str = f"""title: {title}\nlink: {link}"""
  return return_str

demo = gr.Interface(fn=search, inputs="textbox", outputs="textbox")
    
if __name__ == "__main__":
    demo.launch()