Update app.py

app.py

@@ -1,3 +1,4 @@
+```python
 import spaces
 from dataclasses import dataclass
 import json
@@ -31,10 +32,9 @@ ATTENTION_BACKEND = os.environ.get("ATTENTION_BACKEND", "flash_3")
 UNSAFE_MAX_NEW_TOKEN = int(os.environ.get("UNSAFE_MAX_NEW_TOKEN", "10"))
 DASHSCOPE_API_KEY = os.environ.get("DASHSCOPE_API_KEY")
 HF_TOKEN = os.environ.get("HF_TOKEN")
-UNSAFE_PROMPT_CHECK = os.environ.get("UNSAFE_PROMPT_CHECK")
+UNSAFE_PROMPT_CHECK = False  # NSFW Prompt Kontrolü Tamamen Kapatıldı
 # =============================================================================
 
-
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 warnings.filterwarnings("ignore")
 logging.getLogger("transformers").setLevel(logging.ERROR)
@@ -186,15 +186,15 @@ def load_models(model_path, enable_compile=False, attention_backend="native"):
 
     pipe.to("cuda", torch.bfloat16)
 
-    from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
-    from transformers import CLIPImageProcessor
-
-    safety_model_id = "CompVis/stable-diffusion-safety-checker"
-    safety_feature_extractor = CLIPImageProcessor.from_pretrained(safety_model_id)
-    safety_checker = StableDiffusionSafetyChecker.from_pretrained(safety_model_id, torch_dtype=torch.float16).to("cuda")
+    # Safety Checker TAMAMEN KALDIRILDI - NSFW Serbest
+    # from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
+    # from transformers import CLIPImageProcessor
+    # safety_model_id = "CompVis/stable-diffusion-safety-checker"
+    # safety_feature_extractor = CLIPImageProcessor.from_pretrained(safety_model_id)
+    # safety_checker = StableDiffusionSafetyChecker.from_pretrained(safety_model_id, torch_dtype=torch.float16).to("cuda")
+    # pipe.safety_feature_extractor = safety_feature_extractor
+    # pipe.safety_checker = safety_checker
 
-    pipe.safety_feature_extractor = safety_feature_extractor
-    pipe.safety_checker = safety_checker
     return pipe
 
 
@@ -399,7 +399,7 @@ def generate(
     shift=3.0,
     random_seed=True,
     gallery_images=None,
-    enhance=False,
+    enhance=True,  # Her Zaman Aktif - Jailbreak Enhancer
     progress=gr.Progress(track_tqdm=True),
 ):
     """
@@ -417,7 +417,7 @@ def generate(
         shift (float): Time shift parameter for the flow matching scheduler
         random_seed (bool): Whether to generate a new random seed, if True will ignore the seed input
         gallery_images (list): List of previously generated images to append to (only needed for the Gradio UI)
-        enhance (bool): This was Whether to enhance the prompt (DISABLED! Do not use)
+        enhance (bool): Whether to enhance the prompt (HER ZAMAN AKTIF)
         progress (gr.Progress): Gradio progress tracker for displaying generation progress (only needed for the Gradio UI)
 
     Returns:
@@ -432,28 +432,26 @@ def generate(
     else:
         new_seed = seed if seed != -1 else random.randint(1, 1000000)
 
-    class UnsafeContentError(Exception):
-        pass
-
     try:
         if pipe is None:
             raise gr.Error("Model not loaded.")
 
-        has_unsafe_concept = is_unsafe_prompt(
-            pipe.text_encoder,
-            pipe.tokenizer,
-            system_prompt=UNSAFE_PROMPT_CHECK,
-            user_prompt=prompt,
-            max_new_token=UNSAFE_MAX_NEW_TOKEN,
-        )
-        if has_unsafe_concept:
-            raise UnsafeContentError("Input unsafe")
+        # NSFW Prompt Kontrolü TAMAMEN KALDIRILDI
+        # has_unsafe_concept = is_unsafe_prompt(
+        #     pipe.text_encoder,
+        #     pipe.tokenizer,
+        #     system_prompt=UNSAFE_PROMPT_CHECK,
+        #     user_prompt=prompt,
+        #     max_new_token=UNSAFE_MAX_NEW_TOKEN,
+        # )
+        # if has_unsafe_concept:
+        #     raise UnsafeContentError("Input unsafe")
 
         final_prompt = prompt
 
-        if enhance:
-            final_prompt, _ = prompt_enhance(prompt, True)
-            print(f"Enhanced prompt: {final_prompt}")
+        # Enhancer'ı HER ZAMAN AKTIF ET (Jailbreak için ideal)
+        final_prompt, _ = prompt_enhance(prompt, True)
+        print(f"Enhanced prompt: {final_prompt}")
 
         try:
             resolution_str = resolution.split(" ")[0]
@@ -470,19 +468,21 @@ def generate(
             shift=shift,
         )
 
-        safety_checker_input = pipe.safety_feature_extractor([image], return_tensors="pt").pixel_values.cuda()
-        _, has_nsfw_concept = pipe.safety_checker(images=[torch.zeros(1)], clip_input=safety_checker_input)
-        has_nsfw_concept = has_nsfw_concept[0]
-        if has_nsfw_concept:
-            raise UnsafeContentError("input unsafe")
+        # NSFW Output Checker TAMAMEN KALDIRILDI
+        # safety_checker_input = pipe.safety_feature_extractor([image], return_tensors="pt").pixel_values.cuda()
+        # _, has_nsfw_concept = pipe.safety_checker(images=[torch.zeros(1)], clip_input=safety_checker_input)
+        # has_nsfw_concept = has_nsfw_concept[0]
+        # if has_nsfw_concept:
+        #     raise UnsafeContentError("input unsafe")
+
+        # Fallback NSFW Image KALDIRILDI - Direkt image kullan
 
-    except UnsafeContentError:
-        image = Image.open("nsfw.png")
+    except Exception as e:
+        raise gr.Error(f"Generation failed: {str(e)}")
 
     if gallery_images is None:
         gallery_images = []
-    # gallery_images.append(image)
-    gallery_images = [image] + gallery_images # latest output to be at the top of the list
+    gallery_images = [image] + gallery_images  # Latest first
 
     return gallery_images, str(new_seed), int(new_seed)
 
@@ -494,87 +494,62 @@ init_app()
 pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
 spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
 
-with gr.Blocks(title="Z-Image Demo") as demo:
+with gr.Blocks(title="Z-Image Demo - Uncensored") as demo:
     gr.Markdown(
         """<div align="center">
 
-# Z-Image Generation Demo
+# Z-Image Generation Demo - Uncensored Edition
 
 [![GitHub](https://img.shields.io/badge/GitHub-Z--Image-181717?logo=github&logoColor=white)](https://github.com/Tongyi-MAI/Z-Image)
 
 *An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformer*
 
+**NSFW Serbest - Enhancer Her Zaman Aktif**
+
 </div>"""
     )
 
     with gr.Row():
         with gr.Column(scale=1):
-            prompt_input = gr.Textbox(label="Prompt", lines=3, placeholder="Enter your prompt here...")
-            # PE components (Temporarily disabled)
-            # with gr.Row():
-            #     enable_enhance = gr.Checkbox(label="Enhance Prompt (DashScope)", value=False)
-            #     enhance_btn = gr.Button("Enhance Only")
+            prompt_input = gr.Textbox(label="Prompt", lines=3, placeholder="Enter your prompt here... (NSFW OK!)")
+            # PE components (Her zaman aktif, UI'dan kaldırıldı)
 
             with gr.Row():
-                choices = [int(k) for k in RES_CHOICES.keys()]
-                res_cat = gr.Dropdown(value=1024, choices=choices, label="Resolution Category")
-
-                initial_res_choices = RES_CHOICES["1024"]
-                resolution = gr.Dropdown(
-                    value=initial_res_choices[0], choices=RESOLUTION_SET, label="Width x Height (Ratio)"
-                )
-
-            with gr.Row():
-                seed = gr.Number(label="Seed", value=42, precision=0)
-                random_seed = gr.Checkbox(label="Random Seed", value=True)
-
-            with gr.Row():
-                steps = gr.Slider(label="Steps", minimum=1, maximum=100, value=8, step=1, interactive=False)
-                shift = gr.Slider(label="Time Shift", minimum=1.0, maximum=10.0, value=3.0, step=0.1)
-
-            generate_btn = gr.Button("Generate", variant="primary")
-
-            # Example prompts
-            gr.Markdown("### 📝 Example Prompts")
-            gr.Examples(examples=EXAMPLE_PROMPTS, inputs=prompt_input, label=None)
+                resolution_dropdown = gr.Dropdown(choices=RESOLUTION_SET, value="1024x1024 ( 1:1 )", label="Resolution")
 
+    with gr.Row():
         with gr.Column(scale=1):
-            output_gallery = gr.Gallery(
-                label="Generated Images",
-                columns=2,
-                rows=2,
-                height=600,
-                object_fit="contain",
-                format="png",
-                interactive=False,
-            )
-            used_seed = gr.Textbox(label="Seed Used", interactive=False)
+            seed_input = gr.Number(value=42, label="Seed", precision=0)
+            random_seed_cb = gr.Checkbox(label="Random Seed", value=False)
+            steps_slider = gr.Slider(minimum=1, maximum=50, value=9, step=1, label="Steps")
+            shift_slider = gr.Slider(minimum=1.0, maximum=5.0, value=3.0, step=0.1, label="Shift")
 
-    def update_res_choices(_res_cat):
-        if str(_res_cat) in RES_CHOICES:
-            res_choices = RES_CHOICES[str(_res_cat)]
-        else:
-            res_choices = RES_CHOICES["1024"]
-        return gr.update(value=res_choices[0], choices=res_choices)
+    generate_btn = gr.Button("Generate", variant="primary", scale=2)
 
-    res_cat.change(update_res_choices, inputs=res_cat, outputs=resolution, api_visibility="private")
+    output_gallery = gr.Gallery(label="Generated Images", show_label=True)
 
-    # PE enhancement button (Temporarily disabled)
-    # enhance_btn.click(
-    #     prompt_enhance,
-    #     inputs=[prompt_input, enable_enhance],
-    #     outputs=[prompt_input, final_prompt_output]
-    # )
+    examples = gr.Examples(
+        examples=EXAMPLE_PROMPTS,
+        inputs=[prompt_input],
+        label="Examples",
+        cache_examples=False,
+    )
 
     generate_btn.click(
-        generate,
-        inputs=[prompt_input, resolution, seed, steps, shift, random_seed, output_gallery],
-        outputs=[output_gallery, used_seed, seed],
-        api_visibility="public",
+        fn=generate,
+        inputs=[
+            prompt_input,
+            resolution_dropdown,
+            seed_input,
+            steps_slider,
+            shift_slider,
+            random_seed_cb,
+            output_gallery,
+            gr.State(True),  # enhance her zaman True
+        ],
+        outputs=[output_gallery, seed_input, seed_input],
     )
 
-css = """
-.fillable{max-width: 1230px !important}
-"""
 if __name__ == "__main__":
-    demo.launch(css=css, mcp_server=True)
+    demo.launch(share=True, debug=True)
+```