update app

app.py

@@ -12,73 +12,6 @@ from diffusers.utils import load_image
 from huggingface_hub import hf_hub_download
 from aura_sr import AuraSR
 from gradio_imageslider import ImageSlider
-from gradio.themes import Soft
-from gradio.themes.utils import colors, fonts, sizes
-
-# --- Theme and CSS Definition ---
-
-# Define the new HotPink color palette
-colors.hot_pink = colors.Color(
-    name="hot_pink",
-    c50="#FFF0F7",
-    c100="#FFE0EF",
-    c200="#FFC2E0",
-    c300="#FFA4D1",
-    c400="#FF87C3",
-    c500="#FF69B4",  # HotPink base color
-    c600="#E65FA2",
-    c700="#CC5490",
-    c800="#B34A7E",
-    c900="#993F6C",
-    c950="#80355A",
-)
-
-class HotPinkTheme(Soft):
-    def __init__(
-        self,
-        *,
-        primary_hue: colors.Color | str = colors.gray,
-        secondary_hue: colors.Color | str = colors.hot_pink, # Use the new color
-        neutral_hue: colors.Color | str = colors.slate,
-        text_size: sizes.Size | str = sizes.text_lg,
-        font: fonts.Font | str | Iterable[fonts.Font | str] = (
-            fonts.GoogleFont("Outfit"), "Arial", "sans-serif",
-        ),
-        font_mono: fonts.Font | str | Iterable[fonts.Font | str] = (
-            fonts.GoogleFont("IBM Plex Mono"), "ui-monospace", "monospace",
-        ),
-    ):
-        super().__init__(
-            primary_hue=primary_hue,
-            secondary_hue=secondary_hue,
-            neutral_hue=neutral_hue,
-            text_size=text_size,
-            font=font,
-            font_mono=font_mono,
-        )
-        super().set(
-            background_fill_primary="*primary_50",
-            background_fill_primary_dark="*primary_900",
-            body_background_fill="linear-gradient(135deg, *primary_200, *primary_100)",
-            body_background_fill_dark="linear-gradient(135deg, *primary_900, *primary_800)",
-            button_primary_text_color="white",
-            button_primary_text_color_hover="white",
-            button_primary_background_fill="linear-gradient(90deg, *secondary_500, *secondary_600)",
-            button_primary_background_fill_hover="linear-gradient(90deg, *secondary_600, *secondary_700)",
-            button_primary_background_fill_dark="linear-gradient(90deg, *secondary_600, *secondary_700)",
-            button_primary_background_fill_hover_dark="linear-gradient(90deg, *secondary_500, *secondary_600)",
-            slider_color="*secondary_500",
-            slider_color_dark="*secondary_600",
-            block_title_text_weight="600",
-            block_border_width="2px",
-            block_shadow="*shadow_drop_lg",
-            button_primary_shadow="*shadow_drop_lg",
-            button_large_padding="12px",
-            block_label_background_fill="*primary_200",
-        )
-
-# Instantiate the new theme
-hot_pink_theme = HotPinkTheme()
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -104,7 +37,7 @@ pipe.load_lora_weights("prithivMLmods/Photo-Restore-i2i", weight_name="Photo-Res
 pipe.load_lora_weights("prithivMLmods/Polaroid-Warm-i2i", weight_name="Polaroid-Warm-i2i.safetensors", adapter_name="polaroid")
 pipe.load_lora_weights("prithivMLmods/Monochrome-Pencil", weight_name="Monochrome-Pencil-i2i.safetensors", adapter_name="pencil")
 pipe.load_lora_weights("prithivMLmods/LZO-1-Preview", weight_name="LZO-1-Preview.safetensors", adapter_name="lzo")
-
+pipe.load_lora_weights("prithivMLmods/Kontext-Watermark-Remover", weight_name="Kontext-Watermark-Remover.safetensors", adapter_name="watermark-remover")
 
 # --- Upscaler Model Initialization ---
 aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2")
@@ -127,7 +60,9 @@ def infer(input_image, prompt, lora_adapter, upscale_image, seed=42, randomize_s
         pipe.set_adapters(["pencil"], adapter_weights=[1.0])
     elif lora_adapter == "LZO-Zoom":
         pipe.set_adapters(["lzo"], adapter_weights=[1.0])
-
+    elif lora_adapter == "Kontext-Watermark-Remover":
+        pipe.set_adapters(["watermark-remover"], adapter_weights=[1.0])
+        
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     
@@ -165,7 +100,7 @@ css="""
 #main-title h1 {font-size: 2.1em !important;}
 """
 
-with gr.Blocks(css=css, theme=hot_pink_theme) as demo:
+with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **Photo-Mate-i2i**", elem_id="main-title")
@@ -218,7 +153,7 @@ with gr.Blocks(css=css, theme=hot_pink_theme) as demo:
                 with gr.Row():
                     lora_adapter = gr.Dropdown(
                         label="Chosen LoRA",
-                        choices=["PhotoCleanser", "PhotoRestorer", "PolaroidWarm", "MonochromePencil", "LZO-Zoom"],
+                        choices=["PhotoCleanser", "PhotoRestorer", "PolaroidWarm", "MonochromePencil", "LZO-Zoom", "Kontext-Watermark-Remover"],
                         value="PhotoCleanser"
                     )
                     
@@ -229,6 +164,7 @@ with gr.Blocks(css=css, theme=hot_pink_theme) as demo:
             examples=[
                 ["photocleanser/2.png", "[photo content], remove the cat from the image while preserving the background and remaining elements, maintaining realism and original details.", "PhotoCleanser"],
                 ["photocleanser/1.png", "[photo content], remove the football from the image while preserving the background and remaining elements, maintaining realism and original details.", "PhotoCleanser"],
+                ["watermark/12.jpeg", "[photo content], remove any watermark text or logos from the image while preserving the background, texture, lighting, and overall realism. Ensure the edited areas blend seamlessly with surrounding details, leaving no visible traces of watermark removal.", "Kontext-Watermark-Remover"],
                 ["photorestore/1.png", "[photo content], restore and enhance the image by repairing any damage, scratches, or fading. Colorize the photo naturally while preserving authentic textures and details, maintaining a realistic and historically accurate look.", "PhotoRestorer"],
                 ["lzo/1.jpg", "[photo content], zoom in on the specified [face close-up], enhancing resolution and detail while preserving sharpness, realism, and original context. Maintain natural proportions and background continuity around the zoomed area.", "LZO-Zoom"],
                 ["photorestore/2.png", "[photo content], restore and enhance the image by repairing any damage, scratches, or fading. Colorize the photo naturally while preserving authentic textures and details, maintaining a realistic and historically accurate look.", "PhotoRestorer"],