Update app.py

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 import numpy as np
 import random
 import spaces  # [uncomment to use ZeroGPU]
-from diffusers import DiffusionPipeline, DPMSolverSDEScheduler
+from diffusers import DiffusionPipeline
 import torch
-from tags import participant_tags, tribe_tags, skin_tone_tags, body_type_tags, tattoo_tags, piercing_tags, expression_tags, eye_tags, hair_style_tags, position_tags, fetish_tags, location_tags, camera_tags, atmosphere_tags  # Import tags here
+from tags import participant_tags, tribe_tags, skin_tone_tags, body_type_tags, tattoo_tags, piercing_tags, expression_tags, eye_tags, hair_style_tags, position_tags, fetish_tags, location_tags, camera_tags, atmosphere_tags
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "John6666/wai-ani-nsfw-ponyxl-v8-sdxl"  # Replace with your desired model
@@ -22,13 +22,16 @@ MAX_IMAGE_SIZE = 1024
 
 @spaces.GPU  # [uncomment to use ZeroGPU]
 def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
-          selected_participant_tags, selected_tribe_tags, selected_skin_tone_tags, selected_body_type_tags, 
-          selected_tattoo_tags, selected_piercing_tags, selected_expression_tags, selected_eye_tags, 
-          selected_hair_style_tags, selected_position_tags, selected_fetish_tags, selected_location_tags, 
-          selected_camera_tags, selected_atmosphere_tags, use_tags, progress=gr.Progress(track_tqdm=True)):
+          selected_participant_tags, selected_tribe_tags, selected_skin_tone_tags, selected_body_type_tags,
+          selected_tattoo_tags, selected_piercing_tags, selected_expression_tags, selected_eye_tags,
+          selected_hair_style_tags, selected_position_tags, selected_fetish_tags, selected_location_tags,
+          selected_camera_tags, selected_atmosphere_tags, active_tab, progress=gr.Progress(track_tqdm=True)):
 
-    # Construct final prompt using selected tags if `use_tags` is True
-    if use_tags:
+    if active_tab == "Prompt Input":
+        # Use the user-provided prompt
+        final_prompt = f'score_9, score_8_up, score_7_up, source_anime, {prompt}'
+    else:
+        # Use tags from the "Tag Selection" tab
         selected_tags = (
             [participant_tags[tag] for tag in selected_participant_tags] +
             [tribe_tags[tag] for tag in selected_tribe_tags] +
@@ -47,8 +50,6 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         )
         tags_text = ', '.join(selected_tags)
         final_prompt = f'score_9, score_8_up, score_7_up, source_anime, {tags_text}'
-    else:
-        final_prompt = f'score_9, score_8_up, score_7_up, source_anime, {prompt}'
 
     # Concatenate user-provided negative prompt with additional restrictions
     additional_negatives = "worst quality, bad quality, jpeg artifacts, source_cartoon, 3d, (censor), monochrome, blurry, lowres, watermark"
@@ -103,7 +104,7 @@ with gr.Blocks(css=css) as demo:
 
         # Tabbed interface to select either Prompt or Tags
         with gr.Tabs() as tabs:
-            with gr.TabItem("Prompt Input"):
+            with gr.TabItem("Prompt Input") as prompt_tab:
                 prompt = gr.Textbox(
                     label="Prompt",
                     show_label=False,
@@ -111,9 +112,8 @@ with gr.Blocks(css=css) as demo:
                     placeholder="Enter your prompt",
                     container=False,
                 )
-                use_tags = gr.State(False)
 
-            with gr.TabItem("Tag Selection"):
+            with gr.TabItem("Tag Selection") as tag_tab:
                 # Tag selection checkboxes for each tag group
                 selected_participant_tags = gr.CheckboxGroup(choices=list(participant_tags.keys()), label="Participant Tags")
                 selected_tribe_tags = gr.CheckboxGroup(choices=list(tribe_tags.keys()), label="Tribe Tags")
@@ -130,8 +130,6 @@ with gr.Blocks(css=css) as demo:
                 selected_camera_tags = gr.CheckboxGroup(choices=list(camera_tags.keys()), label="Camera Tags")
                 selected_atmosphere_tags = gr.CheckboxGroup(choices=list(atmosphere_tags.keys()), label="Atmosphere Tags")
 
-                use_tags = gr.State(True)
-
         # Full-width "Run" button
         run_button = gr.Button("Run", scale=0, elem_id="run-button")
 
@@ -194,11 +192,11 @@ with gr.Blocks(css=css) as demo:
 
         run_button.click(
             infer,
-            inputs=[prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, 
-                    selected_participant_tags, selected_tribe_tags, selected_skin_tone_tags, selected_body_type_tags, 
-                    selected_tattoo_tags, selected_piercing_tags, selected_expression_tags, selected_eye_tags, 
-                    selected_hair_style_tags, selected_position_tags, selected_fetish_tags, selected_location_tags, 
-                    selected_camera_tags, selected_atmosphere_tags, use_tags],
+            inputs=[prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
+                    selected_participant_tags, selected_tribe_tags, selected_skin_tone_tags, selected_body_type_tags,
+                    selected_tattoo_tags, selected_piercing_tags, selected_expression_tags, selected_eye_tags,
+                    selected_hair_style_tags, selected_position_tags, selected_fetish_tags, selected_location_tags,
+                    selected_camera_tags, selected_atmosphere_tags, tabs],
             outputs=[result, seed, prompt_info]
         )