Add share link + result card to text tab, download image button for mobile, hide Gradio toolbar via CSS

app.py

@@ -284,6 +284,16 @@ label span {
     text-decoration: none !important;
     border-bottom: 1px dotted var(--text-muted);
 }
+
+/* Hide Gradio image toolbar (download/share/fullscreen) on result cards */
+#share-card .image-toolbar,
+#text-share-card .image-toolbar,
+#share-card .icon-buttons,
+#text-share-card .icon-buttons,
+#share-card button[aria-label],
+#text-share-card button[aria-label] {
+    display: none !important;
+}
 """
 
 HUMAN_EXAMPLE = (
@@ -361,13 +371,28 @@ def _error_label(msg: str) -> dict:
 
 
 def detect_text(text: str) -> dict:
-    """Classify text as human-written or AI-generated."""
+    """Classify text as human-written or AI-generated (label only)."""
     if not text or len(text.strip().split()) < 10:
         return _error_label("Please enter at least a few sentences (~50 words)")
     predictions = classifier.predict(text, k=2)
     return {label: round(score, 4) for label, score in predictions}
 
 
+def detect_text_full(text: str) -> tuple[dict, str, Image.Image | None]:
+    """Classify text and generate share link + card."""
+    result = detect_text(text)
+    if len(result) != 2 or any(k.startswith("Please") for k in result):
+        return result, "", None
+    # Share link with text snippet as param
+    snippet = text.strip()[:200]
+    share_link = f"{SPACE_URL}/?text={urllib.parse.quote(snippet, safe='')}"
+    # Card image
+    top_label = max(result, key=result.get)
+    preview = text.strip()[:80] + ("..." if len(text.strip()) > 80 else "")
+    card = make_result_card(preview, top_label, result[top_label])
+    return result, share_link, card
+
+
 SPACE_URL = "https://adaptive-classifier-ai-detector.hf.space"
 
 
@@ -419,10 +444,12 @@ def make_result_card(url: str, label: str, confidence: float) -> Image.Image:
     if fill_w > 0:
         draw.rounded_rectangle([bar_x, bar_y, bar_x + fill_w, bar_y + bar_h], radius=12, fill=result_color)
 
-    # URL
-    display_url = url[:80] + "..." if len(url) > 80 else url
-    draw.text((60, 285), "URL analyzed:", fill=muted_color, font=font_small)
-    draw.text((60, 308), display_url, fill=text_color, font=font_mono)
+    # Source text/URL
+    display_src = url[:80] + "..." if len(url) > 80 else url
+    is_url = url.startswith("http") or url.startswith("www")
+    src_label = "URL analyzed:" if is_url else "Text analyzed:"
+    draw.text((60, 285), src_label, fill=muted_color, font=font_small)
+    draw.text((60, 308), display_src, fill=text_color, font=font_mono)
 
     # Footer
     draw.text((60, H - 60), "adaptive-classifier-ai-detector.hf.space", fill=muted_color, font=font_small)
@@ -487,8 +514,62 @@ with gr.Blocks(css=CSS, title="AI Text Detector", theme=gr.themes.Base()) as dem
 
             with gr.Group(elem_classes="result-card"):
                 text_output = gr.Label(num_top_classes=2, label="Result")
+                text_share_link = gr.Textbox(visible=False, elem_id="text-share-url")
+                with gr.Row(visible=False) as text_share_row:
+                    text_copy_link_btn = gr.Button("Copy Link", size="sm", elem_classes="detect-btn")
+                    text_copy_img_btn = gr.Button("Copy Image", size="sm", elem_classes="detect-btn")
+                    text_dl_img_btn = gr.Button("Download Image", size="sm", elem_classes="detect-btn")
+                text_share_card = gr.Image(
+                    label="Result card",
+                    visible=False,
+                    type="pil",
+                    elem_id="text-share-card",
+                )
+
+            def run_text_detection(text):
+                result, link, card = detect_text_full(text)
+                has_result = bool(link)
+                return (
+                    result,
+                    gr.update(value=link, visible=has_result),
+                    gr.update(visible=has_result),
+                    gr.update(value=card, visible=has_result),
+                )
+
+            text_btn.click(
+                fn=run_text_detection,
+                inputs=text_input,
+                outputs=[text_output, text_share_link, text_share_row, text_share_card],
+                api_name="detect",
+            )
 
-            text_btn.click(fn=detect_text, inputs=text_input, outputs=text_output, api_name="detect")
+            text_copy_link_btn.click(
+                fn=None, inputs=text_share_link, outputs=None,
+                js="(url) => { navigator.clipboard.writeText(url); }",
+            )
+            text_copy_img_btn.click(
+                fn=None, inputs=None, outputs=None,
+                js="""() => {
+                    const img = document.querySelector('#text-share-card img');
+                    if (img) {
+                        const c = document.createElement('canvas');
+                        c.width = img.naturalWidth; c.height = img.naturalHeight;
+                        c.getContext('2d').drawImage(img, 0, 0);
+                        c.toBlob(b => navigator.clipboard.write([new ClipboardItem({'image/png': b})]), 'image/png');
+                    }
+                }""",
+            )
+            text_dl_img_btn.click(
+                fn=None, inputs=None, outputs=None,
+                js="""() => {
+                    const img = document.querySelector('#text-share-card img');
+                    if (img) {
+                        const a = document.createElement('a');
+                        a.href = img.src; a.download = 'ai-detector-result.png';
+                        a.click();
+                    }
+                }""",
+            )
 
             gr.HTML('<div class="examples-heading">Try an example</div>')
             gr.Examples(
@@ -513,14 +594,12 @@ with gr.Blocks(css=CSS, title="AI Text Detector", theme=gr.themes.Base()) as dem
                 with gr.Row(visible=False) as share_row:
                     copy_link_btn = gr.Button("Copy Link", size="sm", elem_classes="detect-btn")
                     copy_img_btn = gr.Button("Copy Image", size="sm", elem_classes="detect-btn")
+                    dl_img_btn = gr.Button("Download Image", size="sm", elem_classes="detect-btn")
                 share_card = gr.Image(
                     label="Result card",
                     visible=False,
                     type="pil",
                     elem_id="share-card",
-                    show_download_button=False,
-                    show_share_button=False,
-                    show_fullscreen_button=False,
                 )
 
             with gr.Group(elem_classes="input-card"):
@@ -576,6 +655,18 @@ with gr.Blocks(css=CSS, title="AI Text Detector", theme=gr.themes.Base()) as dem
                 }""",
             )
 
+            dl_img_btn.click(
+                fn=None, inputs=None, outputs=None,
+                js="""() => {
+                    const img = document.querySelector('#share-card img');
+                    if (img) {
+                        const a = document.createElement('a');
+                        a.href = img.src; a.download = 'ai-detector-result.png';
+                        a.click();
+                    }
+                }""",
+            )
+
             gr.HTML('<div class="examples-heading">Try an example</div>')
             gr.Examples(
                 examples=[
@@ -600,25 +691,44 @@ with gr.Blocks(css=CSS, title="AI Text Detector", theme=gr.themes.Base()) as dem
     """)
 
     def _on_load(request: gr.Request):
-        url = request.query_params.get("url", "")
-        if url:
-            result, preview, link, card = detect_url(url)
-            has_result = bool(link)
+        text_param = request.query_params.get("text", "")
+        url_param = request.query_params.get("url", "")
+        # Default: no changes
+        no_change = (
+            gr.update(), "", {}, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False),
+            "", {}, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), "",
+        )
+        if text_param:
+            result, link, card = detect_text_full(text_param)
+            has = bool(link)
+            return (
+                gr.update(selected="text-tab"),
+                text_param, result,
+                gr.update(value=link, visible=has), gr.update(visible=has), gr.update(value=card, visible=has),
+                "", {}, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), "",
+            )
+        if url_param:
+            result, preview, link, card = detect_url(url_param)
+            has = bool(link)
             return (
                 gr.update(selected="url-tab"),
-                url,
-                result,
-                gr.update(value=link, visible=has_result),
-                gr.update(visible=has_result),
-                gr.update(value=card, visible=has_result),
+                "", {}, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False),
+                url_param, result,
+                gr.update(value=link, visible=has), gr.update(visible=has), gr.update(value=card, visible=has),
                 preview,
             )
-        return gr.update(), "", {}, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), ""
+        return no_change
 
     demo.load(
         fn=_on_load,
         inputs=None,
-        outputs=[tabs, url_input, url_output, share_link, share_row, share_card, url_preview],
+        outputs=[
+            tabs,
+            # text tab outputs
+            text_input, text_output, text_share_link, text_share_row, text_share_card,
+            # url tab outputs
+            url_input, url_output, share_link, share_row, share_card, url_preview,
+        ],
     )
 
 if __name__ == "__main__":