Update app.py

app.py

@@ -1,26 +1,48 @@
 import os
 import gradio as gr
 from dotenv import load_dotenv
-from utils import EXAMPLE_ITEMS, analyze, run_example
+from utils import EXAMPLE_ITEMS, analyze
 
 load_dotenv()
 TOKEN = os.getenv("HF_TOKEN")
 
-with gr.Blocks(title="Visual Content Moderation") as demo:
-    gr.Markdown("# Viddexa - Visual Content Moderation")
-    gr.Markdown("Official demo for [moderators](https://github.com/viddexa/moderators) package.")
-    gr.Markdown("Upload an image or provide a URL to get an instant content analysis.")
-    gr.Markdown(
-        "🔗 **Project Links:** "
-        "[Model: nsfw-mini](https://huggingface.co/viddexa/nsfw-mini) | "
-        "[Model: nsfw-nano](https://huggingface.co/viddexa/nsfw-nano) | "
-        "[Arxiv](https://arxiv.org/abs/2312.16338) | "
-        "[GitHub](https://github.com/viddexa/moderators) | "
-        "[PyPI](https://pypi.org/project/moderators/)"
-    )
+CSS = """
+.header {
+    text-align: center;
+    padding: 2rem 1rem;
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 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; }
+.footer {
+    text-align: center;
+    padding: 1rem;
+    color: #888;
+    margin-top: 2rem;
+}
+"""
+
+with gr.Blocks(title="Visual Content Moderation", css=CSS) as demo:
+    gr.HTML("""
+    <div class="header">
+        <h1>Viddexa - Visual Content Moderation</h1>
+        <p>Official demo for <a href="https://github.com/viddexa/moderators" target="_blank">moderators</a> package</p>
+        <p>
+            <a href="https://huggingface.co/viddexa/nsfw-mini" target="_blank">Model: nsfw-mini</a> | 
+            <a href="https://huggingface.co/viddexa/nsfw-nano" target="_blank">Model: nsfw-nano</a> | 
+            <a href="https://arxiv.org/abs/2312.16338" target="_blank">Arxiv</a> | 
+            <a href="https://github.com/viddexa/moderators" target="_blank">GitHub</a> | 
+            <a href="https://pypi.org/project/moderators/" target="_blank">PyPI</a>
+        </p>
+    </div>
+    """)
     
     with gr.Accordion("📄 BibTeX entry for citation", open=False):
-        gr.Code(
+        gr.Textbox(
             value="""@article{akyon2023nudity,
   title={State-of-the-art in nudity classification: A comparative analysis},
   author={Akyon, Fatih Cagatay and Temizel, Alptekin},
@@ -29,8 +51,9 @@ with gr.Blocks(title="Visual Content Moderation") as demo:
   year={2023},
   organization={IEEE}
 }""",
-            language="bibtex",
             show_label=False,
+            lines=8,
+            max_lines=8,
         )
     
     with gr.Row():
@@ -52,12 +75,12 @@ with gr.Blocks(title="Visual Content Moderation") as demo:
         with gr.Column(scale=1):
             output = gr.Label(label="Classification Scores", num_top_classes=10)
     
-    gr.Markdown("### Examples")
-    gallery = gr.Gallery(
-        value=[[url, f"Example {i+1}"] for i, (url, _) in enumerate(EXAMPLE_ITEMS)],
-        columns=[2, 4],
-        height=200,
-        allow_preview=False,
+    gr.Examples(
+        examples=[[None, url, model] for url, model in EXAMPLE_ITEMS],
+        inputs=[image_input, url_input, model_choice],
+        outputs=output,
+        fn=lambda img, url, model: analyze(img, url, model, TOKEN),
+        cache_examples=False,
     )
     
     analyze_btn.click(
@@ -66,9 +89,6 @@ with gr.Blocks(title="Visual Content Moderation") as demo:
         outputs=output,
     )
     
-    gallery.select(
-        fn=lambda evt: run_example(evt, TOKEN),
-        outputs=[output, model_choice, url_input],
-    )
+    gr.HTML('<div class="footer">Developed by Viddexa</div>')
 
 demo.launch()