Update app.py

app.py

@@ -1,33 +1,64 @@
+# AdVision AI: Simple Ad Detector & Suggestion Tool for Hugging Face
+
 import gradio as gr
+from PIL import Image, ImageDraw
+import torch
 import requests
-from bs4 import BeautifulSoup
-
-def analyze_ads(url):
-    try:
-        html = requests.get(url, timeout=5).text
-        soup = BeautifulSoup(html, "html.parser")
+from io import BytesIO
+import numpy as np
+from ultralytics import YOLO
 
-        ads = soup.find_all(lambda tag: tag.name in ["iframe", "div", "img"]
-                            and any(k in (tag.get("id", "") + str(tag.get("class", ""))).lower()
-                                    for k in ["ad", "banner", "sponsor", "promo", "google"]))
+# Load YOLOv8 model (assumes pretrained for ad/banner detection)
+# You can also use your custom model from Roboflow or YOLO
+model = YOLO("yolov8n.pt")  # Change to custom ad-detection model if available
 
-        total_ads = len(ads)
+# Function to analyze ad positions and give suggestions
+def analyze_ad_positions(detections, image_height):
+    insights = []
+    for box in detections:
+        x1, y1, x2, y2 = box[:4]
+        center_y = (y1 + y2) / 2
 
-        if total_ads == 0:
-            return "📭 Koi ad nahi mila. Aap top ya sidebar par ad laga sakte hain."
-        elif total_ads <= 2:
-            return f"🧐 {total_ads} ad mila. Thoda aur visible jagah pe ads lagane ki zarurat hai."
+        if center_y < image_height * 0.33:
+            pos = "Top"
+            suggestion = "Good visibility ✅"
+        elif center_y < image_height * 0.66:
+            pos = "Middle"
+            suggestion = "Moderate visibility ⚠️ Consider moving up"
         else:
-            return f"✅ {total_ads} ads mile. Achha coverage hai! Par top section par focus karein."
-
-    except Exception as e:
-        return "❌ Galat URL ya website load nahi hui."
-
-# Gradio interface
-gr.Interface(
-    fn=analyze_ads,
-    inputs=gr.Textbox(lines=1, placeholder="Apni website ka URL daalein..."),
-    outputs="text",
-    title="📊 AdVision AI",
-    description="Website ad visibility checker. Apne site ka URL dijiye aur ad analysis lijiye!"
-).launch()
+            pos = "Bottom"
+            suggestion = "Low visibility ❌ Move above the fold"
+
+        insights.append(f"Ad at {pos} → {suggestion}")
+    return insights
+
+# Main prediction function
+def detect_ads(image):
+    img = image.convert("RGB")
+    image_array = np.array(img)
+
+    results = model(image_array)[0]  # First result
+    detections = []
+    draw = ImageDraw.Draw(img)
+
+    for box in results.boxes.xyxy:
+        x1, y1, x2, y2 = map(int, box.tolist())
+        draw.rectangle([x1, y1, x2, y2], outline="red", width=3)
+        draw.text((x1, y1 - 10), "Ad", fill="red")
+        detections.append((x1, y1, x2, y2))
+
+    suggestions = analyze_ad_positions(detections, img.height)
+    suggestions_text = "\n".join(suggestions) if suggestions else "No ads detected."
+
+    return img, suggestions_text
+
+# Gradio Interface
+interface = gr.Interface(
+    fn=detect_ads,
+    inputs=gr.Image(type="pil"),
+    outputs=[gr.Image(type="pil"), gr.Textbox()],
+    title="AdVision AI",
+    description="Upload a webpage screenshot. Detects ad placements and gives visibility suggestions."
+)
+
+interface.launch()