Add url params parsing

README.md

@@ -10,3 +10,5 @@ pinned: false
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+ - Added url param parsing, e.g. ?transform=HorizontalFlip will open app with selected HorizontalFlip

app.py

@@ -6,12 +6,12 @@ import gradio as gr
 import inspect
 import io
 import numpy as np
-import os
 
+from dataclasses import dataclass
 from copy import deepcopy
 from functools import wraps
 from PIL import Image, ImageDraw
-from typing import get_type_hints
+from typing import get_type_hints, Optional
 from mixpanel import Mixpanel
 
 from utils import is_not_supported_transform
@@ -90,6 +90,11 @@ for mask in BASE64_DEFAULT_MASKS:
     )
 
 
+@dataclass
+class RequestParams:
+    user_ip: str
+    transform_name: Optional[str]
+
 def track_event(event_name, user_id="unknown", properties=None):
     if properties is None:
         properties = {}
@@ -97,6 +102,15 @@ def track_event(event_name, user_id="unknown", properties=None):
     logger.info(f"Event tracked: {event_name} - {properties}")
 
 
+def get_params(request: gr.Request) -> RequestParams:
+    """Parse input request parameters."""
+    ip = request.client.host
+    transform_name = request.query_params.get("transform", None)
+    params = RequestParams(user_ip=ip, transform_name=transform_name)
+    track_event("app_opened", user_id=params.user_ip, properties={"transform_name": params.transform_name})
+    return params
+
+
 def run_with_retry(compose):
     @wraps(compose)
     def wrapper(*args, **kwargs):
@@ -125,7 +139,7 @@ def run_with_retry(compose):
     return wrapper
 
 
-def draw_boxes(image, boxes, color=(255, 0, 0), thickness=2) -> np.ndarray:
+def draw_boxes(image, boxes, color=(255, 0, 0), thickness=1) -> np.ndarray:
     """Draw boxes with PIL."""
     pil_image = Image.fromarray(image)
     draw = ImageDraw.Draw(pil_image)
@@ -206,15 +220,13 @@ def get_formatted_signature(function_or_class, indentation=4):
     return result
 
 
-def get_formatted_transform(transform_number):
-    transform_name = transforms_keys[transform_number]
+def get_formatted_transform(transform_name):
     track_event("transform_selected", properties={"transform_name": transform_name})
     transform = transforms_map[transform_name]
     return f"A.{transform.__name__}{get_formatted_signature(transform)}"
 
 
-def get_formatted_transform_docs(transform_number):
-    transform_name = transforms_keys[transform_number]
+def get_formatted_transform_docs(transform_name):
     transform = transforms_map[transform_name]
     return transform.__doc__.strip("\n")
 
@@ -282,6 +294,10 @@ def update_code_and_docs(select):
     docs = get_formatted_transform_docs(select)
     return code, docs
 
+def update_code_and_docs_on_start(url_params: gr.Request):
+    params = get_params(url_params)
+    transform_name = params.transform_name if params.transform_name is not None else DEFAULT_TRANSFORM
+    return gr.update(value=transform_name)
 
 with gr.Blocks() as demo:
     gr.Markdown(HEADER)
@@ -292,23 +308,19 @@ with gr.Blocks() as demo:
                     label="Select a transformation",
                     choices=transforms_keys,
                     value=DEFAULT_TRANSFORM,
-                    type="index",
+                    type="value",
                     interactive=True,
                 )
                 with gr.Accordion("Documentation (click to expand)", open=False):
                     docs = gr.TextArea(
-                        get_formatted_transform_docs(
-                            transforms_keys.index(DEFAULT_TRANSFORM)
-                        ),
+                        get_formatted_transform_docs(DEFAULT_TRANSFORM),
                         show_label=False,
                         interactive=False,
                     )
                 code = gr.Code(
                     label="Code",
                     language="python",
-                    value=get_formatted_transform(
-                        transforms_keys.index(DEFAULT_TRANSFORM)
-                    ),
+                    value=get_formatted_transform(DEFAULT_TRANSFORM),
                     interactive=True,
                     lines=5,
                 )
@@ -333,12 +345,13 @@ with gr.Blocks() as demo:
             columns=3,
             show_label=False,
         )
-
     select.change(fn=update_code_and_docs, inputs=[select], outputs=[code, docs])
     button.click(
         fn=update_augmented_images, inputs=[image, code], outputs=[augmented_image]
     )
-
+    demo.load(
+        update_code_and_docs_on_start, inputs=None, outputs=[select]
+    )
 
 if __name__ == "__main__":
     demo.launch()