Update app.py

app.py

@@ -20,12 +20,35 @@ pipe = pipe.to(device)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
+# First set of tags
+tag_options_1 = {
+    "Fantasy": "fantasy",
+    "Sci-Fi": "sci-fi",
+    "Realistic": "realistic",
+    "Cyberpunk": "cyberpunk",
+    "Noir": "noir",
+}
+
+# Second set of tags
+tag_options_2 = {
+    "Vibrant Colors": "vibrant colors",
+    "Dark Theme": "dark theme",
+    "Minimalist": "minimalist",
+    "Photorealistic": "photorealistic",
+    "Abstract": "abstract",
+}
+
 @spaces.GPU  # [uncomment to use ZeroGPU]
-def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, tag_selection, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, tag_selection_1, tag_selection_2, use_tags, progress=gr.Progress(track_tqdm=True)):
 
-    # Combine selected tags with the user input prompt
-    tags_text = ', '.join(tag_selection)
-    final_prompt = f'score_9, score_8_up, score_7_up,source_anime, {tags_text}, {prompt}'
+    # Determine prompt based on the tab selection
+    if use_tags:
+        selected_tags_1 = [tag_options_1[tag] for tag in tag_selection_1 if tag in tag_options_1]
+        selected_tags_2 = [tag_options_2[tag] for tag in tag_selection_2 if tag in tag_options_2]
+        tags_text = ', '.join(selected_tags_1 + selected_tags_2)
+        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}'
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -57,12 +80,6 @@ css = """
 }
 """
 
-# Define a list of example tags
-tag_options = [
-    "fantasy", "sci-fi", "realistic", "cyberpunk", "noir", "surreal", 
-    "colorful", "detailed", "high resolution", "anime style"
-]
-
 with gr.Blocks(css=css) as demo:
 
     with gr.Column(elem_id="col-container"):
@@ -70,23 +87,29 @@ with gr.Blocks(css=css) as demo:
         # Text-to-Image Gradio Template
         """)
 
-        with gr.Row():
-            prompt = gr.Text(
-                label="Prompt",
-                show_label=False,
-                max_lines=1,
-                placeholder="Enter your prompt",
-                container=False,
-            )
+        # Tabbed interface to select either Prompt or Tags
+        with gr.Tabs() as tabs:
+            # Prompt-only tab
+            with gr.TabItem("Prompt Input"):
+                prompt = gr.Text(
+                    label="Prompt",
+                    show_label=False,
+                    max_lines=1,
+                    placeholder="Enter your prompt",
+                    container=False,
+                )
+                use_tags = gr.State(False)  # This keeps track of whether tags are used
 
-            run_button = gr.Button("Run", scale=0)
+            # Tag-based input tab
+            with gr.TabItem("Tag Selection"):
+                with gr.Row():
+                    tag_selection_1 = gr.CheckboxGroup(choices=list(tag_options_1.keys()), label="Select Tags (Style)")
+                    tag_selection_2 = gr.CheckboxGroup(choices=list(tag_options_2.keys()), label="Select Tags (Theme)")
+                use_tags = gr.State(True)  # When this tab is selected, tags are used
 
+        run_button = gr.Button("Run", scale=0)
         result = gr.Image(label="Result", show_label=False)
 
-        with gr.Row():
-            # Checkbox group for selectable tags
-            tag_selection = gr.CheckboxGroup(choices=tag_options, label="Select Tags")
-
         with gr.Accordion("Advanced Settings", open=False):
             negative_prompt = gr.Text(
                 label="Negative prompt",
@@ -143,10 +166,20 @@ with gr.Blocks(css=css) as demo:
             examples=examples,
             inputs=[prompt]
         )
+
+    # Use `run_button.click` event to check which tab is selected
+    def check_tab(prompt, tag_selection_1, tag_selection_2, selected_tab):
+        if selected_tab == "Prompt Input":
+            return False  # Use prompt only
+        return True  # Use tags if on "Tag Selection"
+
+    # Set the selected tab
+    tabs.change(check_tab, inputs=[prompt, tag_selection_1, tag_selection_2, tabs], outputs=use_tags)
+
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
-        inputs=[prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, tag_selection],
+        inputs=[prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, tag_selection_1, tag_selection_2, use_tags],
         outputs=[result, seed]
     )