import gradio as gr
from search_engines import Google

search_engine = Google()

def scrape_search_results(query, num_results=5):
    try:
        results = search_engine.search(query, num_results=num_results)
        search_results = [
            {"title": result.title, "url": result.url, "description": result.description}
            for result in results
        ]
        return search_results
    except Exception as e:
        return [{"error": str(e)}]

def search_gradio(query, num_results):
    return scrape_search_results(query, num_results)

interface = gr.Interface(
    fn=search_gradio,
    inputs=[
        gr.Textbox(label="Search Query"),
        gr.Slider(label="Number of Results", minimum=1, maximum=10, step=1, default=5)
    ],
    outputs="json",
    title="Search Engine Scraper",
    description="Scrapes Google search engine results based on input query and number of results."
)

if __name__ == "__main__":
    interface.launch()