Update app.py

app.py

@@ -113,6 +113,34 @@ caption.caption {
     cursor: text;
 }
 """
+
+js = """
+function aff() {
+const links = document.querySelectorAll('a');
+
+const currentUrl = new URL(window.location.href);
+const currentParams = currentUrl.searchParams.toString();
+
+links.forEach(link => {
+    const href = link.getAttribute('href');
+    if (href && (href.startsWith('https://faceonlive.pocketsflow.com') || href.startsWith('https://faceonlive.com'))) {
+        const targetUrl = new URL(href);
+        // Append current page parameters to the link
+        currentParams.split('&').forEach(param => {
+            if (param) {
+                const [key, value] = param.split('=');
+                targetUrl.searchParams.set(key, value);
+            }
+        });
+        link.setAttribute('href', targetUrl.toString());
+        console.log(`Updated Link: ${targetUrl.toString()}`);
+    }
+});
+
+  return ''
+}
+"""
+
 with gr.Blocks(css=custom_css) as demo:
     gr.Markdown(
         """
@@ -135,7 +163,8 @@ with gr.Blocks(css=custom_css) as demo:
 
     search_face_button.click(search_face, inputs=[image, token], outputs=[output], api_name=False)
 
-    gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online"><img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online&labelColor=%23ff8a65&countColor=%2337d67a&style=flat&labelStyle=upper" /></a>')
-
+    gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online&labelColor=%23ff8a65&countColor=%2337d67a&style=flat&labelStyle=upper" /></a>')
+    html = gr.HTML()
+    demo.load(None, inputs=None, outputs=html, js=js)
 
 demo.queue(api_open=False, default_concurrency_limit=4).launch(server_name="0.0.0.0", server_port=7860, show_api=False)