Create app.py

app.py

@@ -0,0 +1,97 @@
+import gradio as gr
+from selenium import webdriver
+from selenium.webdriver.chrome.service import Service
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
+from webdriver_manager.chrome import ChromeDriverManager
+import time
+
+def _make_driver():
+    chrome_options = Options()
+    chrome_options.add_argument("--headless")
+    chrome_options.add_argument("--no-sandbox")
+    chrome_options.add_argument("--disable-dev-shm-usage")
+    driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options)
+    driver.set_page_load_timeout(15)
+    return driver
+
+def browse_and_extract(url: str, selector: str = "body") -> str:
+    """Open a webpage and extract text of a CSS selector."""
+    try:
+        driver = _make_driver()
+        driver.get(url)
+        elem = driver.find_element(By.CSS_SELECTOR, selector)
+        text = elem.text
+        driver.quit()
+        return text
+    except Exception as e:
+        return f"Error: {e}"
+
+def screenshot(url: str) -> str:
+    """Take a screenshot of a webpage."""
+    try:
+        driver = _make_driver()
+        driver.get(url)
+        path = "/tmp/screenshot.png"
+        driver.save_screenshot(path)
+        driver.quit()
+        return path
+    except Exception as e:
+        return f"Error: {e}"
+
+def click(url: str, selector: str) -> str:
+    """Open a webpage, click an element, and return the new page title."""
+    try:
+        driver = _make_driver()
+        driver.get(url)
+        elem = driver.find_element(By.CSS_SELECTOR, selector)
+        elem.click()
+        time.sleep(2)  # let navigation happen
+        title = driver.title
+        driver.quit()
+        return f"Clicked {selector}, new title: {title}"
+    except Exception as e:
+        return f"Error: {e}"
+
+def fill(url: str, selector: str, text: str) -> str:
+    """Open a webpage, fill an input box, and return page title."""
+    try:
+        driver = _make_driver()
+        driver.get(url)
+        elem = driver.find_element(By.CSS_SELECTOR, selector)
+        elem.clear()
+        elem.send_keys(text)
+        driver.quit()
+        return f"Filled {selector} with '{text}'"
+    except Exception as e:
+        return f"Error: {e}"
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🌐 MCP Browser (Selenium + Chromium)")
+
+    with gr.Tab("Extract Text"):
+        url_in = gr.Textbox(label="URL")
+        sel_in = gr.Textbox(label="CSS Selector", value="body")
+        out = gr.Textbox(label="Extracted Text", lines=15)
+        gr.Button("Run").click(browse_and_extract, [url_in, sel_in], out)
+
+    with gr.Tab("Screenshot"):
+        url_in2 = gr.Textbox(label="URL")
+        out_img = gr.Image(label="Screenshot")
+        gr.Button("Take Screenshot").click(screenshot, url_in2, out_img)
+
+    with gr.Tab("Click"):
+        url_in3 = gr.Textbox(label="URL")
+        sel_in3 = gr.Textbox(label="CSS Selector")
+        out3 = gr.Textbox(label="Result")
+        gr.Button("Click").click(click, [url_in3, sel_in3], out3)
+
+    with gr.Tab("Fill"):
+        url_in4 = gr.Textbox(label="URL")
+        sel_in4 = gr.Textbox(label="CSS Selector")
+        txt_in4 = gr.Textbox(label="Text to Fill")
+        out4 = gr.Textbox(label="Result")
+        gr.Button("Fill").click(fill, [url_in4, sel_in4, txt_in4], out4)
+
+if __name__ == "__main__":
+    demo.launch(mcp_server=True)