Update app.py

app.py

@@ -102,6 +102,9 @@ with gr.Blocks(css=css) as demo:
         # Add a textbox to display the prompts used for generation
         prompt_info = gr.Textbox(label="Prompts Used", lines=3, interactive=False)
 
+        # State to track active tab
+        active_tab = gr.State("Prompt Input")
+
         # Tabbed interface to select either Prompt or Tags
         with gr.Tabs() as tabs:
             with gr.TabItem("Prompt Input") as prompt_tab:
@@ -112,6 +115,7 @@ with gr.Blocks(css=css) as demo:
                     placeholder="Enter your prompt",
                     container=False,
                 )
+                prompt_tab.select(lambda: "Prompt Input", inputs=None, outputs=active_tab)
 
             with gr.TabItem("Tag Selection") as tag_tab:
                 # Tag selection checkboxes for each tag group
@@ -129,6 +133,7 @@ with gr.Blocks(css=css) as demo:
                 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)
 
         # Full-width "Run" button
         run_button = gr.Button("Run", scale=0, elem_id="run-button")
@@ -196,7 +201,7 @@ with gr.Blocks(css=css) as demo:
                     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],
+                    selected_camera_tags, selected_atmosphere_tags, active_tab],
             outputs=[result, seed, prompt_info]
         )