Update app.py

app.py

@@ -117,7 +117,25 @@ js="""
     <script>console.log("BOOOOOOOOOOOOOOOOBS");</script>
 """
 
-with gr.Blocks(fill_width=True, head=js) as demo:
+css="""
+    .Running {
+        background-color: green;
+        color: white;
+    }
+    .off {
+        background-color: red;
+        color: white;
+    }
+"""
+
+def toggle_button(butState):
+    if butState == "Running":
+        return "Off", gr.Button(variant="secondary", value="Off", elem_classes=["off"])
+    else:
+        return "Running", gr.Button(variant="primary", value="Running", elem_classes=["Running"])
+        
+
+with gr.Blocks(fill_width=True, head=js, css=css) as demo:
     with gr.Tab(str(num_models) + ' Models'):
         with gr.Column(scale=2):
             with gr.Group():
@@ -134,7 +152,11 @@ with gr.Blocks(fill_width=True, head=js) as demo:
                         seed_rand = gr.Button("Randomize Seed 🎲", size="sm", variant="secondary")
                         seed_rand.click(randomize_seed, None, [seed], queue=False)
             with gr.Row():
-                gen_button = gr.Button(f'Generate up to {int(num_models)} images', variant='primary', scale=3)
+                # gen_button = gr.Button(f'Generate up to {int(num_models)} images', variant='primary', scale=3)
+                butState = gr.State(value="Off")
+                gen_button = gr.Button(value="Off", variant="secondary", elem_classes=["off"], scale=3)
+                gen_button.click(fn=toggle_button, inputs=butState, outputs=[butState, button])
+                
                 random_button = gr.Button(f'Randomize Models', variant='secondary', scale=1)
 
         with gr.Column(scale=1):