Try-on--test

app.py

@@ -190,7 +190,7 @@ def upload_image(image, upload_type, did, ip, user_id, user_name):
         upload_url = upload_image_url
     else:
         upload_url = upload_background_image_url
-    image_format = "PNG"
+    image_format = image.format if image.format else "PNG"
     mime_type = f"image/{image_format.lower()}"
     timestamp = str(int(time.time()))
     signature = generate_signature(
@@ -395,7 +395,6 @@ with gr.Blocks(css=css) as WeShop:
             gr.Markdown("#### Step 1: Upload a garment image")
             main_image_input = gr.Image(
                 type="pil",
-                format='png',
                 label="Main Image",
                 elem_classes=["image-container", "hide-buttons"]
             )
@@ -411,7 +410,6 @@ with gr.Blocks(css=css) as WeShop:
             gr.Markdown("#### Step 2: Upload a person image")
             background_image_input = gr.Image(
                 type="pil",
-                format='png',
                 label="Background reference image",
                 elem_classes=["image-container", "hide-buttons"]
             )
@@ -428,7 +426,6 @@ with gr.Blocks(css=css) as WeShop:
                     gr.Markdown("#### Step 3: Press 'Generate' to get the result")
                     output = gr.Image(
                         type="pil",
-                        format='png',
                         label="Result",
                         elem_classes=["image-container", "hide-buttons"],
                         interactive=False
@@ -456,4 +453,14 @@ with gr.Blocks(css=css) as WeShop:
                     inputs=[main_image_input, background_image_input, output],
                     elem_id="example-res-images",
                 )
+    main_image_input.upload(
+        fn=preprocess_image,
+        inputs=[main_image_input],
+        outputs=main_image_input
+    )
+    background_image_input.upload(
+        fn=preprocess_background_image,
+        inputs=[background_image_input],
+        outputs=background_image_input
+    )
 WeShop.queue(api_open=False).launch(show_api=False)

assets/title.html

@@ -85,9 +85,7 @@
         <br>
         <div class="content-text">
             <strong>How It Works: </strong>Digitally dress any model in any garment. Upload a clear image of the
-            clothing and a photo of your model, and our AI will seamlessly outfit them. Explore more with <a
-                href="https://www.weshop.ai/tools/virtualtryon" style="font-weight: bold;" target="_blank">WeShopAI
-            Virtual Try-On</a>.
+            clothing and a photo of your model, and our AI will seamlessly outfit them.
         </div>
         <br>
         <div class="content-text">
@@ -101,7 +99,6 @@
             <span class="nav-links">
                 <a href="https://huggingface.co/spaces/WeShopAI/WeShopAI-Fashion-Model-Pose-Change">WeShopAI Fashion Model Pose Change</a>
                 <a href="https://huggingface.co/spaces/WeShopAI/WeShopAI-Bad-Hand-Fixer">WeShopAI Bad Hand Fixer</a>
-                <a href="https://huggingface.co/spaces/WeShopAI/WeShopAI-Swap-Face-And-BG">WeShopAI Swap Face & BG</a>
             </span>
         </div>
         <br>