Update app.py

app.py

@@ -115,26 +115,23 @@ def display_quote(index):
     st.markdown(f"### {number}. {topic}")
     st.markdown(quote)
 
+
+
 # Streamlit app
 def main():
 
     # Session state to hold the value of AutoRepeat button across reruns
     if "auto_repeat" not in st.session_state:
         st.session_state.auto_repeat = "On"
-
+    if "current_index" not in st.session_state:
+        st.session_state.current_index = random.randint(0, len(quotes)-1)
+    
     st.title("Quote Timer")
 
     # AutoRepeat radio button
     st.session_state.auto_repeat = st.radio("🔄 AutoRepeat", ["On", "Off"])
 
-    # Check if we should change the quote (using a hidden button)
-    if st.button("change_quote", key="hidden_button", on_click=_change_quote):
-        pass
-
-    # Display a random quote
-    if "current_index" not in st.session_state:
-        st.session_state.current_index = random.randint(0, len(quotes)-1)
-    
+    # Display the current quote
     display_quote(st.session_state.current_index)
 
     timer_html = """
@@ -144,11 +141,17 @@ def main():
         const timerElement = document.getElementById('timer');
         
         function updateTimer() {
-            if (countdown > 2) { // Change here
+            if (countdown > 1) {
                 timerElement.innerHTML = `Time left: ${countdown} seconds`;
                 countdown -= 1;
             } else {
-                document.querySelector("[data-testid='hidden_button']").click();
+                countdown = 5;
+                timerElement.innerHTML = `Time left: ${countdown} seconds`;
+                fetch('/stream').then(response => response.text()).then(data => {
+                    if (data) {
+                        location.reload();
+                    }
+                });
             }
         }