Update app.py

app.py

@@ -32,27 +32,16 @@ def predict_salary(age, education_num, sex, capital_gain, capital_loss, hours_pe
 
     return label, confidence, fig
 
-toggle_theme = gr.State(value="dark")
-
-with gr.Blocks(css="""
+# Custom CSS for styling
+dark_css = """
     body { background-color: #0a0a23; color: white; }
     .gr-box { background-color: #1a1a3d !important; border-radius: 12px; padding: 20px; }
     h1, h3, p { color: white; text-align: center; font-family: 'Segoe UI', sans-serif; }
     .gr-button { background-color: #1db954 !important; color: white !important; border-radius: 10px; font-size: 1.1em; }
     .gr-radio .gr-form { display: flex; justify-content: center !important; }
 """
-) as demo:
-
-    def switch_theme(theme_choice):
-        if theme_choice == "dark":
-            return gr.themes.Base().set(body_background="#0a0a23", body_text_color="white"), "dark"
-        else:
-            return gr.themes.Default(), "light"
 
-    with gr.Row():
-        theme_toggle = gr.Radio(["dark", "light"], value="dark", label="Theme", interactive=True)
-
-    theme_toggle.change(switch_theme, inputs=theme_toggle, outputs=[demo.theme, toggle_theme])
+with gr.Blocks(css=dark_css) as demo:
 
     gr.Markdown("""
         <div style='max-width: 700px; margin: 0 auto;'>
@@ -99,3 +88,4 @@ with gr.Blocks(css="""
 demo.launch()
 
 
+