Update app.py

app.py

@@ -16,8 +16,11 @@ st.set_page_config(page_title="Moodboard AI", layout="wide")
 # -- Dark Mode Toggle --
 if 'dark_mode' not in st.session_state:
     st.session_state.dark_mode = False
-if st.checkbox("🌙 Toggle Dark Mode"):
-    st.session_state.dark_mode = not st.session_state.dark_mode
+
+dark_mode_toggle = st.toggle("🌙 Toggle Dark Mode", value=st.session_state.dark_mode)
+if dark_mode_toggle != st.session_state.dark_mode:
+    st.session_state.dark_mode = dark_mode_toggle
+    st.rerun()
 
 # -- Apply Dark Mode Styling --
 if st.session_state.dark_mode:
@@ -30,10 +33,23 @@ if st.session_state.dark_mode:
             h1, h2, h3, h4, h5, h6 {
                 color: #ffffff !important;
             }
-            .stTextInput > div > div > input {
+            .stTextInput > div > div > input,
+            .stTextArea > div > textarea {
                 background-color: #222222;
                 color: #ffffff;
             }
+            .stButton>button {
+                background-color: #333333;
+                color: #ffffff;
+            }
+        </style>
+    """, unsafe_allow_html=True)
+else:
+    st.markdown("""
+        <style>
+            h1, h2, h3, h4, h5, h6 {
+                color: #000000 !important;
+            }
         </style>
     """, unsafe_allow_html=True)
 
@@ -139,7 +155,7 @@ if st.session_state.get("generate"):
             fonts = fetch_google_fonts()
             for font in fonts:
                 st.markdown(
-                    f"<p style='font-family:{font};font-size:24px;color:#ffffff'>{font}</p>",
+                    f"<p style='font-family:{font};font-size:24px;color:{'#ffffff' if st.session_state.dark_mode else '#000000'}'>{font}</p>",
                     unsafe_allow_html=True
                 )