chore: Update UI

app.py

@@ -3,7 +3,10 @@ import PIL.Image
 import torch
 from transformers import AutoModelForZeroShotObjectDetection, AutoProcessor
 
+from kofi import SCRIPT
+
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+HEADLINE = "# [LLMDet](https://github.com/iSEE-Laboratory/LLMDet) Arena ✨"
 
 
 class Detector:
@@ -64,7 +67,7 @@ def _postprocess(detections):
     return annotations
 
 
-def detect_objects(image, labels, confidence_threshold):
+def run(image, labels, confidence_threshold):
     labels = [label.strip() for label in labels.split(",")]
 
     detections = []
@@ -79,10 +82,8 @@ def detect_objects(image, labels, confidence_threshold):
     return tuple((image, det) for det in detections)
 
 
-with gr.Blocks(delete_cache=(5, 10)) as demo:
-    gr.Markdown(
-        "# LLMDet Arena ✨\n ### [Paper](https://arxiv.org/abs/2501.18954) - [Repository](https://github.com/iSEE-Laboratory/LLMDet)"
-    )
+with gr.Blocks(delete_cache=(5, 10), js=SCRIPT) as demo:
+    gr.Markdown(HEADLINE)
 
     with gr.Row():
         with gr.Column():
@@ -111,7 +112,7 @@ with gr.Blocks(delete_cache=(5, 10)) as demo:
             )
 
     with gr.Row():
-        detect_button = gr.Button("Detect Objects")
+        detect_button = gr.Button("Run")
 
     with gr.Row():
         gr.Markdown("## Output Annotated Images")
@@ -123,7 +124,7 @@ with gr.Blocks(delete_cache=(5, 10)) as demo:
 
     # Connect the button to the detection function
     detect_button.click(
-        fn=detect_objects,
+        fn=run,
         inputs=[image_input, text_input, confidence_slider],
         outputs=[
             output_annotated_image_tiny,
@@ -160,10 +161,14 @@ with gr.Blocks(delete_cache=(5, 10)) as demo:
                 output_annotated_image_base,
                 output_annotated_image_large,
             ],
-            fn=detect_objects,
+            fn=run,
             cache_examples=True,
         )
 
+    with gr.Row():
+        with gr.Column():
+            gr.HTML('<div id="kofi" style="text-align: center;"></div>')
+
 
 if __name__ == "__main__":
     demo.launch()

kofi.py

@@ -0,0 +1,42 @@
+SCRIPT = """
+async function loadKofi() {
+    var kofiwidget2 = kofiwidget2 ||
+    (function () {
+        var style = "";
+        var html = "";
+        var color = "";
+        var text = "";
+        var id = "";
+        return {
+            init: function (pText, pColor, pId) {
+                color = pColor;
+                text = pText;
+                id = pId;
+                style =
+                    "img.kofiimg{display: initial!important;vertical-align:middle;height:13px!important;width:20px!important;padding-top:0!important;padding-bottom:0!important;border:none;margin-top:0;margin-right:7px!important;margin-left:0!important;margin-bottom:3px!important;content:url('https://storage.ko-fi.com/cdn/cup-border.png')}.kofiimg:after{vertical-align:middle;height:25px;padding-top:0;padding-bottom:0;border:none;margin-top:0;margin-right:6px;margin-left:0;margin-bottom:4px!important;content:url('https://storage.ko-fi.com/cdn/whitelogo.svg')}.btn-container{display:inline-block!important;white-space:nowrap;min-width:160px}span.kofitext{color:#fff !important;letter-spacing: -0.15px!important;text-wrap:none;vertical-align:middle;line-height:30px !important;padding:0;text-align:center;text-decoration:none!important; text-shadow: 0 1px 1px rgba(34, 34, 34, 0.05);}.kofitext a{color:#fff !important;text-decoration:none:important;}.kofitext a:hover{color:#fff !important;text-decoration:none}a.kofi-button{box-shadow: 1px 1px 0px rgba(0, 0, 0, 0.2);line-height:27px!important;min-width:150px;display:inline-block!important;background-color:#29abe0;padding:2px 12px !important;text-align:center !important;border-radius:7px;color:#fff;cursor:pointer;overflow-wrap:break-word;vertical-align:middle;border:0 none #fff !important;font-family:'Quicksand',Helvetica,Century Gothic,sans-serif !important;text-decoration:none;text-shadow:none;font-weight:700!important;font-size:14px !important}a.kofi-button:visited{color:#fff !important;text-decoration:none !important}a.kofi-button:hover{opacity:.85;color:#f5f5f5 !important;text-decoration:none !important}a.kofi-button:active{color:#f5f5f5 !important;text-decoration:none !important}.kofitext img.kofiimg {height:15px!important;width:22px!important;display: initial;animation: kofi-wiggle 3s infinite;}";
+                style = style + "@keyframes kofi-wiggle{0%{transform:rotate(0) scale(1)}60%{transform:rotate(0) scale(1)}75%{transform:rotate(0) scale(1.12)}80%{transform:rotate(0) scale(1.1)}84%{transform:rotate(-10deg) scale(1.1)}88%{transform:rotate(10deg) scale(1.1)}92%{transform:rotate(-10deg) scale(1.1)}96%{transform:rotate(10deg) scale(1.1)}100%{transform:rotate(0) scale(1)}}";
+                style = "<style>" + style + "</style>";
+                html =
+                    "<link href='https://fonts.googleapis.com/css?family=Quicksand:400,700' rel='stylesheet' type='text/css'>";
+                html +=
+                    '<div class=btn-container><a title="Support me on ko-fi.com" class="kofi-button" style="background-color:[color]; text-decoration: none;" href="https://ko-fi.com/[id]" target="_blank"> <span class="kofitext"><img src="https://storage.ko-fi.com/cdn/cup-border.png" alt="Ko-fi donations" class="kofiimg"/>[text]</span></a></div>';
+            },
+            getHTML: function () {
+                var rtn = style + html.replace("[color]", color).replace("[text]", text).replace("[id]", id);
+                return rtn;
+            },
+            draw: function () {
+                document.writeln(style + html.replace("[color]", color).replace("[text]", text).replace("[id]", id));
+            }
+        };
+    }());
+    const checkElement = setInterval(() => {
+        const kofiDiv = document.getElementById('kofi');
+        if (kofiDiv) {
+            clearInterval(checkElement);
+            kofiwidget2.init('Buy me a coffee', '#f87316', 'Y8Y0STZND');
+            kofiDiv.innerHTML = kofiwidget2.getHTML();
+        }
+    }, 100);
+}
+"""