Updated the API client to support RapidAPI, small tweaks and fixes

app.py

@@ -16,6 +16,8 @@ EXAMPLE_PATH = os.path.join(os.path.dirname(__file__), "examples")
 API_URL = os.getenv("TRY_ON_DIFFUSION_DEMO_API_URL", "http://localhost:8000")
 API_KEY = os.getenv("TRY_ON_DIFFUSION_DEMO_API_KEY", "")
 
+SHOW_RAPIDAPI_LINK = os.getenv("TRY_ON_DIFFUSION_DEMO_SHOW_RAPIDAPI_LINK", "1") == "1"
+
 CONCURRENCY_LIMIT = int(os.getenv("TRY_ON_DIFFUSION_DEMO_CONCURRENCY_LIMIT", "2"))
 
 logging.basicConfig(level=LOG_LEVEL, format=LOG_FORMAT, datefmt=LOG_DATE_FORMAT)
@@ -68,7 +70,7 @@ def try_on(
     )
 
     if result.status_code == 200:
-        return cv2.cvtColor(result.image, cv2.COLOR_RGB2BGR), f"<h3>Success</h3><p>Seed: {result.seed}</p>"
+        return cv2.cvtColor(result.image, cv2.COLOR_BGR2RGB), f"<h3>Success</h3><p>Seed: {result.seed}</p>"
     else:
         error_message = f"<h3>Error {result.status_code}</h3>"
 
@@ -78,7 +80,7 @@ def try_on(
         return None, error_message
 
 
-with gr.Blocks(theme=gr.themes.Soft()) as app:
+with gr.Blocks(theme=gr.themes.Soft(), delete_cache=(3600, 3600)) as app:
     gr.HTML(
         f"""
         <div style="width: 100%; background-color: #001537; border-radius: 10px; padding-left: 10px">
@@ -93,13 +95,26 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             custom diffusion-based pipeline for fast and flexible multi-modal virtual try-on.
             Clothing, avatar and background can be specified by reference images or text prompts allowing for clothing 
             transfer, avatar replacement, fashion image generation and other virtual try-on related tasks. 
-            Feel free to <a href="https://texelmoda.com/contact/" target="_blank">contact us</a> if you are interested 
-            in commercial API usage or partnership.
-        </p>
-        <br/>
     """
     )
 
+    if SHOW_RAPIDAPI_LINK:
+        gr.Button(
+            value="Check out the API @ RapidAPI.com",
+            link="https://rapidapi.com/texelmoda-texelmoda-apis/api/try-on-diffusion",
+            icon="https://files.readme.io/9336831-small-rapid-logo-favicon.png",
+        )
+
+        gr.HTML(
+            """
+            <center>
+                <a href="https://huggingface.co/spaces/texelmoda/try-on-diffusion/blob/main/docs/API.md">API Documentation</a>
+            </center>
+            """
+        )
+
+    gr.HTML("</p>")
+
     with gr.Row():
         with gr.Column():
             gr.HTML(
@@ -247,9 +262,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
                 info="Seed used for generation, specify -1 for random seed for each generation.",
             )
 
-            generate_button = gr.Button(value="Generate", elem_id="button")
+            generate_button = gr.Button(value="Generate", variant="primary")
 
-            result_image = gr.Image(label="Result", show_share_button=False)
+            result_image = gr.Image(label="Result", show_share_button=False, format="jpeg")
             result_details = gr.HTML(label="Details")
 
     generate_button.click(

docs/API.md

@@ -33,7 +33,7 @@
 Virtual Try-On Diffusion [VTON-D] by [Texel.Moda](https://texelmoda.com) is a custom diffusion-based pipeline for fast 
 and flexible multi-modal virtual try-on. Clothing, avatar and background can be specified by reference images or text 
 prompts allowing for clothing transfer, avatar replacement, fashion image generation and other virtual try-on related 
-tasks. Check out the [demo on HuggingFace](https://huggingface.co/spaces/texelmoda/try-on-diffusion) to try the API in
+tasks. Check out the [demo on Hugging Face](https://huggingface.co/spaces/texelmoda/try-on-diffusion) to try the API in
 a user-friendly way.
 
 ## Consuming the API
@@ -60,7 +60,7 @@ curl --request POST \
 ```
 
 For a simple Python client implementation please see the 
-[HuggingFace demo application source](https://huggingface.co/spaces/texelmoda/try-on-diffusion/blob/main/try_on_diffusion_client.py).
+[Hugging Face demo application source](https://huggingface.co/spaces/texelmoda/try-on-diffusion/blob/main/try_on_diffusion_client.py).
 
 ## Try-On Endpoints
 

try_on_diffusion_client.py

@@ -2,6 +2,7 @@ import cv2
 import numpy as np
 import requests
 from requests_toolbelt.multipart.encoder import MultipartEncoder
+from urllib.parse import urlparse
 import logging
 import json
 from io import BytesIO
@@ -26,6 +27,13 @@ class TryOnDiffusionClient:
         if self._base_url[-1] == "/":
             self._base_url = self._base_url[:-1]
 
+        parsed_url = urlparse(self._base_url)
+
+        self._rapidapi_host = parsed_url.netloc if parsed_url.netloc.endswith(".rapidapi.com") else None
+
+        if self._rapidapi_host is not None:
+            self._logger.info(f"Using RapidAPI proxy: {self._rapidapi_host}")
+
     @staticmethod
     def _image_to_upload_file(image: np.ndarray) -> tuple:
         _, jpeg_data = cv2.imencode(".jpg", image, [int(cv2.IMWRITE_JPEG_QUALITY), 99])
@@ -44,14 +52,12 @@ class TryOnDiffusionClient:
         avatar_sex: str = None,
         background_image: np.ndarray = None,
         background_prompt: str = None,
-        negative_prompt: str = None,
-        num_images: int = 1,
         seed: int = -1,
         raw_response: bool = False,
     ) -> TryOnDiffusionAPIResponse:
         url = self._base_url + "/try-on-file"
 
-        request_data = {"num_images": str(num_images), "seed": str(seed)}
+        request_data = {"seed": str(seed)}
 
         if clothing_image is not None:
             request_data["clothing_image"] = self._image_to_upload_file(clothing_image)
@@ -74,16 +80,21 @@ class TryOnDiffusionClient:
         if background_prompt is not None:
             request_data["background_prompt"] = background_prompt
 
-        if negative_prompt is not None:
-            request_data["negative_prompt"] = negative_prompt
-
         multipart_data = MultipartEncoder(fields=request_data)
 
+        headers = {"Content-Type": multipart_data.content_type}
+
+        if self._rapidapi_host is not None:
+            headers["X-RapidAPI-Key"] = self._api_key
+            headers["X-RapidAPI-Host"] = self._rapidapi_host
+        else:
+            headers["X-API-Key"] = self._api_key
+
         try:
             response = requests.post(
                 url,
                 data=multipart_data,
-                headers={"Content-Type": multipart_data.content_type, "X-API-Key": self._api_key},
+                headers=headers,
             )
         except Exception as e:
             self._logger.error(e, exc_info=True)