Update app.py

app.py

@@ -1,37 +1,29 @@
 import gradio as gr
-from selenium import webdriver
-from selenium.webdriver.chrome.options import Options
-from webdriver_manager.chrome import ChromeDriverManager
-import time
+from extract import take_webdata
+from PIL import Image
 
-def capture_screenshot(url):
-    options = Options()
-    options.add_argument("--headless")
-    options.add_argument("--no-sandbox")
-    options.add_argument("--disable-dev-shm-usage")
-    options.add_argument("--window-size=1920,1080")
-
-    driver = webdriver.Chrome(ChromeDriverManager().install(), options=options)
-
-    screenshot_path = "screenshot.png"
+def visualize(url):
     try:
-        driver.get(url)
-        time.sleep(2)
-        driver.save_screenshot(screenshot_path)
+        html_image, html_content = take_webdata(url)
+        if not html_content:
+            return None, "❌ Error: empty HTML content"
+        if not html_image:
+            return None, "❌ Error: empty image preview"
+        return html_image, html_content
     except Exception as e:
-        return f"Error: {str(e)}"
-    finally:
-        driver.quit()
+        return None, f"❌ Error: {str(e)}"
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🌐 Website Content Extractor")
+    gr.Markdown("Enter a URL below to fetch and preview website content.")
+
+    url_input = gr.Textbox(label="Enter a URL")
+    submit_button = gr.Button("Proceed")
 
-    return screenshot_path
+    with gr.Row():
+        img_output = gr.Image(label="Website Preview", type="pil")
+        text_output = gr.Textbox(label="Website Title/Content")
 
-demo = gr.Interface(
-    fn=capture_screenshot,
-    inputs=gr.Textbox(label="Enter URL"),
-    outputs=gr.Image(label="Screenshot"),
-    title="Web Page Screenshot Tool",
-    description="Enter a URL and get a screenshot using headless Chrome.",
-)
+    submit_button.click(fn=visualize, inputs=[url_input], outputs=[img_output, text_output])
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()