Add specimen type selector with URINE default

app.py

@@ -25,6 +25,8 @@ SPACE_ROOT = Path(__file__).resolve().parent
 WATERMARK_TEXT = "go to carb-connect.com fully use"
 UPGRADE_MESSAGE = "Go to carb-connect.com to keep using this app"
 UPGRADE_URL = "https://carb-connect.com"
+SPECIMEN_TYPE_OPTIONS = ["URINE", "BLOOD (AEROBIC)", "BLOOD (ANAEROBIC)"]
+DEFAULT_SPECIMEN_TYPE = "URINE"
 
 CUSTOM_CSS = """
 :root {
@@ -253,14 +255,19 @@ def _load_example_choices() -> list[tuple[str, str]]:
     return choices
 
 
-def _post_image_echo(image_path: str, hf_token: str) -> httpx.Response:
+def _normalize_specimen_type(specimen_type: str | None) -> str:
+    return specimen_type if specimen_type in SPECIMEN_TYPE_OPTIONS else DEFAULT_SPECIMEN_TYPE
+
+
+def _post_image_echo(image_path: str, hf_token: str, specimen_type: str | None) -> httpx.Response:
     infer_url = f"{TRIAL_GATEWAY_URL}/v1/trial/infer"
     headers = {"Authorization": f"Bearer {hf_token}"}
     mime_type = mimetypes.guess_type(image_path)[0] or "application/octet-stream"
+    form_data = {"specimen_type": _normalize_specimen_type(specimen_type)}
 
     with open(image_path, "rb") as img_file:
         files = {"image": (Path(image_path).name, img_file, mime_type)}
-        return httpx.post(infer_url, headers=headers, files=files, timeout=45.0)
+        return httpx.post(infer_url, headers=headers, files=files, data=form_data, timeout=45.0)
 
 
 def _content_type_to_format(content_type: str) -> str:
@@ -319,6 +326,7 @@ def _apply_watermark(image_bytes: bytes, content_type: str) -> bytes:
 
 def _run_image_trial(
     input_image_path: str | None,
+    specimen_type: str | None,
     upgrade_cta_visible: bool,
     oauth_token: gr.OAuthToken | None = None,
 ) -> tuple[str | None, str, str, bool]:
@@ -341,7 +349,7 @@ def _run_image_trial(
         )
 
     try:
-        response = _post_image_echo(input_image_path, oauth_token.token)
+        response = _post_image_echo(input_image_path, oauth_token.token, specimen_type)
     except httpx.HTTPError as error:
         return (
             None,
@@ -464,6 +472,11 @@ with gr.Blocks(
             if not EXAMPLE_CHOICES:
                 gr.Markdown(NO_EXAMPLES_MESSAGE)
             input_image = gr.Image(type="filepath", label="Input image")
+            specimen_type_input = gr.Dropdown(
+                label="Type of specimen",
+                choices=SPECIMEN_TYPE_OPTIONS,
+                value=DEFAULT_SPECIMEN_TYPE,
+            )
             run_button = gr.Button("Send image", elem_id="run-btn")
         with gr.Column(scale=1, elem_classes=["panel"]):
             output_image = gr.Image(type="filepath", label="Output image")
@@ -478,7 +491,7 @@ with gr.Blocks(
 
     run_button.click(
         _run_image_trial,
-        inputs=[input_image, upgrade_cta_state],
+        inputs=[input_image, specimen_type_input, upgrade_cta_state],
         outputs=[output_image, trial_status, upgrade_cta_html, upgrade_cta_state],
         show_progress="full",
     )