Update app.py

app.py

@@ -1,34 +1,66 @@
 import gradio as gr
 import requests
 import concurrent.futures
+import time
+import os
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from PIL import Image
+import io
 
-def open_links(link, count):
-    def open_link(url):
-        try:
-            response = requests.get(url, timeout=10)
-            return f"Status: {response.status_code}"
-        except requests.exceptions.RequestException as e:
-            return f"Error: {str(e)}"
+def capture_screenshot(url):
+    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(options=chrome_options)
+    driver.get(url)
+    time.sleep(2)  # Wait for the page to load
+    screenshot = driver.get_screenshot_as_png()
+    driver.quit()
+    
+    return Image.open(io.BytesIO(screenshot))
 
+def open_link(url, index):
+    start_time = time.time()
+    try:
+        response = requests.get(url, timeout=10)
+        elapsed_time = time.time() - start_time
+        status = f"Request {index+1}: Status {response.status_code}, Time: {elapsed_time:.2f}s, Size: {len(response.content)} bytes"
+        
+        screenshot = capture_screenshot(url)
+        return status, screenshot
+    except Exception as e:
+        return f"Request {index+1}: Error: {str(e)}", None
+
+def open_links(link, count):
     results = []
-    with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
-        futures = [executor.submit(open_link, link) for _ in range(count)]
+    screenshots = []
+    with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor:
+        futures = [executor.submit(open_link, link, i) for i in range(count)]
         for future in concurrent.futures.as_completed(futures):
-            results.append(future.result())
+            status, screenshot = future.result()
+            results.append(status)
+            if screenshot:
+                screenshots.append(screenshot)
     
-    return "\n".join(results)
+    return "\n".join(results), screenshots[0] if screenshots else None
 
 iface = gr.Interface(
     fn=open_links,
     inputs=[
         gr.Textbox(label="Enter link", placeholder="https://example.com"),
-        gr.Slider(minimum=1, maximum=50, step=1, value=5, label="Number of times to open")
+        gr.Slider(minimum=1, maximum=10, step=1, value=1, label="Number of times to open")
+    ],
+    outputs=[
+        gr.Textbox(label="Results"),
+        gr.Image(label="Screenshot")
     ],
-    outputs=gr.Textbox(label="Results"),
-    title="Link Opener",
-    description="Enter a link and choose how many times to open it. The app will attempt to open the link and return the status for each attempt.",
+    title="Link Opener with Screenshot",
+    description="Enter a link and choose how many times to open it. The app will attempt to open the link, return the status, response time, content size for each attempt, and display a screenshot of the first successful attempt.",
     theme="huggingface",
-    examples=[["https://www.example.com", 5]]
+    examples=[["https://www.example.com", 1]]
 )
 
 iface.launch()