- Format documents

- Reload polls list
- Fix poll exception on deleted poll
- Clear user votes after refresh

home.py

@@ -17,6 +17,7 @@ def is_valid_new_poll(poll_name):
             return False
     return True
 
+
 def is_valid_join_poll(poll_name):
     poll = utils.find_poll(poll_name)
     if poll:
@@ -26,6 +27,8 @@ def is_valid_join_poll(poll_name):
         else:
             return True
     return False
+
+
 def is_valid_admin_password(poll_name, admin_password):
     poll = utils.find_poll(poll_name)
     if poll:
@@ -33,6 +36,7 @@ def is_valid_admin_password(poll_name, admin_password):
             return True
     return False
 
+
 def join_poll(poll_name, admin_password=""):
     if not is_valid_join_poll(poll_name):
         st.error(f"Invalid poll name {poll_name}")
@@ -46,6 +50,7 @@ def join_poll(poll_name, admin_password=""):
     if "poll" not in st.query_params:
         st.query_params.from_dict({"poll": poll_name})
 
+
 def new_poll(poll_name, admin_password):
     if not is_valid_new_poll(poll_name):
         return
@@ -56,7 +61,7 @@ def new_poll(poll_name, admin_password):
         "name": poll_name,
         "admin_password": admin_password,
         "expiry": datetime.datetime.now() + datetime.timedelta(hours=3),
-        "votes" : {},
+        "votes": {},
     }
 
     utils.load_polls().append(poll)
@@ -64,8 +69,10 @@ def new_poll(poll_name, admin_password):
 
 
 def show_polls():
+    st.write("---")
     with st.container(border=True):
         st.header("Poll list:")
+        st.button("Reload Polls", use_container_width=True)
 
         polls = utils.load_polls()
         if len(polls) == 0:
@@ -103,14 +110,16 @@ def show_join_poll(column):
             poll_name = st.text_input("Join Poll")
             poll_admin_password = ""
             if st.checkbox("Login as Admin", value=False):
-                poll_admin_password = st.text_input("Admin Password", type="password", key="join_admin_password")
+                poll_admin_password = st.text_input(
+                    "Admin Password", type="password", key="join_admin_password"
+                )
             st.button(
                 "Join Poll",
                 type="primary",
                 key="btn_join",
                 use_container_width=True,
                 on_click=join_poll,
-                args=(poll_name,poll_admin_password),
+                args=(poll_name, poll_admin_password),
             )
 
 
@@ -125,7 +134,7 @@ def show_new_poll(column):
                 key="btn_new",
                 use_container_width=True,
                 on_click=new_poll,
-                args=(poll_name,poll_admin_password),
+                args=(poll_name, poll_admin_password),
             )
 
 

poll.py

@@ -2,6 +2,8 @@ import utils
 import streamlit as st
 import plotly.express as px
 import pandas as pd
+import time
+
 
 def exit_poll():
     set_votes([])
@@ -10,32 +12,64 @@ def exit_poll():
     if "selected_poll_admin_password" in st.session_state:
         del st.session_state["selected_poll_admin_password"]
 
+
 def set_votes(votes=[]):
     poll = utils.find_poll(st.session_state["selected_poll_name"])
+    if poll is None:
+        on_poll_not_exists()
+        return
     uuid = st.session_state["UUID"]
-    poll["votes"][uuid] = votes
+    if votes == []:
+        if uuid in poll["votes"]:
+            del poll["votes"][uuid]
+    else:
+        poll["votes"][uuid] = votes
+
 
 def vote():
     votes = st.session_state.get("vote_selected", [])
     set_votes(votes)
 
+
 def load_voting():
+    poll = utils.find_poll(st.session_state["selected_poll_name"])
+    if poll is None:
+        on_poll_not_exists()
+        return
+    current_votes = poll["votes"].get(st.session_state["UUID"], [])
     st.multiselect(
-    "How do you feel?",
-    utils.get_options(),
-    [],
-    on_change=vote,
-    key="vote_selected")
+        "How do you feel?",
+        utils.get_options(),
+        current_votes,
+        on_change=vote,
+        key="vote_selected",
+    )
+
 
 def is_admin():
     password = st.session_state.get("selected_poll_admin_password", "")
     poll = utils.find_poll(st.session_state["selected_poll_name"])
+    if poll is None:
+        return False
     return poll["admin_password"] == password
 
+
+def on_poll_not_exists():
+    st.error(f"Poll {st.session_state['selected_poll_name']} does not exist")
+    st.query_params.pop("poll")
+    del st.session_state["selected_poll_name"]
+    if "selected_poll_admin_password" in st.session_state:
+        del st.session_state["selected_poll_admin_password"]
+
+
 def clear_votes_section():
     poll = utils.find_poll(st.session_state["selected_poll_name"])
+    if poll is None:
+        on_poll_not_exists()
+        return
     poll["votes"] = {}
 
+
 def delete_poll():
     utils.load_polls().remove(utils.find_poll(st.session_state["selected_poll_name"]))
     st.query_params.pop("poll")
@@ -45,53 +79,73 @@ def delete_poll():
 def load_votes():
     selected_poll_name = st.session_state["selected_poll_name"]
     poll = utils.find_poll(selected_poll_name)
+    if poll is None:
+        on_poll_not_exists()
+        return
     votes = {x: 0 for x in utils.get_options()}
     votes_count = 0
     users_count = 0
-    for _,user_votes in poll["votes"].items():
-        users_count +=1
+    for _, user_votes in poll["votes"].items():
+        users_count += 1
         for vote in user_votes:
-            votes_count +=1
+            votes_count += 1
             if vote in votes:
                 votes[vote] += 1
             else:
                 votes[vote] = 1
 
-
-
-    df = pd.DataFrame(list(votes.items()), columns=['Status', 'Count'])
+    df = pd.DataFrame(list(votes.items()), columns=["Status", "Count"])
     if df["Count"].sum() == 0:
         st.warning("No votes yet!")
     else:
         st.success(f"Users:{users_count} | Votes:{votes_count}")
-        fig = px.pie(df, values='Count', names='Status', title="Audience Feedback",)
+        fig = px.pie(
+            df,
+            values="Count",
+            names="Status",
+            title="Audience Feedback",
+        )
         st.plotly_chart(fig)
         st.write(df)
     st.button("Clear Votes", on_click=clear_votes_section, use_container_width=True)
     st.button("Delete Poll", on_click=delete_poll, use_container_width=True)
 
+
 def show_qrcode():
     selected_poll_name = st.session_state["selected_poll_name"]
     columns = st.columns(5)
     with columns[2]:
         st.image(utils.generate_qr_code_image_link(selected_poll_name))
 
+
+def dummy_refresh():
+    selected_poll_name = st.session_state["selected_poll_name"]
+    poll = utils.find_poll(selected_poll_name)
+    if poll is None:
+        on_poll_not_exists()
+        return
+
+
 def load():
+    dummy_refresh()
     selected_poll_name = st.session_state["selected_poll_name"]
     st.write("")
     columns = st.columns(3)
     with columns[0]:
         st.button("Exit", on_click=exit_poll, use_container_width=True)
     with columns[1]:
-        st.button("Refresh", use_container_width=True)
+        st.button("Refresh", use_container_width=True, on_click=dummy_refresh)
     with columns[2]:
-        st.link_button("Link To Poll", "?poll="+selected_poll_name, type="primary", use_container_width=True)
+        st.link_button(
+            "Link To Poll",
+            "?poll=" + selected_poll_name,
+            type="primary",
+            use_container_width=True,
+        )
 
     st.title(selected_poll_name)
     if not is_admin():
         load_voting()
     else:
         load_votes()
-
-    
-    show_qrcode()
+    show_qrcode()