Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,15 +2,31 @@ import streamlit as st
 import pandas as pd
 import random
 import os
+import json
 from datetime import datetime
 
 # Page config to use full width
 st.set_page_config(layout="wide")
 
-# Load movie data
+# Constants
 BASE_DIR = os.path.dirname(__file__)
 MOVIES_PATH = os.path.join(BASE_DIR, "movies.csv")
+RATINGS_JSON_PATH = os.path.join(BASE_DIR, "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:
+            return json.load(f)
+    return []
+
+def save_rating_to_json(entry):
+    all_ratings = load_ratings()
+    all_ratings.append(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)
@@ -107,9 +123,11 @@ if movie_id:
         st.title(movie_row["clean_title"])
         st.caption(f"Genres: {movie_row['genres']}")
         st.caption(f"Release Year: {movie_row['year']}")
-        rating = st.radio("Rate this movie", [1, 2, 3, 4, 5], horizontal=True)
+        rating = st.radio("Rate this movie", [1, 2, 3, 4, 5], key=f"detail_{movie_id}")
         if st.button("Submit Rating"):
-            st.session_state.rated.append({"movie": movie_row["title"], "rating": rating, "timestamp": datetime.now(), "source": "detail"})
+            entry = {"movie": movie_row["title"], "rating": rating, "timestamp": datetime.now().isoformat(), "source": "detail"}
+            st.session_state.rated.append(entry)
+            save_rating_to_json(entry)
             st.success(f"Thanks for rating '{movie_row['title']}' with ⭐{rating}")
     except:
         st.error("Movie not found.")
@@ -129,6 +147,8 @@ elif page:
         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())
+        for r in st.session_state.rated:
+            save_rating_to_json(r)
         if st.button("🔄 Start Over"):
             del st.session_state.queue
             st.rerun()
@@ -138,7 +158,9 @@ elif page:
         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, "timestamp": datetime.now(), "source": "quiz"})
+            entry = {"movie": movie, "rating": rating, "timestamp": datetime.now().isoformat(), "source": "quiz"}
+            st.session_state.rated.append(entry)
+            save_rating_to_json(entry)
             st.session_state.index += 1
             st.rerun()
 
@@ -170,7 +192,9 @@ elif search_query:
             st.markdown(f"[➡️ Go to Detail Page](?movie_id={row['movieId']})")
             rating = st.radio(f"Rate '{row['title']}'", [1, 2, 3, 4, 5], key=row["title"])
             if st.button(f"Submit rating for {row['title']}", key=f"submit_{row['title']}"):
-                st.session_state.rated.append({"movie": row["title"], "rating": rating, "timestamp": datetime.now(), "source": "search"})
+                entry = {"movie": row["title"], "rating": rating, "timestamp": datetime.now().isoformat(), "source": "search"}
+                st.session_state.rated.append(entry)
+                save_rating_to_json(entry)
                 st.success(f"Thanks for rating '{row['title']}' with ⭐{rating}")
 
 else: