Update app.py

app.py

@@ -33,6 +33,7 @@ def load_tags(active_tab):
     
     return tags_module
 
+@spaces.GPU
 @spaces.GPU
 def infer(
     prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
@@ -61,25 +62,44 @@ def infer(
     camera_tags = tags_module.camera_tags
     atmosphere_tags = tags_module.atmosphere_tags
 
-    # Handle the active tab and generate the prompt accordingly
-    tag_list = (
-        [participant_tags[tag] for tag in selected_participant_tags] +
-        [tribe_tags[tag] for tag in selected_tribe_tags] +
-        [role_tags[tag] for tag in selected_role_tags] +
-        [skin_tone_tags[tag] for tag in selected_skin_tone_tags] +
-        [body_type_tags[tag] for tag in selected_body_type_tags] +
-        [tattoo_tags[tag] for tag in selected_tattoo_tags] +
-        [piercing_tags[tag] for tag in selected_piercing_tags] +
-        [expression_tags[tag] for tag in selected_expression_tags] +
-        [eye_tags[tag] for tag in selected_eye_tags] +
-        [hair_style_tags[tag] for tag in selected_hair_style_tags] +
-        [position_tags[tag] for tag in selected_position_tags] +
-        [fetish_tags[tag] for tag in selected_fetish_tags] +
-        [location_tags[tag] for tag in selected_location_tags] +
-        [camera_tags[tag] for tag in selected_camera_tags] +
-        [atmosphere_tags[tag] for tag in selected_atmosphere_tags]
-    )
+    # Debug: Check if the tag modules are loaded correctly
+    print(f"Loaded tags for {active_tab}:")
+    print(f"participant_tags: {list(participant_tags.keys())}")
+    print(f"tribe_tags: {list(tribe_tags.keys())}")
+    print(f"role_tags: {list(role_tags.keys())}")
+    print(f"skin_tone_tags: {list(skin_tone_tags.keys())}")
 
+    # Handle the active tab and generate the prompt accordingly
+    tag_list = []
+
+    # Dynamically build tag list based on selected checkboxes
+    for tag_list_group, selected_tags in [
+        (participant_tags, selected_participant_tags),
+        (tribe_tags, selected_tribe_tags),
+        (role_tags, selected_role_tags),
+        (skin_tone_tags, selected_skin_tone_tags),
+        (body_type_tags, selected_body_type_tags),
+        (tattoo_tags, selected_tattoo_tags),
+        (piercing_tags, selected_piercing_tags),
+        (expression_tags, selected_expression_tags),
+        (eye_tags, selected_eye_tags),
+        (hair_style_tags, selected_hair_style_tags),
+        (position_tags, selected_position_tags),
+        (fetish_tags, selected_fetish_tags),
+        (location_tags, selected_location_tags),
+        (camera_tags, selected_camera_tags),
+        (atmosphere_tags, selected_atmosphere_tags)
+    ]:
+        # Add the selected tags for this category
+        for tag in selected_tags:
+            if tag in tag_list_group:
+                tag_list.append(tag_list_group[tag])
+            else:
+                print(f"Tag '{tag}' not found in {tag_list_group}")
+
+    # Debug: Print the final prompt that is being generated
+    print(f"Final Prompt: {final_prompt}")
+    
     final_prompt = f"score_9, score_8_up, score_7_up, source_anime, {', '.join(tag_list)}"
 
     # Concatenate additional negative prompts
@@ -90,6 +110,9 @@ def infer(
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
 
+    # Debug: Ensure the final negative prompt is correct
+    print(f"Negative Prompt: {full_negative_prompt}")
+
     image = pipe(
         prompt=final_prompt,
         negative_prompt=full_negative_prompt,
@@ -100,8 +123,12 @@ def infer(
         generator=generator
     ).images[0]
 
+    # Debug: Confirm the seed and image generation
+    print(f"Seed used for generation: {seed}")
+    
     return image, seed, f"Prompt: {final_prompt}\nNegative Prompt: {full_negative_prompt}"
 
+
 # CSS for the layout
 css = """
 #col-container {