Update src/streamlit_app.py

src/streamlit_app.py

@@ -11,76 +11,110 @@ MOVIES_PATH = os.path.join(BASE_DIR, "movies.csv")
 @st.cache_data
 def load_movies():
     df = pd.read_csv(MOVIES_PATH)
-    return df["title"].dropna().unique().tolist()
+    return df.dropna(subset=["title"])
+
+movie_df = load_movies()
+movie_titles = movie_df["title"].unique().tolist()
 
 # Initialize session state
-if "movies" not in st.session_state:
-    st.session_state.movies = []
-    st.session_state.queue = []
-    st.session_state.index = 0
+if "rated" not in st.session_state:
     st.session_state.rated = []
-    st.session_state.skipped = 0
-    st.session_state.max_questions = 5
-    st.session_state.finished = False
-    st.session_state.show_slider = True
+if "quiz_history" not in st.session_state:
+    st.session_state.quiz_history = []
 
+# Navigation bar
 st.markdown("""
     <style>
-        html, body, [class*="css"] {
-            background-color: #0d1b2a !important;
-            color: white !important;
+        .nav-bar {
+            display: flex;
+            justify-content: space-between;
+            align-items: center;
+            background-color: #1b263b;
+            padding: 1rem;
+        }
+        .nav-item {
+            color: white;
+            font-size: 20px;
+            text-decoration: none;
+            margin: 0 1rem;
+            cursor: pointer;
+        }
+        .search-bar {
+            width: 50%;
+            padding: 0.5rem;
+            font-size: 16px;
         }
     </style>
+    <div class="nav-bar">
+        <span class="nav-item" onclick="window.location.href='/'">🏠 Home</span>
+        <input type="text" id="search" class="search-bar" placeholder="Search movies..." onkeyup="filterMovies()">
+        <span class="nav-item" onclick="window.location.href='/?recommend=true'">🎯 Quiz</span>
+    </div>
+    <script>
+        function filterMovies() {
+            const input = document.getElementById('search').value.toLowerCase();
+            const elements = document.querySelectorAll('.movie-option');
+            elements.forEach(el => {
+                el.style.display = el.textContent.toLowerCase().includes(input) ? 'block' : 'none';
+            });
+        }
+    </script>
 """, unsafe_allow_html=True)
 
-st.title("🎬 Movie Recommender")
-st.write("Rate a few movies to get personalized recommendations")
+page = st.query_params.get("recommend")
 
-if not st.session_state.movies:
-    movie_titles = load_movies()
-    random.shuffle(movie_titles)
-    st.session_state.movies = movie_titles
+if page:
+    # Quiz Flow
+    if "queue" not in st.session_state:
+        random.shuffle(movie_titles)
+        st.session_state.queue = movie_titles.copy()
+        st.session_state.index = 0
+        st.session_state.rated = []
 
-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.session_state.show_slider = False
+    if st.session_state.index >= len(st.session_state.queue):
+        st.subheader("🍿 Top 10 Recommendations")
+        remaining = [m for m in movie_titles 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):
+            st.markdown(f"**#{i}. {movie}** — Match score: `{round(random.uniform(75, 100), 1)}%`")
+        st.session_state.quiz_history.append(st.session_state.rated.copy())
+        if st.button("🔄 Start Over"):
+            del st.session_state.queue
             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}** — Match score: `{score}%`")
-
-    if st.button("🔄 Start Over"):
-        for key in ["movies", "queue", "index", "rated", "skipped", "max_questions", "finished", "show_slider"]:
-            st.session_state.pop(key, None)
-        st.rerun()
-else:
-    movie = st.session_state.queue[st.session_state.index]
-
-    st.markdown(f"### 🎞️ Movie #{len(st.session_state.rated) + 1} 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"):
+    else:
+        movie = st.session_state.queue[st.session_state.index]
+        st.markdown(f"### 🎞️ Movie #{len(st.session_state.rated) + 1}")
+        st.markdown(f"## **{movie}**")
+        rating = st.radio("Rate this movie", [1, 2, 3, 4, 5], horizontal=True)
+        if st.button("Submit Rating"):
+            st.session_state.rated.append({"movie": movie, "rating": rating})
             st.session_state.index += 1
+            st.rerun()
+else:
+    # Home Page
+    st.title("🏠 Welcome to MovieMatch")
+    st.subheader("Your past quiz results")
+    for i, result in enumerate(st.session_state.quiz_history):
+        st.markdown(f"**Quiz {i+1} Summary:**")
+        for r in result:
+            st.markdown(f"- {r['movie']}: ⭐ {r['rating']}")
+    st.subheader("Overall summary")
+    all_ratings = [r for result in st.session_state.quiz_history for r in result]
+    summary_df = pd.DataFrame(all_ratings)
+    if not summary_df.empty:
+        avg_rating = summary_df.groupby("movie")["rating"].mean().sort_values(ascending=False)
+        st.write(avg_rating.reset_index().rename(columns={"rating": "Average Rating"}))
+    else:
+        st.info("No ratings yet.")
 
-    if len(st.session_state.rated) >= st.session_state.max_questions:
-        st.session_state.finished = True
-        st.rerun()
+    st.subheader("Search & Rate")
+    query = st.text_input("Search for a movie")
+    if query:
+        matches = [m for m in movie_titles if query.lower() in m.lower()]
+        for match in matches:
+            with st.expander(match):
+                st.write("Short description placeholder")
+                rate = st.radio(f"Rate {match}", [1, 2, 3, 4, 5], key=match)
+                if st.button(f"Submit {match}", key=f"submit_{match}"):
+                    st.session_state.rated.append({"movie": match, "rating": rate})
+                    st.success(f"Thanks for rating {match} ⭐{rate}")