Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import numpy as np
 import random
-import spaces  # Uncomment if you're using ZeroGPU
-from diffusers import DiffusionPipeline
 import torch
+from diffusers import DiffusionPipeline
+import importlib  # to import tag modules dynamically
 from tags import participant_tags, tribe_tags, role_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"
@@ -21,7 +21,21 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 
-@spaces.GPU  # Uncomment if using ZeroGPU
+# Function to load tags dynamically based on the selected tab
+def load_tags(active_tab):
+    if active_tab == "Gay":
+        tags_module = importlib.import_module('tags_gay')  # dynamically import the tags_gay module
+    elif active_tab == "Straight":
+        tags_module = importlib.import_module('tags_straight')  # dynamically import the tags_straight module
+    elif active_tab == "Lesbian":
+        tags_module = importlib.import_module('tags_lesbian')  # dynamically import the tags_lesbian module
+    else:
+        raise ValueError(f"Unknown tab: {active_tab}")
+    
+    return tags_module
+
+
+@gradio.Interface
 def infer(
     prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
     selected_participant_tags, selected_tribe_tags, selected_role_tags, selected_skin_tone_tags, selected_body_type_tags,
@@ -29,28 +43,46 @@ def infer(
     selected_position_tags, selected_fetish_tags, selected_location_tags, selected_camera_tags, selected_atmosphere_tags,
     active_tab, progress=gr.Progress(track_tqdm=True)
 ):
+    # Dynamically load the correct tags module based on active tab
+    tags_module = load_tags(active_tab)
+
+    # Now use the tags from the loaded module
+    participant_tags = tags_module.participant_tags
+    tribe_tags = tags_module.tribe_tags
+    role_tags = tags_module.role_tags
+    skin_tone_tags = tags_module.skin_tone_tags
+    body_type_tags = tags_module.body_type_tags
+    tattoo_tags = tags_module.tattoo_tags
+    piercing_tags = tags_module.piercing_tags
+    expression_tags = tags_module.expression_tags
+    eye_tags = tags_module.eye_tags
+    hair_style_tags = tags_module.hair_style_tags
+    position_tags = tags_module.position_tags
+    fetish_tags = tags_module.fetish_tags
+    location_tags = tags_module.location_tags
+    camera_tags = tags_module.camera_tags
+    atmosphere_tags = tags_module.atmosphere_tags
+
     # Handle the active tab and generate the prompt accordingly
-    if active_tab == "Prompt Input":
-        final_prompt = f"score_9, score_8_up, score_7_up, source_anime, {prompt}"
-    else:
-        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]
-        )
-        final_prompt = f"score_9, score_8_up, score_7_up, source_anime, {', '.join(tag_list)}"
+    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]
+    )
+
+    final_prompt = f"score_9, score_8_up, score_7_up, source_anime, {', '.join(tag_list)}"
 
     # Concatenate additional negative prompts
     additional_negatives = "worst quality, bad quality, jpeg artifacts, source_cartoon, 3d, (censor), monochrome, blurry, lowres, watermark"
@@ -72,7 +104,7 @@ def infer(
 
     return image, seed, f"Prompt: {final_prompt}\nNegative Prompt: {full_negative_prompt}"
 
-
+# CSS for the layout
 css = """
 #col-container {
     margin: 0 auto;