update_app_v2

app.py

@@ -66,7 +66,7 @@ def call_baseten_generate(
     try:
         if not URL:
             raise ValueError("The URL environment variable is not set.")
-        
+
         response = requests.post(URL, headers=headers, json=payload)
         if response.status_code == 200:
             data = response.json()
@@ -94,6 +94,7 @@ Mode = TypedDict(
         "default_height": int,
         "default_width": int,
         "models": list[str],
+        "remove_bg": bool,
     },
 )
 
@@ -107,9 +108,10 @@ MODE_DEFAULTS: dict[str, Mode] = {
         "models": [
             "zendsd_512_146000",
             "subject_99000_512",
-            #"zen_pers_11000",
+            # "zen_pers_11000",
             "zen_26000_512",
         ],
+        "remove_bg": True,
     },
     "Background Generation": {
         "model": "gen_back_3000_1024",
@@ -119,19 +121,19 @@ MODE_DEFAULTS: dict[str, Mode] = {
         "default_width": 1024,
         "models": [
             "bgwlight_15000_1024",
-            #"rmgb_12000_1024",
+            # "rmgb_12000_1024",
             "bg_canny_58000_1024",
-            #"gen_back_3000_1024",
+            # "gen_back_3000_1024",
             "gen_back_7000_1024",
-            #"gen_bckgnd_18000_512",
-            #"gen_bckgnd_18000_512",
-            #"loose_25000_512",
-            #"looser_23000_1024",
-            #"looser_bg_gen_21000_1280",
-            #"old_looser_46000_1024",
-            #"relight_bg_gen_31000_1024",
-            
+            # "gen_bckgnd_18000_512",
+            # "gen_bckgnd_18000_512",
+            # "loose_25000_512",
+            # "looser_23000_1024",
+            # "looser_bg_gen_21000_1280",
+            # "old_looser_46000_1024",
+            # "relight_bg_gen_31000_1024",
         ],
+        "remove_bg": True,
     },
     "Canny": {
         "model": "canny_21000_1024",
@@ -140,6 +142,7 @@ MODE_DEFAULTS: dict[str, Mode] = {
         "default_height": 1024,
         "default_width": 1024,
         "models": ["canny_21000_1024"],
+        "remove_bg": True,
     },
     "Depth": {
         "model": "depth_9800_1024",
@@ -150,6 +153,7 @@ MODE_DEFAULTS: dict[str, Mode] = {
         "models": [
             "depth_9800_1024",
         ],
+        "remove_bg": True,
     },
     "Deblurring": {
         "model": "slight_deblurr_18000",
@@ -157,30 +161,20 @@ MODE_DEFAULTS: dict[str, Mode] = {
         "default_strength": 1.2,
         "default_height": 1024,
         "default_width": 1024,
-        "models": [#"slight_deblurr_18000", 
-                   "deblurr_1024_10000"],
+        "models": ["deblurr_1024_10000"],  # "slight_deblurr_18000",
+        "remove_bg": False,
     },
 }
 
 
 header = """
 # 🌍 ZenCtrl / FLUX
-<div style="text-align: center; display: flex; justify-content: left; gap: 5px;">
-    <a href="https://huggingface.co/fotographerai/zenctrl_tools">
-    <img src="https://img.shields.io/badge/🤗-Model-ffbd45.svg" alt="Weights">
-    </a>
-    <a href="https://github.com/FotographerAI/ZenCtrl">
-    <img src="https://img.shields.io/badge/GitHub-Code-blue.svg?logo=github" alt="GitHub">
-    </a>
-    <a href="https://fotographer.ai/">
-    <img src="https://img.shields.io/badge/LP-Visit-9cf" alt="LP">
-    </a>
-    <a href="https://x.com/fotographerait">
-    <img src="https://img.shields.io/twitter/follow/FotographerAI?style=social" alt="Twitter">
-    </a>
-    <a href="https://discord.com/invite/b9RuYQ3F8k">
-    <img src="https://img.shields.io/badge/Discord-Join-7289da.svg?logo=discord" alt="Discord">
-    </a>
+<div align="center" style="line-height: 1;">
+    <a href="https://github.com/FotographerAI/ZenCtrl/tree/main" target="_blank" style="margin: 2px;" name="github_repo_link"><img src="https://img.shields.io/badge/GitHub-Repo-181717.svg" alt="GitHub Repo" style="display: inline-block; vertical-align: middle;"></a>
+    <a href="https://huggingface.co/spaces/fotographerai/ZenCtrl" target="_blank" name="huggingface_space_link"><img src="https://img.shields.io/badge/🤗_HuggingFace-Space-ffbd45.svg" alt="HuggingFace Space" style="display: inline-block; vertical-align: middle;"></a>
+    <a href="https://discord.com/invite/b9RuYQ3F8k" target="_blank" style="margin: 2px;" name="discord_link"><img src="https://img.shields.io/badge/Discord-Join-7289da.svg?logo=discord" alt="Discord" style="display: inline-block; vertical-align: middle;"></a>
+    <a href="https://fotographer.ai/" target="_blank" style="margin: 2px;" name="lp_link"><img src="https://img.shields.io/badge/Website-Landing_Page-blue" alt="LP" style="display: inline-block; vertical-align: middle;"></a>
+    <a href="https://x.com/FotographerAI" target="_blank" style="margin: 2px;" name="twitter_link"><img src="https://img.shields.io/twitter/follow/FotographerAI?style=social" alt="X" style="display: inline-block; vertical-align: middle;"></a>
 </div>
 """
 
@@ -231,7 +225,7 @@ with gr.Blocks(title="🌍 ZenCtrl") as demo:
                                 interactive=True,
                             )
                             remove_bg_checkbox = gr.Checkbox(
-                                label="Remove Background", value=False
+                                label="Remove Background", value=defaults["remove_bg"]
                             )
 
                     with gr.Column(scale=2):