Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,14 +5,11 @@ import os
 import json
 from datetime import datetime
 
-# Page config to use full width
 st.set_page_config(layout="wide")
 
-# Constants
 MOVIES_PATH = os.path.join(os.path.dirname(__file__), "movies.csv")
 RATINGS_JSON_PATH = "/tmp/ratings.json"
 
-# Load or create ratings JSON
 def load_ratings():
     if os.path.exists(RATINGS_JSON_PATH):
         with open(RATINGS_JSON_PATH, "r") as f:
@@ -26,11 +23,10 @@ def save_rating_to_json(entry):
     with open(RATINGS_JSON_PATH, "w") as f:
         json.dump(all_ratings, f, indent=2, default=str)
 
-# Load movie data
 @st.cache_data
 def load_movies():
     df = pd.read_csv(MOVIES_PATH)
-    df["year"] = df["title"].str.extract(r'\((\d{4})\)').fillna("Unknown")
+    df["year"] = df["title"].str.extract(r'\((\d{4})\)').fillna("0").astype(int)
     df["clean_title"] = df["title"].str.replace(r'\(\d{4}\)', '', regex=True).str.strip()
     df["genres"] = df["genres"].fillna("Unknown")
     return df
@@ -117,17 +113,34 @@ def render_star_rating(rating):
 all_ratings_data = load_ratings()
 
 if movie_id:
-    # Detail Page
-    ...  # Unverändert
+    ...  # Detail Page unverändert
 
 elif page:
-    # 🎲 Random Movie Rating Flow
     st.title("🎲 Rate Random Movies")
-    if "queue" not in st.session_state:
-        random.shuffle(movie_titles)
-        st.session_state.queue = movie_titles.copy()
+
+    # Filter UI
+    genres_all = sorted(set("|".join(movie_df["genres"]).split("|")))
+    selected_genres = st.multiselect("Filter by genres", genres_all)
+
+    min_year = movie_df["year"].replace(0, pd.NA).dropna().min()
+    max_year = movie_df["year"].max()
+    selected_year_range = st.slider("Filter by release year", min_year, max_year, (min_year, max_year))
+
+    # Filter movie list
+    filtered_df = movie_df.copy()
+    if selected_genres:
+        genre_filter = filtered_df["genres"].apply(lambda g: any(gen in g for gen in selected_genres))
+        filtered_df = filtered_df[genre_filter]
+    filtered_df = filtered_df[(filtered_df["year"] >= selected_year_range[0]) & (filtered_df["year"] <= selected_year_range[1])]
+
+    filtered_titles = filtered_df["title"].tolist()
+
+    if "queue" not in st.session_state or st.session_state.get("filter_hash") != hash((tuple(selected_genres), selected_year_range)):
+        random.shuffle(filtered_titles)
+        st.session_state.queue = filtered_titles
         st.session_state.index = 0
         st.session_state.rated = []
+        st.session_state.filter_hash = hash((tuple(selected_genres), selected_year_range))
 
     if st.session_state.index >= len(st.session_state.queue):
         st.subheader("🎉 You've rated enough movies!")