Create app.py

app.py

@@ -0,0 +1,39 @@
+import gradio as gr
+import httpx
+from PIL import Image
+from io import BytesIO
+import os
+
+SCRAPFLY_API_KEY = os.getenv("SCRAPFLY_API_KEY")  # Use HF Spaces secrets
+
+client = httpx.Client(
+    timeout=60,
+    headers={'accept-encoding': 'gzip'},
+    params={"key": SCRAPFLY_API_KEY}
+)
+
+def take_screenshot(url, capture="fullpage"):
+    try:
+        response = client.get(
+            'https://api.scrapfly.io/screenshot',
+            params={"url": url, "capture": capture}
+        )
+        response.raise_for_status()
+        img = Image.open(BytesIO(response.content))
+        return img
+    except Exception as e:
+        return f"Error: {e}"
+
+demo = gr.Interface(
+    fn=take_screenshot,
+    inputs=[
+        gr.Textbox(label="Enter URL", placeholder="https://example.com"),
+        gr.Radio(["fullpage", "viewport"], label="Capture Type", value="fullpage")
+    ],
+    outputs=gr.Image(type="pil", label="Screenshot"),
+    title="Website Screenshot Tool",
+    description="Takes screenshots of any website using ScrapFly API."
+)
+
+if __name__ == "__main__":
+    demo.launch()