Update src/streamlit_app.py

src/streamlit_app.py

@@ -248,48 +248,46 @@ elif page:
         for genre in sublist
     ))
 
-    selected_genres = st.multiselect("Select genres:", options=all_genres, placeholder="Choose a genre")
+    selected_genres = st.multiselect(
+        "Select genres:",
+        options=all_genres,
+        default=st.session_state.get("selected_genres", []),
+        placeholder="Choose a genre"
+    )
+    st.session_state["selected_genres"] = selected_genres
 
     if "rating_mode_state" not in st.session_state:
         st.session_state.rating_mode_state = {
             "movie_pool": pd.DataFrame(),
-            "index": 0,
-            "genre_filter": [],
-            "active": False
+            "index": 0
         }
 
-    if st.button("Start Rating Session"):
-        if not selected_genres:
-            st.warning("Please select at least one genre.")
-            st.stop()
-
-        genre_filter = movie_df["genres"].apply(
-            lambda g: any(genre in g.split("|") for genre in selected_genres)
-        )
-        filtered_df = movie_df[genre_filter].copy()
+    mode = st.session_state.rating_mode_state
 
-        if filtered_df.empty:
-            st.warning("No movies found for selected genres.")
+    if mode["movie_pool"].empty or mode.get("genre_filter") != selected_genres:
+        if selected_genres:
+            genre_filter = movie_df["genres"].apply(
+                lambda g: any(genre in g.split("|") for genre in selected_genres)
+            )
+            filtered_df = movie_df[genre_filter].copy()
+            if not filtered_df.empty:
+                mode["movie_pool"] = filtered_df.sample(frac=1, random_state=42).reset_index(drop=True)
+                mode["index"] = 0
+                mode["genre_filter"] = selected_genres
+            else:
+                st.warning("No movies found for selected genres.")
+                st.stop()
+        else:
+            st.info("Please select at least one genre.")
             st.stop()
 
-        st.session_state.rating_mode_state["movie_pool"] = filtered_df.sample(frac=1, random_state=42).reset_index(drop=True)
-        st.session_state.rating_mode_state["index"] = 0
-        st.session_state.rating_mode_state["genre_filter"] = selected_genres
-        st.session_state.rating_mode_state["active"] = True
-        st.rerun()
-
-    mode = st.session_state.rating_mode_state
-    if not mode["active"]:
-        st.info("Please select genres and start a session.")
-        st.stop()
-
     movie_pool = mode["movie_pool"]
     index = mode["index"]
 
-    if movie_pool.empty or index >= len(movie_pool):
-        st.success("✅ You've rated all movies in this session.")
-        if st.button("🔁 Start New Session"):
-            mode["active"] = False
+    if index >= len(movie_pool):
+        st.success("You've gone through all movies in this selection.")
+        if st.button("Restart"):
+            mode["index"] = 0
             st.rerun()
         st.stop()
 
@@ -299,11 +297,11 @@ elif page:
     col1, col2 = st.columns([1, 2])
     with col1:
         if poster_url and "placeholder.com" not in poster_url:
-            st.image(poster_url, width=300)
+            st.image(poster_url, width=200)  # Einheitlich kleine Breite
         else:
             st.markdown("""
-            <div style='width:300px;border:2px dashed gray;height:450px;display:flex;align-items:center;justify-content:center;color:gray;'>
-                No picture available
+            <div style='width:200px;height:300px;border:2px dashed gray;display:flex;align-items:center;justify-content:center;color:gray;font-size:12px;'>
+                No<br>Image
             </div>
             """, unsafe_allow_html=True)
 
@@ -314,13 +312,10 @@ elif page:
         if tmdb_link:
             st.markdown(f"<a href='{tmdb_link}' target='_blank'>View on TMDb</a>", unsafe_allow_html=True)
 
-        st.markdown(f"### Movie {index + 1} of {len(movie_pool)}")
-
-        # Bestehende Bewertung laden
         existing_rating = next((r["rating"] for r in all_ratings_data if r["movie_id"] == movie["movieId"]), None)
         default_rating = existing_rating if existing_rating else 1
 
-        rating_key = f"rating_{movie['movieId']}_{index}"
+        rating_key = f"rating_{movie['movieId']}"
         rating = st.radio(
             "Rate this movie:",
             options=[1, 2, 3, 4, 5],
@@ -331,7 +326,7 @@ elif page:
 
         col_submit, col_skip = st.columns([1, 1])
         with col_submit:
-            if st.button("Submit Rating", key=f"submit_{movie['movieId']}_{index}") and rating:
+            if st.button("Submit Rating", key=f"submit_{movie['movieId']}"):
                 save_rating_to_json({
                     "movie_id": int(movie["movieId"]),
                     "rating": rating,
@@ -339,17 +334,11 @@ elif page:
                 })
                 mode["index"] += 1
                 st.rerun()
-                st.stop()
 
         with col_skip:
-            if st.button("Didn't Watch", key=f"skip_{movie['movieId']}_{index}"):
+            if st.button("Didn't Watch", key=f"skip_{movie['movieId']}"):
                 mode["index"] += 1
                 st.rerun()
-                st.stop()
-
-
-
-
 
 elif search_query:
     st.title(f"Search Results for '{search_query}'")