image format

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 = image.format if image.format else "PNG"
+    image_format = "PNG"
     mime_type = f"image/{image_format.lower()}"
     timestamp = str(int(time.time()))
     signature = generate_signature(
@@ -395,6 +395,7 @@ 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"]
             )
@@ -410,6 +411,7 @@ 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"]
             )
@@ -426,6 +428,7 @@ 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
@@ -453,14 +456,4 @@ 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)