Update app.py

app.py

@@ -96,140 +96,52 @@ import streamlit as st
 import json
 import webbrowser
 
-# Function to load the history of clicks from the text file
-def load_history():
-    try:
-        with open("click_history.txt", "r") as f:
-            return json.load(f)
-    except FileNotFoundError:
-        return {url: 0 for url in urls}
-
-# Function to save the updated history of clicks to the text file
-def save_history(history):
-    with open("click_history.txt", "w") as f:
-        json.dump(history, f)
-
-# Load the history of clicks
-history = load_history()
-
-# Display the buttons for each URL
-for url, name, emoji in zip(urls, url_names, url_emojis):
-    if st.button(f"{emoji} {name}"):
-        # Open the URL in a new browser tab using JavaScript
-        st.write('<script>window.open("'+url+'", "_blank");</script>', unsafe_allow_html=True)
-        # Update the history of clicks
-        history[url] += 1
-        save_history(history)
-    # Display the number of times the URL was opened below its corresponding button
-    st.write(f"Clicked: {history[url]} times")
-
-import time
-from bokeh.plotting import figure
-from bokeh.models import ColumnDataSource
-
-# ... [rest of the initial code remains unchanged] ...
-
-# 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 = 0  # Use 0 as a default index
-
-    # Load the history of clicks
-    history = load_history()
-
-    # Display the buttons for each URL
-    for url, name, emoji in zip(urls, url_names, url_emojis):
-        #if st.button(f"{emoji} {name}"):
-        if st.button(f"{emoji} {name}", key=url):  # using the URL as the unique key
-            # Open the URL in a new browser tab using JavaScript
-            st.write('<script>window.open("'+url+'", "_blank");</script>', unsafe_allow_html=True)
-            # Update the history of clicks
-            history[url] += 1
-            save_history(history)
-        # Display the number of times the URL was opened below its corresponding button
-        st.write(f"Clicked: {history[url]} times")
-
-def get_url_emojis(url_names):
-    url_emojis = []
-    for name in url_names:
-        associated_emoji = "🔗"
-        for keyword, emoji in emoji_mapping.items():
-            if keyword in name:
-                associated_emoji = emoji
-                break
-        url_emojis.append(associated_emoji)
-    return url_emojis
+def get_emoji(name):
+    for key, emoji in emoji_mapping.items():
+        if key in name:
+            return emoji
+    return "🔗"
 
-def load_history():
+def load_votes():
     try:
-        with open("click_history.txt", "r") as f:
+        with open("votes.json", "r") as f:
             return json.load(f)
     except FileNotFoundError:
         return {url: 0 for url in urls}
 
-def save_history(history):
-    with open("click_history.txt", "w") as f:
-        json.dump(history, f)
+def save_votes(votes):
+    with open("votes.json", "w") as f:
+        json.dump(votes, f)
 
 def main():
-    if "auto_repeat" not in st.session_state:
-        st.session_state.auto_repeat = "On"
+    votes = load_votes()
     
-    history = load_history()
+    # Create sorted list of items
+    items = [{"url": url, "name": url.split('/')[-1], 
+              "emoji": get_emoji(url.split('/')[-1]), 
+              "votes": votes[url]} for url in urls]
+    items.sort(key=lambda x: (-x["votes"], x["name"]))
     
-    # Create list of dictionaries with all data
-    url_data = [{'url': url, 'name': name, 'emoji': emoji, 'clicks': history[url]} 
-                for url, name, emoji in zip(urls, url_names, get_url_emojis(url_names))]
+    # Display buttons in grid
+    cols = st.columns(4)
+    for i, item in enumerate(items):
+        with cols[i % 4]:
+            if st.button(f"{item['emoji']} {item['name']}", key=item['url']):
+                votes[item['url']] += 1
+                save_votes(votes)
+                st.rerun()
+            st.write(f"Votes: {item['votes']}")
     
-    # Sort alphabetically on initial load if no clicks
-    if all(data['clicks'] == 0 for data in url_data):
-        url_data.sort(key=lambda x: x['name'].lower())
-    else:
-        # Sort by clicks (descending) and then alphabetically for ties
-        url_data.sort(key=lambda x: (-x['clicks'], x['name'].lower()))
-
-    # Display buttons in columns
-    num_cols = min(4, len(url_data))
-    cols = st.columns(num_cols)
-
-    for i, data in enumerate(url_data):
-        col = cols[i % num_cols]
-        with col:
-            if st.button(f"{data['emoji']} {data['name']}", key=f"btn_{data['url']}"):
-                st.write(f'<script>window.open("{data["url"]}", "_blank");</script>', 
-                        unsafe_allow_html=True)
-                history[data['url']] += 1
-                save_history(history)
-                st.experimental_rerun()  # Rerun to update sorting
-            st.write(f"Clicked: {data['clicks']} times")
-
-    # Display graph for non-zero clicks
-    non_zero_data = [d for d in url_data if d['clicks'] > 0]
-    if non_zero_data:
-        source = ColumnDataSource(data=dict(
-            urls=[d['name'] for d in non_zero_data],
-            counts=[d['clicks'] for d in non_zero_data]
-        ))
-
-        p = figure(x_range=[d['name'] for d in non_zero_data], 
-                  height=350, 
-                  title="Click Counts per URL",
-                  toolbar_location=None)
-        p.vbar(x='urls', top='counts', width=0.9, source=source)
+    # Display vote graph
+    if any(votes.values()):
+        source = ColumnDataSource({
+            'names': [i["name"] for i in items if votes[i["url"]] > 0],
+            'votes': [i["votes"] for i in items if votes[i["url"]] > 0]
+        })
+        p = figure(x_range=source.data['names'], height=350, title="Vote Counts")
+        p.vbar(x='names', top='votes', width=0.9, source=source)
         p.xaxis.major_label_orientation = 1.2
         st.bokeh_chart(p)
 
-    # Timer logic
-    if st.session_state.auto_repeat == "On":
-        timer_placeholder = st.empty()
-        for i in range(10, 0, -1):
-            timer_placeholder.text(f"Reloading in {i} seconds...")
-            time.sleep(1)
-        st.rerun()
-
 if __name__ == "__main__":
     main()