refactor: simplify app.py

app.py

@@ -1,139 +1,74 @@
 import os
 import gradio as gr
-from urllib.request import urlretrieve
-from utils import (
-    EXAMPLE_ITEMS,
-    analyze_image_with_status,
-    run_example_by_index,
-)
+from dotenv import load_dotenv
+from utils import EXAMPLE_ITEMS, analyze, run_example
 
-VIDDEXA_TOKEN = os.getenv("HF_TOKEN")
+load_dotenv()
+TOKEN = os.getenv("HF_TOKEN")
 
-CUSTOM_CSS = """
-.verdict-safe { background-color: #D5F5E3; border: 2px solid #2ECC71; color: #1D8348; }
-.verdict-sensitive { background-color: #FCF3CF; border: 2px solid #F1C40F; color: #B7950B; }
-.verdict-nsfw { background-color: #FADBD8; border: 2px solid #E74C3C; color: #B03A2E; }
-.verdict-card { padding: 20px; border-radius: 10px; text-align: center; font-size: 24px; font-weight: bold; }
-.viddexa-logo { background-color: #FFFFFF; padding: 8px; border-radius: 8px; }
-footer {visibility: hidden}
-"""
-
-with gr.Blocks(
-    theme=gr.themes.Default(primary_hue=gr.themes.colors.teal),
-    title="Visual Content Moderation",
-    css=CUSTOM_CSS,
-    analytics_enabled=False,
-) as demo:
-    gr.HTML("""
-    <div class="viddexa-header" style="text-align: center; max-width: 800px; margin: 0 auto;">
-        <img class="viddexa-logo" src="https://aky-tech.com/images/viddexa-logo.svg" alt="Viddexa logo" style="max-width: 320px; width: 80%; height: auto; display: block; margin: 10px auto 6px;" />
-        <p style="font-size: 1.2em; color: #888;">
-            Official demo for <a href="https://github.com/viddexa/moderators" target="_blank"><code>moderators</code></a> package.
-            <br>
-            Upload an image or provide a URL to get an instant content analysis.
-        </p>
-        <p>
-            🔗 <b>Project Links:</b>
-            <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>
-        <p>
-        <details style="margin-top: 10px;"> 
-        <summary style="cursor: pointer; font-weight: bold;">📄 BibTeX entry for citation</summary>
-        <pre style="background-color: rgba(128, 128, 128, 0.1); border: 1px solid rgba(128, 128, 128, 0.3); padding: 15px; border-radius: 5px; text-align: left; overflow-x: auto; margin: 10px 0;"><code style="font-family: monospace;">@article{akyon2023nudity,
+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/)"
+    )
+    
+    with gr.Accordion("📄 BibTeX entry for citation", open=False):
+        gr.Code(
+            value="""@article{akyon2023nudity,
   title={State-of-the-art in nudity classification: A comparative analysis},
   author={Akyon, Fatih Cagatay and Temizel, Alptekin},
   booktitle={2023 IEEE International Conference on Acoustics, Speech, and Signal Processing Workshops (ICASSPW)},
   pages={1--5},
   year={2023},
   organization={IEEE}
-}</code></pre>
-    </details>
-        </p>
-    </div>
-    """)
-
-    with gr.Row(variant="panel"):
-        with gr.Column(scale=1, min_width=350):
-            gr.Markdown("## ⚙️ Step 1: Configure Settings")
+}""",
+            language="bibtex",
+            show_label=False,
+        )
+    
+    with gr.Row():
+        with gr.Column(scale=1):
             model_choice = gr.Dropdown(
                 choices=["viddexa/nsfw-detection-mini", "viddexa/nsfw-detection-nano"],
                 value="viddexa/nsfw-detection-mini",
-                label="Moderation Model",
-                info="Choose the faster 'nano' or the better 'mini' model.",
+                label="Model",
             )
-
-            gr.Markdown("## 🖼️ Step 2: Provide an Image")
+            
             with gr.Tabs():
-                with gr.TabItem("Upload Image"):
-                    image_input = gr.Image(type="filepath", label="Drag & drop a file or click to upload")
-                with gr.TabItem("From URL"):
-                    image_url_input = gr.Textbox(
-                        label="Image URL",
-                        placeholder="https://example.com/image.jpg",
-                    )
-
-            run_btn = gr.Button("Start Analysis", variant="primary", scale=2)
-
-        with gr.Column(scale=2, min_width=500):
-            gr.Markdown("## 📊 Step 3: Review Results")
-            verdict_output = gr.HTML(label="Final Verdict")
-            label_output = gr.Label(label="Classification Scores", num_top_classes=4, show_label=True)
-            markdown_output = gr.Markdown(label="All Scores")
-
-            with gr.Accordion("Show Raw JSON Output", open=False):
-                json_output = gr.JSON(label="Model Output (JSON)")
-
-    gr.Markdown("## 🎯 Try an Example (click an image)")
-    gallery_items = [[url, caption] for (url, _model, caption) in EXAMPLE_ITEMS]
-    examples_gallery = gr.Gallery(
-        label="Try an Example",
-        value=gallery_items,
+                with gr.TabItem("Upload"):
+                    image_input = gr.Image(type="filepath", label="Image")
+                with gr.TabItem("URL"):
+                    url_input = gr.Textbox(label="Image URL", placeholder="https://...")
+            
+            analyze_btn = gr.Button("Analyze", variant="primary")
+        
+        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,
     )
-
-    status_md = gr.Markdown("Last analysed: —", elem_id="last-example-status")
-
-    examples_gallery.select(
-        fn=lambda evt: run_example_by_index(evt, VIDDEXA_TOKEN),
-        outputs=[
-            verdict_output,
-            label_output,
-            markdown_output,
-            json_output,
-            model_choice,
-            image_url_input,
-            status_md,
-        ],
+    
+    analyze_btn.click(
+        fn=lambda img, url, model: analyze(img, url, model, TOKEN),
+        inputs=[image_input, url_input, model_choice],
+        outputs=output,
     )
-
-    run_btn.click(
-        fn=lambda img, url, model, progress: analyze_image_with_status(img, url, model, VIDDEXA_TOKEN, progress),
-        inputs=[image_input, image_url_input, model_choice],
-        outputs=[verdict_output, label_output, markdown_output, json_output, status_md],
+    
+    gallery.select(
+        fn=lambda evt: run_example(evt, TOKEN),
+        outputs=[output, model_choice, url_input],
     )
 
-    gr.HTML("""
-    <div style="text-align: center; margin-top: 20px; color: #888;">
-        <p>Developed by Viddexa.</p>
-    </div>
-    """)
-
-if not os.path.exists("examples"):
-    os.makedirs("examples")
-    print("Created 'examples' directory.")
-    try:
-        urlretrieve(
-            "https://images.pexels.com/photos/36717/amazing-animal-beautiful-beautifull.jpg",
-            "examples/safe_nature.jpg"
-        )
-        print("Downloaded an example image to 'examples/safe_nature.jpg'")
-    except Exception as e:
-        print(f"Could not download example image: {e}")
-
 demo.launch()