Update app.py

app.py

@@ -13,14 +13,19 @@ GOOGLE_FONTS_API_KEY = os.environ.get("GOOGLE_FONTS_API_KEY")
 
 st.set_page_config(page_title="Moodboard AI", layout="wide")
 
-# -- Dark Mode Toggle --
+# -- Init Session State --
 if 'dark_mode' not in st.session_state:
     st.session_state.dark_mode = False
+if 'generate' not in st.session_state:
+    st.session_state.generate = False
+if 'remix' not in st.session_state:
+    st.session_state.remix = random.randint(1, 10000)
 
-dark_mode_toggle = st.toggle("🌙 Toggle Dark Mode", value=st.session_state.dark_mode)
+# -- Dark Mode Toggle (no forced rerun) --
+dark_mode_toggle = st.toggle("🌙 Toggle Dark Mode", value=st.session_state.dark_mode, key="dark_toggle")
 if dark_mode_toggle != st.session_state.dark_mode:
     st.session_state.dark_mode = dark_mode_toggle
-    st.rerun()
+    st.experimental_rerun()
 
 # -- Apply Dark Mode Styling --
 if st.session_state.dark_mode:
@@ -56,7 +61,7 @@ else:
 # -- Title --
 st.title("🎨 Moodboard AI – Smart Visual Inspiration Generator")
 
-# -- Display API Status --
+# -- API Status Display --
 if UNSPLASH_ACCESS_KEY:
     st.success("✅ Unsplash API key loaded successfully.")
 else:
@@ -70,14 +75,15 @@ else:
 # -- Theme Input --
 theme = st.text_input("Enter a theme or keyword (e.g. 'vintage tech', 'coastal minimalism'):", "futuristic tech").lower()
 
-# -- Remix Button --
-if st.button("🔁 Remix Moodboard"):
-    st.session_state.remix = random.randint(1, 10000)
-    st.session_state.generate = True
-
-# -- Generate Moodboard Button --
-if st.button("✨ Generate Moodboard"):
-    st.session_state.generate = True
+# -- Control Buttons --
+col1, col2 = st.columns(2)
+with col1:
+    if st.button("🔁 Remix Moodboard"):
+        st.session_state.remix = random.randint(1, 10000)
+        st.session_state.generate = True
+with col2:
+    if st.button("✨ Generate Moodboard"):
+        st.session_state.generate = True
 
 # -- Fetch Images from Unsplash --
 def fetch_images_from_unsplash(query, count=5):
@@ -131,7 +137,8 @@ def fetch_google_fonts():
 # -- Show Moodboard --
 if st.session_state.get("generate"):
     with st.spinner("Generating your moodboard..."):
-        image_urls = fetch_images_from_unsplash(theme)
+        query = f"{theme} {st.session_state.remix}"
+        image_urls = fetch_images_from_unsplash(query)
         if not image_urls:
             st.warning("No images found. Try another keyword or check your API key.")
         else:
@@ -160,4 +167,4 @@ if st.session_state.get("generate"):
                 )
 
             st.success("✅ Moodboard ready! Right-click or screenshot to save.")
-    st.session_state.generate = False  # Reset flag
+    st.session_state.generate = False  # Reset