Update utils.py

utils.py

@@ -2,6 +2,7 @@ import gradio as gr
 from typing import Any, Dict
 from urllib.request import urlopen, Request
 from io import BytesIO
+from pathlib import Path
 from PIL import Image
 from functools import lru_cache
 
@@ -14,6 +15,36 @@ EXAMPLE_ITEMS = [
     ("https://img.freepik.com/premium-photo/portrait-beautiful-young-woman_1048944-5548042.jpg", "viddexa/nsfw-detection-nano"),
 ]
 
+CUSTOM_CSS = """
+.header {
+    text-align: center;
+    padding: 2rem 1rem;
+    background: linear-gradient(135deg, #14b8a6 0%, #0d9488 100%);
+    color: white;
+    border-radius: 8px;
+    margin-bottom: 2rem;
+}
+.header h1 { margin: 0; font-size: 2.5em; }
+.header p { margin: 0.5rem 0; opacity: 0.9; }
+.header a { color: white; text-decoration: underline; }
+
+/* Hide built-in Gradio footer */
+footer {
+    display: none !important;
+}
+
+/* Custom footer styling */
+.custom-footer {
+    text-align: center;
+    padding: 1rem;
+    color: #888;
+    margin-top: 2rem;
+}
+"""
+
+EXAMPLES_DIR = Path("examples")
+EXAMPLES_DIR.mkdir(exist_ok=True)
+
 
 @lru_cache(maxsize=32)
 def download_image(url: str) -> Image.Image:
@@ -23,6 +54,22 @@ def download_image(url: str) -> Image.Image:
         return Image.open(BytesIO(resp.read())).convert("RGB")
 
 
+def prepare_examples():
+    """Download and prepare example images for display."""
+    examples = []
+    for i, (url, model) in enumerate(EXAMPLE_ITEMS):
+        img_path = EXAMPLES_DIR / f"example_{i}.jpg"
+        if not img_path.exists():
+            try:
+                img = download_image(url)
+                img.save(img_path, "JPEG", quality=95)
+            except Exception as e:
+                print(f"Warning: Could not download example {i}: {e}")
+                continue
+        examples.append([str(img_path), model])
+    return examples
+
+
 def load_model(model_id: str, token: str | None = None) -> Any:
     """Load model with caching."""
     if model_id not in _MODEL_CACHE: