Update src/streamlit_app.py

src/streamlit_app.py

@@ -250,20 +250,21 @@ elif page:
     ))
     selected_genres = st.multiselect("Select genres:", options=all_genres, placeholder="Choose a genre")
 
-    # Initialer Zustand
+    # Initialzustand setzen
     if "rating_mode_state" not in st.session_state:
         st.session_state.rating_mode_state = {
             "movie_pool": [],
             "index": 0,
-            "genre_filter": []
+            "genre_filter": [],
+            "initialized": False
         }
 
-    # Filter neu anwenden, wenn sich Genres geändert haben
-    if st.session_state.rating_mode_state["genre_filter"] != selected_genres:
+    # Neuinitialisierung per Button starten
+    if st.button("Start Rating Session") or not st.session_state.rating_mode_state["initialized"]:
         genre_filter = movie_df["genres"].apply(
             lambda g: any(genre in g.split("|") for genre in selected_genres)
         ) if selected_genres else pd.Series([True] * len(movie_df))
-        
+
         filtered_df = movie_df[genre_filter].copy()
 
         if filtered_df.empty:
@@ -271,17 +272,22 @@ elif page:
             st.stop()
 
         st.session_state.rating_mode_state = {
-            "movie_pool": filtered_df.sample(frac=1).reset_index(drop=True),
+            "movie_pool": filtered_df.sample(frac=1, random_state=42).reset_index(drop=True),
             "index": 0,
-            "genre_filter": selected_genres
+            "genre_filter": selected_genres,
+            "initialized": True
         }
 
     # Hole aktuellen Film
     movie_pool = st.session_state.rating_mode_state["movie_pool"]
     index = st.session_state.rating_mode_state["index"]
 
+    if not movie_pool:
+        st.info("Please start a rating session.")
+        st.stop()
+
     if index >= len(movie_pool):
-        st.info("No more movies to rate.")
+        st.success("✅ You've rated all movies in this session.")
         st.stop()
 
     movie = movie_pool.iloc[index]
@@ -310,11 +316,10 @@ elif page:
         st.markdown(f"### Movie {index + 1} of {len(movie_pool)}")
         rating_key = f"rating_{movie['movieId']}_{index}"
         rating = st.radio("Rate this movie:", [1, 2, 3, 4, 5], horizontal=True, key=rating_key)
-        
+
         col_submit, col_skip = st.columns([1, 1])
         with col_submit:
             if st.button("Submit Rating", key=f"submit_{movie['movieId']}_{index}") and rating:
-
                 save_rating_to_json({
                     "movie_id": int(movie["movieId"]),
                     "rating": rating,
@@ -324,7 +329,7 @@ elif page:
                 st.rerun()
 
         with col_skip:
-            if st.button("Didn't Watch"):
+            if st.button("Didn't Watch", key=f"skip_{movie['movieId']}_{index}"):
                 st.session_state.rating_mode_state["index"] += 1
                 st.rerun()