lenawilli commited on
Commit
fdc9ef7
ยท
verified ยท
1 Parent(s): a91e9f1

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +6 -2
src/streamlit_app.py CHANGED
@@ -57,11 +57,15 @@ elif st.session_state.finished or st.session_state.index >= len(st.session_state
57
  for i, movie in enumerate(recommendations, 1):
58
  score = round(random.uniform(75, 100), 1)
59
  st.markdown(f"**#{i}. {movie}** โ€” Likelihood: `{score}%`")
 
 
 
 
 
60
  else:
61
  movie = st.session_state.queue[st.session_state.index]
62
- card_number = len(st.session_state.rated) + 1
63
 
64
- st.markdown(f"### ๐ŸŽž๏ธ Movie #{card_number} of {st.session_state.max_questions}")
65
  st.markdown(f"## **{movie}**")
66
 
67
  col1, col2, col3 = st.columns([1, 1, 1])
 
57
  for i, movie in enumerate(recommendations, 1):
58
  score = round(random.uniform(75, 100), 1)
59
  st.markdown(f"**#{i}. {movie}** โ€” Likelihood: `{score}%`")
60
+
61
+ if st.button("๐Ÿ”„ Start Over"):
62
+ for key in ["movies", "queue", "index", "rated", "skipped", "max_questions", "finished", "show_slider"]:
63
+ st.session_state.pop(key, None)
64
+ st.rerun()
65
  else:
66
  movie = st.session_state.queue[st.session_state.index]
 
67
 
68
+ st.markdown(f"### ๐ŸŽž๏ธ Movie #{len(st.session_state.rated) + 1} of {st.session_state.max_questions}")
69
  st.markdown(f"## **{movie}**")
70
 
71
  col1, col2, col3 = st.columns([1, 1, 1])