bugfix

app.py

@@ -222,6 +222,7 @@ def process(
     image = image.resize((width, height), Image.LANCZOS)
     mask = mask.resize((width, height), Image.LANCZOS)
     mask = process_mask(mask, mask_inflation=mask_inflation_slider, mask_blur=mask_blur_slider)
+
     image = run_flux(
         image=image,
         mask=mask,
@@ -237,8 +238,9 @@ def process(
     )
 
     if upload_to_r2:
-        url = upload_image_to_r2(image, account_id, access_key, secret_key, bucket)
-        result = {"status": "success", "url": url}
+        with calculateDuration("upload image"):
+            url = upload_image_to_r2(image, account_id, access_key, secret_key, bucket)
+            result = {"status": "success", "url": url}
     else:
         result = {"status": "success", "message": "Image generated but not uploaded"}
 
@@ -369,8 +371,7 @@ with gr.Blocks() as demo:
                     secret_key = gr.Textbox(label="Secret Key", placeholder="Enter R2 secret key here")
                 
         with gr.Column():
-
-            output_json_component = gr.Textbox()
+            output_json_component = gr.Code(label="JSON Result", language="json")
 
     submit_button_component.click(
         fn=process,