Update app.py

app.py

@@ -1,8 +1,9 @@
-
 import gradio as gr
 import requests, time, socket, ssl, random
 from ping3 import ping
 import dns.resolver
+from bs4 import BeautifulSoup
+
 
 def dns_lookup(domain):
     try:
@@ -12,6 +13,7 @@ def dns_lookup(domain):
     except:
         return None
 
+
 def ssl_handshake(domain):
     try:
         ctx = ssl.create_default_context()
@@ -24,6 +26,7 @@ def ssl_handshake(domain):
     except:
         return None
 
+
 def test_speed(url):
     if not url.startswith("http"):
         url = "http://" + url
@@ -39,14 +42,13 @@ def test_speed(url):
     p = ping(domain, timeout=1)
     result["Ping (ms)"] = round(p * 1000, 2) if p else None
 
-    # SSL handshake
+    # SSL
     ssl_time = ssl_handshake(domain)
     result["SSL Handshake (ms)"] = ssl_time
 
     try:
         session = requests.Session()
-        # Random query param to bypass cache
-        cache_bypass_url = url + ("?" + str(random.randint(100000, 999999)))
+        cache_bypass_url = url + "?" + str(random.randint(100000, 999999))
 
         start_ttfb = time.time()
         r = session.get(cache_bypass_url, stream=True, timeout=20)
@@ -57,7 +59,7 @@ def test_speed(url):
         result["Final URL"] = r.url
         result["Redirect Chain"] = [resp.url for resp in r.history] + [r.url]
 
-        # Full download
+        # Download
         start_dl = time.time()
         content = r.content
         download_time = (time.time() - start_dl) * 1000
@@ -65,24 +67,37 @@ def test_speed(url):
         result["Download Time (ms)"] = round(download_time, 2)
         result["Page Size (KB)"] = round(len(content) / 1024, 2)
 
-        # Approximate Speed Index
-        # Very rough approximation: DNS + SSL + TTFB + 50% of download time
-        speed_index = sum(x for x in [dns_time, ssl_time, ttfb, download_time*0.5] if x)
+        # ---- ASSET COUNT ----
+        soup = BeautifulSoup(content, "html.parser")
+
+        images = soup.find_all("img")
+        videos = soup.find_all("video")
+        audios = soup.find_all("audio")
+
+        result["Images Count"] = len(images)
+        result["Videos Count"] = len(videos)
+        result["Audios Count"] = len(audios)
+        result["Total Assets"] = len(images) + len(videos) + len(audios)
+
+        # Speed index
+        speed_index = sum(
+            x for x in [dns_time, ssl_time, ttfb, download_time * 0.5] if x
+        )
         result["Speed Index (ms)"] = round(speed_index, 2)
         result["Speed Index (s)"] = round(speed_index / 1000, 2)
 
     except Exception as e:
         return {"Error": str(e)}
 
-    # Simple speed rating based on TTFB
+    # Rating
     if ttfb < 200:
         rating = "Fast"
     elif ttfb < 600:
         rating = "Medium"
     else:
         rating = "Slow"
-    result["Speed Rating"] = rating
 
+    result["Speed Rating"] = rating
     return result