Update app.py

app.py

@@ -436,9 +436,22 @@ def calculate_dot_product_and_representations_independent(query_model_choice, do
     return full_output
 
 
-# Function to generate share link (simulated)
-def generate_share_link():
-    return "🔗 Share link functionality would be available when running with share=True"
+# Function to get the share URL (will be populated when demo launches)
+share_url = None
+
+def get_share_url():
+    global share_url
+    if share_url:
+        return f"🔗 Share this app: {share_url}"
+    else:
+        return "🔗 Share URL will be available after launching with share=True"
+
+def copy_share_url():
+    global share_url
+    if share_url:
+        return share_url
+    else:
+        return "Share URL not available yet"
 
 # --- Gradio Interface Setup with Tabs ---
 with gr.Blocks(title="SPLADE Demos", css=css) as demo:
@@ -468,15 +481,16 @@ with gr.Blocks(title="SPLADE Demos", css=css) as demo:
                     # Add share button underneath the input field
                     with gr.Row(elem_classes="share-button-container"):
                         share_button = gr.Button(
-                            "📤 Generate Share Link", 
+                            "🔗 Get Share Link", 
                             elem_classes="custom-share-button",
                             size="sm"
                         )
                     
-                    share_output = gr.Textbox(
+                    share_display = gr.Textbox(
                         label="Share Link",
-                        interactive=False,
-                        visible=False
+                        interactive=True,
+                        visible=False,
+                        placeholder="Share URL will appear here..."
                     )
                     
                     info_output_display = gr.Markdown(
@@ -487,6 +501,16 @@ with gr.Blocks(title="SPLADE Demos", css=css) as demo:
                 with gr.Column(scale=2): # Right column for the main representation output
                     main_representation_output = gr.Markdown()
 
+                        
+            # Connect share button
+            share_button.click(
+                fn=copy_share_url,
+                outputs=share_display
+            ).then(
+                fn=lambda: gr.update(visible=True),
+                outputs=share_display
+            )
+
             # Connect the interface elements
             model_radio.change(
                 fn=predict_representation_explorer,
@@ -552,5 +576,12 @@ with gr.Blocks(title="SPLADE Demos", css=css) as demo:
                 allow_flagging="never" # Keep this to keep the share button at the bottom of THIS interface
             )
 
-# Crucial step: Launch the demo with share=True
-demo.launch(share=True)
+# Crucial step: Launch the demo with share=True and capture the share URL
+if __name__ == "__main__":
+    # Launch and capture the share URL
+    demo_info = demo.launch(share=True, return_url=True)
+    if hasattr(demo_info, 'share_url') and demo_info.share_url:
+        share_url = demo_info.share_url
+        print(f"Share URL: {share_url}")
+    else:
+        print("Share URL not available")