Update app.py

app.py

@@ -113,13 +113,11 @@ def preset_tf_for_ui(selected_gpu: str, dtype: str):
 # Theme setter (returns HTML snippet to run client-side JS)
 # ------------------------
 def set_theme(theme_name: str):
-    # map button label -> class name used in CSS
     cls = {
         "Blue": "theme-blue",
         "Green": "theme-green",
         "Purple": "theme-purple",
     }.get(theme_name, "theme-blue")
-    # return script that sets the root class
     return f"<script>document.documentElement.className='{cls}';</script>"
 
 # ------------------------
@@ -163,15 +161,15 @@ with gr.Blocks() as demo:
     gpu_dropdown.change(_update_tf, inputs=[gpu_dropdown, dtype_dropdown], outputs=[tf_override])
     dtype_dropdown.change(_update_tf, inputs=[gpu_dropdown, dtype_dropdown], outputs=[tf_override])
 
-    # button clicks
+    # Run button computes estimate
     run_btn.click(estimate_time,
                   inputs=[params, tokens, gpu_dropdown, dtype_dropdown, tf_override, utilization],
                   outputs=[result])
 
-    # theme buttons now call the Python set_theme and return HTML that runs client-side
-    btn_blue.click(set_theme, inputs=["Blue"], outputs=[theme_script])
-    btn_green.click(set_theme, inputs=["Green"], outputs=[theme_script])
-    btn_purple.click(set_theme, inputs=["Purple"], outputs=[theme_script])
+    # THE FIXED THEME BUTTONS: use zero-arg lambdas that return the HTML script value
+    btn_blue.click(lambda: set_theme("Blue"), outputs=[theme_script])
+    btn_green.click(lambda: set_theme("Green"), outputs=[theme_script])
+    btn_purple.click(lambda: set_theme("Purple"), outputs=[theme_script])
 
     gr.HTML("<div class='small-muted'>Tip: GPU presets are TFLOPs per dtype. You can edit the TFLOPs number to override. Utilization reduces theoretical peak to realistic throughput.</div>")