Update app.py

app.py

@@ -6,9 +6,6 @@ import random
 themes = {
     "Theme 1": ["Node 1.1", "Node 1.2", "Node 1.3", "Node 1.4", "Node 1.5", "Node 1.6", "Node 1.7", "Node 1.8", "Node 1.9", "Node 1.10"],
     "Theme 2": ["Node 2.1", "Node 2.2", "Node 2.3", "Node 2.4", "Node 2.5", "Node 2.6", "Node 2.7", "Node 2.8", "Node 2.9", "Node 2.10"],
-    "Theme 3": ["Node 3.1", "Node 3.2", "Node 3.3", "Node 3.4", "Node 3.5", "Node 3.6", "Node 3.7", "Node 3.8", "Node 3.9", "Node 3.10"],
-    "Theme 4": ["Node 4.1", "Node 4.2", "Node 4.3", "Node 4.4", "Node 4.5", "Node 4.6", "Node 4.7", "Node 4.8", "Node 4.9", "Node 4.10"],
-    "Theme 5": ["Node 5.1", "Node 5.2", "Node 5.3", "Node 5.4", "Node 5.5", "Node 5.6", "Node 5.7", "Node 5.8", "Node 5.9", "Node 5.10"],
 }
 
 # Define emojis for top 1 through 10
@@ -31,26 +28,24 @@ def random_timer(countdown):
 st.title("Spinning Model Visualization")
 st.write("Select themes to display the spinning model and adjust the number of seconds before the next refresh.")
 
-# Add input widgets for the five themes and refresh interval
+# Add input widgets for the two themes and refresh interval
 theme_1 = st.text_input("Theme 1:", "Theme 1")
 theme_2 = st.text_input("Theme 2:", "Theme 2")
-theme_3 = st.text_input("Theme 3:", "Theme 3")
-theme_4 = st.text_input("Theme 4:", "Theme 4")
-theme_5 = st.text_input("Theme 5:", "Theme 5")
-refresh_interval = st.slider("Refresh interval (in seconds):",1, 10, 5)
+refresh_interval = st.slider("Refresh interval (in seconds):", 1, 10, 5)
 
-selected_themes = [themes.get(theme, []) for theme in [theme_1, theme_2, theme_3, theme_4, theme_5]]
+# Get the selected themes and their nodes
+selected_theme_1 = themes.get(theme_1, [])
+selected_theme_2 = themes.get(theme_2, [])
 
-i = 0
 while True:
-    # Get the selected theme and shuffle the nodes
-    selected_theme = selected_themes[i]
+    # Randomly select a theme to display and shuffle the nodes
+    if random.randint(0, 1):
+        selected_theme = selected_theme_1
+        theme = theme_1
+    else:
+        selected_theme = selected_theme_2
+        theme = theme_2
     random.shuffle(selected_theme)
-
-spinning_model(f"Theme {i+1}", selected_theme, emojis)
-selected_theme = random_timer(refresh_interval)
-
-# Increment the index and loop back to the first theme if all themes have been shown
-i = (i + 1) % 5
-if i == 0:
-    st.write("---")
+    
+    spinning_model(theme, selected_theme, emojis)
+    selected_theme = random_timer(refresh_interval)