Optimize UI: Fixed heights, side-by-side layout, removed share footer, deleted APPROACH.md

APPROACH.md

@@ -1,24 +0,0 @@
-# SynthID Bypass: Headless ComfyUI Implementation
-
-This application has been transitioned to a **Headless ComfyUI backend** to ensure a 100% exact replication of the original research.
-
-## Why this change?
-While `diffusers` is a powerful library, the `00quebec/Synthid-Bypass` research relies on a highly specialized stack of technologies:
-1.  **Z-Image-Turbo (S3-DiT)**: A specific architecture that differs from standard Stable Diffusion.
-2.  **Union ControlNet**: A multi-mode ControlNet that handles structural guidance in a unique way.
-3.  **ComfyUI Custom Nodes**: Specifically `Impact Pack` for face restoration and `SeedVR2` for upscaling.
-
-By running the actual ComfyUI engine in the background of the Hugging Face Space, we guarantee:
-- **Identical Model Loading**: Using the exact `.safetensors` files from the research.
-- **Identical Logic**: Processing images through the exact same node graph.
-
-## Architecture
-- **Backend**: Headless ComfyUI server.
-- **Frontend**: Gradio UI acting as a client.
-- **Environment**: Hugging Face ZeroGPU (with model offloading to CPU when idle).
-
-## Deployment Note
-The first run on a fresh Hugging Face Space will involve:
-1. Cloning ComfyUI and 6+ custom node repositories.
-2. Downloading approximately 10GB of model weights.
-This may lead to a long initial "Building" phase, but ensures the most faithful output possible.

app.py

@@ -101,14 +101,37 @@ def remove_watermark(input_image):
     finally:
         proc.terminate()
 
-# Simple UI
-view = gr.Interface(
-    fn=remove_watermark,
-    inputs=gr.Image(type="pil", label="Upload AI Generated Image"),
-    outputs=gr.Image(type="pil", label="Bypass Result"),
-    title="SynthID Bypass (Exact ComfyUI Replication)",
-    description="This Space replicates the research paper 'Synthid-Bypass' using the exact Z-Image-Turbo models. Note: Initial boot takes ~5 minutes to download 10GB of models."
-)
+# Premium UI with Fixed Height and No Share Buttons
+css = """
+#container {
+    max-width: 1200px;
+    margin: 0 auto;
+}
+.image-preview {
+    max-height: 512px !important;
+}
+footer {display: none !important;}
+"""
+
+with gr.Blocks(css=css, title="SynthID Bypass") as demo:
+    with gr.Column(elem_id="container"):
+        gr.Markdown("# SynthID Bypass (Exact ComfyUI Replication)")
+        gr.Markdown("Replicating 'Synthid-Bypass' with exact Z-Image-Turbo models. Note: Initial boot takes ~5 minutes to download 10GB of models.")
+        
+        with gr.Row():
+            with gr.Column():
+                input_img = gr.Image(type="pil", label="Input Image", height=512)
+            with gr.Column():
+                output_img = gr.Image(type="pil", label="Reconstructed Image (Bypass)", height=512, interactive=False)
+        
+        submit_btn = gr.Button("Remove Watermark", variant="primary")
+        
+        submit_btn.click(
+            fn=remove_watermark,
+            inputs=[input_img],
+            outputs=[output_img]
+        )
 
 if __name__ == "__main__":
-    view.launch()
+    # Disable flagging and share to keep UI clean
+    demo.launch(show_api=False)