Update app.py

app.py

@@ -1,63 +1,16 @@
 import os
 import gradio as gr
 from dotenv import load_dotenv
-from pathlib import Path
-from utils import EXAMPLE_ITEMS, analyze, download_image
+from utils import analyze, prepare_examples, CUSTOM_CSS
 
 load_dotenv()
 TOKEN = os.getenv("HF_TOKEN")
 
-# Prepare example images
-EXAMPLES_DIR = Path("examples")
-EXAMPLES_DIR.mkdir(exist_ok=True)
-
-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
-
 EXAMPLES = prepare_examples()
 
-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;
-}
-"""
-
 with gr.Blocks(
     title="Visual Content Moderation",
-    css=CSS,
+    css=CUSTOM_CSS,
     theme=gr.themes.Default(primary_hue=gr.themes.colors.teal),
     analytics_enabled=False,
 ) as demo:
@@ -115,7 +68,7 @@ with gr.Blocks(
         inputs=[image_input, model_choice],
         outputs=output,
         fn=lambda img, model: analyze(img, None, model, TOKEN),
-        cache_examples=False,
+        cache_examples=True,
         run_on_click=True,
     )