Add permalink_short textbox

app.py

@@ -157,12 +157,11 @@ def load_data(request: gr.Request, model_3d, image_slider):
     query_params = _resolve_short_id_to_query_params(query_params)
 
     # Extract URLs from query parameters (which may have been updated)
+    short_id = query_params.get("sid", None)
     model_url = query_params.get("3d", None)
     hm_url = query_params.get("hm", None)
     img_url = query_params.get("image", None)
 
-    # Check if any relevant parameters are set to decide if we should clear query_params for permalink generation
-    # This logic ensures that if only 'sid' was passed and it resolved, query_params will now contain 3d, hm, image.
     # If 'sid' was passed but didn't resolve, or no params were passed, then model_url, hm_url, img_url will be None.
     has_loadable_params = bool(model_url or hm_url or img_url)
 
@@ -199,17 +198,25 @@ def load_data(request: gr.Request, model_3d, image_slider):
 
     # If any loadable query parameters were effectively present (either directly or via sid), generate a permalink.
     permalink = ""
+    permalink_short = ""
     if has_loadable_params:
+        # Generate permalink using the processed URLs.
         try:
             permalink_model_url = query_params.get("3d", model_url) 
             permalink_hm_url = query_params.get("hm", hm_url if len(slider_images) > 1 and hm_url else (slider_images[1] if len(slider_images) > 1 else None) )
             permalink_img_url = query_params.get("image", img_url if slider_images and img_url else (slider_images[0] if slider_images else None) )
-
             permalink = storage.generate_permalink_from_urls(permalink_model_url, permalink_hm_url, permalink_img_url)
+            if not short_id:
+                # If no short ID was provided, generate a new one.
+                result, short_id = storage.gen_full_url(full_url=permalink, repo_id=constants.HF_REPO_ID, json_file=constants.SHORTENER_JSON_FILE)
+            permalink_short = f"{constants.APP_BASE_URL}/?sid={short_id}"
+            print(f"Generated permalink: {result} (short ID: {short_id})")
         except Exception as e:
             print(f"Error generating permalink: {e}")
+
+
     
-    return model_url, slider_images, permalink
+    return model_url, slider_images, permalink, permalink_short
 
 def process_upload(files, current_model, current_images):
     """
@@ -368,16 +375,25 @@ with gr.Blocks(css_paths="style_20250503.css", title="3D viewer", theme='Surn/be
                     key="permalink",
                     elem_classes="solid centered",
                     max_lines=5,
-                    lines=3
+                    lines=4
                 )
                 gr.Markdown("### Copy the permalink to share your model and images.", elem_classes="solid centered",)
+                permalink_short = gr.Textbox(
+                    show_copy_button=True,
+                    label="Shortened Permalink",
+                    elem_id="short_permalink",
+                    key="permalink",
+                    elem_classes="solid centered",
+                    max_lines=5,
+                    lines=3
+                )
     with gr.Row():
         gr.HTML(value=getVersions(), visible=True, elem_id="versions")
     
     viewer3d.load(
         load_data,
         inputs=[model_3d, image_slider],
-        outputs=[model_3d, image_slider, permalink],
+        outputs=[model_3d, image_slider, permalink, permalink_short],
         scroll_to_output=True
     ).then(
         lambda link: (gr.update(visible=True), gr.update(interactive=False))