Update app.py

app.py

@@ -30,8 +30,9 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
     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
+    
+    elif active_tab == "Straight" :
+        # Use tags from the "Gay" tab
         selected_tags = (
             [participant_tags[tag] for tag in selected_participant_tags] +
             [tribe_tags[tag] for tag in selected_tribe_tags] +
@@ -51,6 +52,48 @@ 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}'
 
+    elif active_tab == "Gay" :
+        # Use tags from the "Gay" tab
+        selected_tags = (
+            [participant_tags[tag] for tag in selected_participant_tags] +
+            [tribe_tags[tag] for tag in selected_tribe_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]
+        )
+        tags_text = ', '.join(selected_tags)
+        final_prompt = f'score_9, score_8_up, score_7_up, source_anime, {tags_text}'
+        
+    elif active_tab == "Lesbian" :
+        # Use tags from the "Lesbien" tab
+        selected_tags = (
+            [participant_tags[tag] for tag in selected_participant_tags] +
+            [tribe_tags[tag] for tag in selected_tribe_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]
+        )
+        tags_text = ', '.join(selected_tags)
+        final_prompt = f'score_9, score_8_up, score_7_up, source_anime, {tags_text}'
+    
     # Concatenate user-provided negative prompt with additional restrictions
     additional_negatives = "worst quality, bad quality, jpeg artifacts, source_cartoon, 3d, (censor), monochrome, blurry, lowres, watermark"
     full_negative_prompt = f"{additional_negatives}, {negative_prompt}"
@@ -94,7 +137,7 @@ css = """
 with gr.Blocks(css=css) as demo:
 
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("""# Text-to-Image Gradio Template""")
+        gr.Markdown("""# Rainbow Media X""")
 
         # Display result image at the top
         result = gr.Image(label="Result", show_label=False)
@@ -117,7 +160,44 @@ with gr.Blocks(css=css) as demo:
                 )
                 prompt_tab.select(lambda: "Prompt Input", inputs=None, outputs=active_tab)
 
-            with gr.TabItem("Tag Selection") as tag_tab:
+            with gr.TabItem("Straight") as straight_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")
+                selected_skin_tone_tags = gr.CheckboxGroup(choices=list(skin_tone_tags.keys()), label="Skin Tone Tags")
+                selected_body_type_tags = gr.CheckboxGroup(choices=list(body_type_tags.keys()), label="Body Type Tags")
+                selected_tattoo_tags = gr.CheckboxGroup(choices=list(tattoo_tags.keys()), label="Tattoo Tags")
+                selected_piercing_tags = gr.CheckboxGroup(choices=list(piercing_tags.keys()), label="Piercing Tags")
+                selected_expression_tags = gr.CheckboxGroup(choices=list(expression_tags.keys()), label="Expression Tags")
+                selected_eye_tags = gr.CheckboxGroup(choices=list(eye_tags.keys()), label="Eye Tags")
+                selected_hair_style_tags = gr.CheckboxGroup(choices=list(hair_style_tags.keys()), label="Hair Style Tags")
+                selected_position_tags = gr.CheckboxGroup(choices=list(position_tags.keys()), label="Position Tags")
+                selected_fetish_tags = gr.CheckboxGroup(choices=list(fetish_tags.keys()), label="Fetish Tags")
+                selected_location_tags = gr.CheckboxGroup(choices=list(location_tags.keys()), label="Location Tags")
+                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")
+                tag_tab.select(lambda: "Tag Selection", inputs=None, outputs=active_tab)
+                
+            
+            with gr.TabItem("Gay") as gay_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")
+                selected_skin_tone_tags = gr.CheckboxGroup(choices=list(skin_tone_tags.keys()), label="Skin Tone Tags")
+                selected_body_type_tags = gr.CheckboxGroup(choices=list(body_type_tags.keys()), label="Body Type Tags")
+                selected_tattoo_tags = gr.CheckboxGroup(choices=list(tattoo_tags.keys()), label="Tattoo Tags")
+                selected_piercing_tags = gr.CheckboxGroup(choices=list(piercing_tags.keys()), label="Piercing Tags")
+                selected_expression_tags = gr.CheckboxGroup(choices=list(expression_tags.keys()), label="Expression Tags")
+                selected_eye_tags = gr.CheckboxGroup(choices=list(eye_tags.keys()), label="Eye Tags")
+                selected_hair_style_tags = gr.CheckboxGroup(choices=list(hair_style_tags.keys()), label="Hair Style Tags")
+                selected_position_tags = gr.CheckboxGroup(choices=list(position_tags.keys()), label="Position Tags")
+                selected_fetish_tags = gr.CheckboxGroup(choices=list(fetish_tags.keys()), label="Fetish Tags")
+                selected_location_tags = gr.CheckboxGroup(choices=list(location_tags.keys()), label="Location Tags")
+                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")
+                tag_tab.select(lambda: "Tag Selection", inputs=None, outputs=active_tab)
+
+            with gr.TabItem("Lesbian") as lesbian_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")