Update src/streamlit_app.py

src/streamlit_app.py

@@ -22,15 +22,13 @@ if "movies" not in st.session_state:
     st.session_state.skipped = 0
     st.session_state.max_questions = 5
     st.session_state.finished = False
+    st.session_state.show_slider = True
 
 st.markdown("""
     <style>
-        body {
-            background-color: #0d1b2a;
-            color: white;
-        }
-        .stSlider > div {
-            background-color: #1b263b;
+        html, body, [class*="css"] {
+            background-color: #0d1b2a !important;
+            color: white !important;
         }
     </style>
 """, unsafe_allow_html=True)
@@ -43,42 +41,42 @@ if not st.session_state.movies:
     random.shuffle(movie_titles)
     st.session_state.movies = movie_titles
 
-if not st.session_state.queue:
+if st.session_state.show_slider:
     with st.container():
         st.subheader("How many movies do you want to rate?")
         st.session_state.max_questions = st.slider("Select number of movies", 1, 15, 5)
         if st.button("Start Rating"):
             st.session_state.queue = st.session_state.movies.copy()
-            st.experimental_rerun()
+            st.session_state.show_slider = False
+            st.rerun()
+elif st.session_state.finished or st.session_state.index >= len(st.session_state.queue):
+    st.subheader("🍿 Top 10 Recommendations")
+    st.write("Based on your preferences, we suggest:")
+    remaining = [m for m in st.session_state.movies if m not in [r["movie"] for r in st.session_state.rated]]
+    recommendations = random.sample(remaining, min(10, len(remaining)))
+    for i, movie in enumerate(recommendations, 1):
+        score = round(random.uniform(75, 100), 1)
+        st.markdown(f"**#{i}. {movie}** — Likelihood: `{score}%`")
 else:
-    if st.session_state.finished or st.session_state.index >= len(st.session_state.queue):
-        st.subheader("🍿 Top 10 Recommendations")
-        st.write("Based on your preferences, we suggest:")
-        remaining = [m for m in st.session_state.movies if m not in [r["movie"] for r in st.session_state.rated]]
-        recommendations = random.sample(remaining, min(10, len(remaining)))
-        for i, movie in enumerate(recommendations, 1):
-            score = round(random.uniform(75, 100), 1)
-            st.markdown(f"**#{i}. {movie}** — Likelihood: `{score}%`")
-    else:
-        movie = st.session_state.queue[st.session_state.index]
-        card_number = len(st.session_state.rated) + 1
+    movie = st.session_state.queue[st.session_state.index]
+    card_number = len(st.session_state.rated) + 1
 
-        st.markdown(f"### 🎞️ Movie #{card_number} of {st.session_state.max_questions}")
-        st.markdown(f"## **{movie}**")
+    st.markdown(f"### 🎞️ Movie #{card_number} of {st.session_state.max_questions}")
+    st.markdown(f"## **{movie}**")
 
-        col1, col2, col3 = st.columns([1, 1, 1])
-        with col1:
-            if st.button("👍 Like"):
-                st.session_state.rated.append({"movie": movie, "rating": "like"})
-                st.session_state.index += 1
-        with col2:
-            if st.button("👎 Dislike"):
-                st.session_state.rated.append({"movie": movie, "rating": "dislike"})
-                st.session_state.index += 1
-        with col3:
-            if st.button("⏭ Didn't Watch"):
-                st.session_state.index += 1
+    col1, col2, col3 = st.columns([1, 1, 1])
+    with col1:
+        if st.button("👍 Like"):
+            st.session_state.rated.append({"movie": movie, "rating": "like"})
+            st.session_state.index += 1
+    with col2:
+        if st.button("👎 Dislike"):
+            st.session_state.rated.append({"movie": movie, "rating": "dislike"})
+            st.session_state.index += 1
+    with col3:
+        if st.button("⏭ Didn't Watch"):
+            st.session_state.index += 1
 
-        if len(st.session_state.rated) >= st.session_state.max_questions:
-            st.session_state.finished = True
-            st.experimental_rerun()
+    if len(st.session_state.rated) >= st.session_state.max_questions:
+        st.session_state.finished = True
+        st.rerun()