Update app.py

app.py

@@ -1,16 +1,38 @@
 import os
 import gradio as gr
 from dotenv import load_dotenv
-from utils import EXAMPLE_ITEMS, analyze
+from pathlib import Path
+from utils import EXAMPLE_ITEMS, analyze, download_image
 
 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, #667eea 0%, #764ba2 100%);
+    background: linear-gradient(135deg, #14b8a6 0%, #0d9488 100%);
     color: white;
     border-radius: 8px;
     margin-bottom: 2rem;
@@ -18,7 +40,14 @@ CSS = """
 .header h1 { margin: 0; font-size: 2.5em; }
 .header p { margin: 0.5rem 0; opacity: 0.9; }
 .header a { color: white; text-decoration: underline; }
-.footer {
+
+/* Hide built-in Gradio footer */
+footer {
+    display: none !important;
+}
+
+/* Custom footer styling */
+.custom-footer {
     text-align: center;
     padding: 1rem;
     color: #888;
@@ -82,10 +111,10 @@ with gr.Blocks(
     
     gr.Markdown("### 🎯 Try an Example")
     gr.Examples(
-        examples=[[url, model] for url, model in EXAMPLE_ITEMS],
-        inputs=[url_input, model_choice],
+        examples=EXAMPLES,
+        inputs=[image_input, model_choice],
         outputs=output,
-        fn=lambda url, model: analyze(None, url, model, TOKEN),
+        fn=lambda img, model: analyze(img, None, model, TOKEN),
         cache_examples=False,
         run_on_click=True,
     )
@@ -96,6 +125,6 @@ with gr.Blocks(
         outputs=output,
     )
     
-    gr.HTML('<div class="footer">Developed by Viddexa</div>')
+    gr.HTML('<div class="custom-footer">Developed by Viddexa</div>')
 
 demo.launch()