Update app.py

app.py

@@ -7,6 +7,7 @@ from datetime import datetime
 import os
 import random
 import time
+import hashlib
 
 # Fun usernames with emojis - the VIP list of quirky characters! 🎉😜
 FUN_USERNAMES = [
@@ -17,12 +18,17 @@ FUN_USERNAMES = [
     "MysticMoose 🦌", "GlitchGnome 🧚", "VortexViper 🐍", "ChronoChimp 🐒"
 ]
 
-# Directory for chat logs - the secret vault where tales are stashed! 🗄️🔒
+# Directories for chat and votes - the secret vaults where tales and tallies are stashed! 🗄️🔒
 CHAT_DIR = "chat_logs"
+VOTE_DIR = "vote_logs"
 os.makedirs(CHAT_DIR, exist_ok=True)
+os.makedirs(VOTE_DIR, exist_ok=True)
 
-# Persistent chat file - the grand tome of all chatter! 📖✨
+# Persistent files - the grand tomes of chatter and votes! 📖✨
 CHAT_FILE = os.path.join(CHAT_DIR, "global_chat.md")
+QUOTE_VOTES_FILE = os.path.join(VOTE_DIR, "quote_votes.md")
+IMAGE_VOTES_FILE = os.path.join(VOTE_DIR, "image_votes.md")
+HISTORY_FILE = os.path.join(VOTE_DIR, "vote_history.md")
 
 # Node name - the app’s codename generator, sneaky and slick! 🕵️‍♂️💾
 def get_node_name():
@@ -72,6 +78,82 @@ def get_user_list(chat_content):
             users.add(user)
     return sorted(list(users))
 
+# Quote loader - the sage pulling wisdom from the ages! 📜🧙
+def load_quotes(source="famous"):
+    """📚 Grabs a stack of wise words from famous folks or custom quips! 🗣️"""
+    famous_quotes = [
+        "The true sign of intelligence is not knowledge but imagination. – Albert Einstein",
+        "I have not failed. I've just found 10,000 ways that won't work. – Thomas Edison",
+        "Innovation distinguishes between a leader and a follower. – Steve Jobs",
+        "Research is what I'm doing when I don't know what I'm doing. – Wernher von Braun",
+        "The only way to discover the limits of the possible is to go beyond them into the impossible. – Arthur C. Clarke",
+        "Success is a science; if you have the conditions, you get the result. – Oscar Wilde",
+        "An expert is a person who has made all the mistakes that can be made in a very narrow field. – Niels Bohr",
+        "The important thing is to not stop questioning. Curiosity has its own reason for existing. – Albert Einstein",
+        "The best way to predict the future is to invent it. – Alan Kay",
+        "If I have seen further it is by standing on the shoulders of Giants. – Isaac Newton",
+        "Logic will get you from A to B. Imagination will take you everywhere. – Albert Einstein",
+        "Imagination is more important than knowledge. Knowledge is limited. Imagination encircles the world. – Albert Einstein",
+        "Science is a way of thinking much more than it is a body of knowledge. – Carl Sagan",
+        "We cannot solve our problems with the same thinking we used when we created them. – Albert Einstein",
+        "The true method of knowledge is experiment. – William Blake",
+        "The scientist is not a person who gives the right answers, he's one who asks the right questions. – Claude Levi-Strauss",
+        "It's kind of fun to do the impossible. – Walt Disney",
+        "Any sufficiently advanced technology is indistinguishable from magic. – Arthur C. Clarke",
+        "Creativity is intelligence having fun. – Albert Einstein",
+        "To invent, you need a good imagination and a pile of junk. – Thomas Edison"
+    ]
+    custom_quotes = [
+        "Every age unfolds a new lesson. Life's chapters evolve, each teaching us anew.",
+        "From infancy to twilight, our journey is painted in growth. Every stage shines with its own wisdom.",
+        "Love is the universal language, transcending boundaries and touching souls.",
+        "Through love, we find connection, unity, and the essence of existence."
+    ]
+    return famous_quotes if source == "famous" else custom_quotes
+
+# Vote saver - the tally keeper counting thumbs up! 👍📊
+def save_vote(file, item, user_hash):
+    """✍️ Tallies a vote in the grand ledger - your opinion matters! 🗳️"""
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    entry = f"[{timestamp}] {user_hash} voted for {item}"
+    try:
+        with open(file, 'a') as f:
+            f.write(f"{entry}\n")
+        with open(HISTORY_FILE, 'a') as f:
+            f.write(f"- {timestamp} - User {user_hash} voted for {item}\n")
+        return True
+    except Exception as e:
+        print(f"Vote save flop: {e}")
+        return False
+
+# Vote loader - the scorekeeper tallying the crowd’s cheers! 🎉🏅
+def load_votes(file):
+    """📈 Counts the votes from the ledger - who’s winning the popularity contest? 🏆"""
+    if not os.path.exists(file):
+        with open(file, 'w') as f:
+            f.write("# Vote Tally\n\nNo votes yet - get clicking! 🖱️\n")
+    try:
+        with open(file, 'r') as f:
+            lines = f.read().strip().split('\n')
+        votes = {}
+        for line in lines[2:]:  # Skip header
+            if line.strip() and 'voted for' in line:
+                item = line.split('voted for ')[1]
+                votes[item] = votes.get(item, 0) + 1
+        return votes
+    except Exception as e:
+        print(f"Vote load oopsie: {e}")
+        return {}
+
+# User hash generator - the secret agent giving you a cool code! 🕵️‍♂️🔑
+def generate_user_hash():
+    """🕵️ Crafts a snazzy 8-digit ID badge - you’re in the club now! 🎟️"""
+    if 'user_hash' not in st.session_state:
+        session_id = str(random.getrandbits(128))
+        hash_object = hashlib.md5(session_id.encode())
+        st.session_state['user_hash'] = hash_object.hexdigest()[:8]
+    return st.session_state['user_hash']
+
 active_connections = {}
 
 # WebSocket handler - the bouncer at the chat rave, keeping it hopping! 🎉🚪
@@ -123,7 +205,7 @@ async def start_websocket_server(host='0.0.0.0', port=8765):
 
 # Chat interface maker - the stage builder for our chat extravaganza! 🎭🏟️
 def create_streamlit_interface(initial_username):
-    """🖌️ Sets up the chat stage with live updates, timers, and refresh rate flair! 🌟"""
+    """🖌️ Sets up the chat stage with live updates, timers, voting, and refresh flair! 🌟"""
     # Custom CSS for a sleek chat box and timer
     st.markdown("""
         <style>
@@ -133,7 +215,7 @@ def create_streamlit_interface(initial_username):
             color: #d4d4d4;
             padding: 10px;
             border-radius: 5px;
-            height: 400px;
+            height: 300px;
             overflow-y: auto;
         }
         .timer {
@@ -145,46 +227,97 @@ def create_streamlit_interface(initial_username):
     """, unsafe_allow_html=True)
 
     # Title and intro
-    st.title(f"Chat Node: {NODE_NAME}")
-    st.markdown("Chat live, switch names, and tweak the refresh rate! 🎉")
+    st.title(f"Chat & Quote Node: {NODE_NAME}")
+    st.markdown("Chat, vote on messages, quotes, and images - keep the party rocking! 🎉")
 
-    # Session state for username and refresh rate
+    # Session state for username, refresh rate, and quote index
     if 'username' not in st.session_state:
         st.session_state.username = initial_username
     if 'refresh_rate' not in st.session_state:
-        st.session_state.refresh_rate = 5  # Default 5 seconds
+        st.session_state.refresh_rate = 5
     if 'last_refresh' not in st.session_state:
         st.session_state.last_refresh = time.time()
+    if 'quote_index' not in st.session_state:
+        st.session_state.quote_index = random.randint(0, len(load_quotes("famous")) - 1)
+    if 'quote_source' not in st.session_state:
+        st.session_state.quote_source = "famous"
 
-    # Chat display
+    # Chat section
+    st.subheader("Chat Room 💬")
     chat_content = load_chat()
-    st.markdown(f"<div class='chat-box'>{chat_content}</div>", unsafe_allow_html=True)
+    chat_lines = chat_content.split('\n')
+    for i, line in enumerate(chat_lines):
+        if line.strip() and ': ' in line:
+            col1, col2 = st.columns([5, 1])
+            with col1:
+                st.markdown(line)
+            with col2:
+                if st.button(f"👍", key=f"chat_vote_{i}"):
+                    user_hash = generate_user_hash()
+                    save_vote(QUOTE_VOTES_FILE, line, user_hash)
+                    st.session_state.last_refresh = time.time()
+                    st.rerun()
 
-    # User switcher
     user_list = get_user_list(chat_content)
     new_username = st.selectbox("Switch User", user_list + [st.session_state.username], index=len(user_list))
     if new_username != st.session_state.username:
         st.session_state.username = new_username
 
-    # Message input and send
     message = st.text_input("Message", placeholder="Type your epic line here! ✍️")
     if st.button("Send 🚀") and message.strip():
         save_chat_entry(st.session_state.username, message)
-        st.session_state.last_refresh = time.time()  # Reset timer on send
+        st.session_state.last_refresh = time.time()
+        st.rerun()
+
+    # Quote section
+    st.subheader("Quote of the Moment 📝")
+    quotes = load_quotes(st.session_state.quote_source)
+    quote = quotes[st.session_state.quote_index]
+    col1, col2 = st.columns([5, 1])
+    with col1:
+        st.markdown(quote)
+    with col2:
+        if st.button("👍 Upvote", key="quote_vote"):
+            user_hash = generate_user_hash()
+            save_vote(QUOTE_VOTES_FILE, quote, user_hash)
+            st.session_state.last_refresh = time.time()
+            st.rerun()
+    if time.time() - st.session_state.last_refresh > 10:  # 10s quote refresh
+        st.session_state.quote_index = (st.session_state.quote_index + 1) % len(quotes)
+        st.session_state.quote_source = "custom" if st.session_state.quote_source == "famous" else "famous"
+        st.session_state.last_refresh = time.time()
         st.rerun()
 
+    # Image voting section
+    st.subheader("Image Voting 🖼️")
+    image_dir = '.'
+    valid_extensions = ('.png', '.jpg', '.jpeg', '.gif', '.bmp', '.tiff', '.webp')
+    images = [f for f in os.listdir(image_dir) if f.lower().endswith(valid_extensions)]
+    if len(images) >= 2:
+        image1, image2 = random.sample(images, 2)
+        col1, col2 = st.columns(2)
+        with col1:
+            st.image(os.path.join(image_dir, image1))
+            if st.button(f"👍 Upvote {image1}", key=f"img_vote_{image1}"):
+                user_hash = generate_user_hash()
+                save_vote(IMAGE_VOTES_FILE, image1, user_hash)
+                st.session_state.last_refresh = time.time()
+                st.rerun()
+        with col2:
+            st.image(os.path.join(image_dir, image2))
+            if st.button(f"👍 Upvote {image2}", key=f"img_vote_{image2}"):
+                user_hash = generate_user_hash()
+                save_vote(IMAGE_VOTES_FILE, image2, user_hash)
+                st.session_state.last_refresh = time.time()
+                st.rerun()
+    else:
+        st.error("Need at least 2 images in the directory for voting!")
+
     # Refresh rate controls
     st.subheader("Set Refresh Rate ⏳")
-    refresh_rate = st.slider(
-        "Refresh Rate (seconds)",
-        min_value=1,
-        max_value=300,  # 5 minutes = 300 seconds
-        value=st.session_state.refresh_rate,
-        step=1
-    )
+    refresh_rate = st.slider("Refresh Rate (seconds)", min_value=1, max_value=300, value=st.session_state.refresh_rate, step=1)
     st.session_state.refresh_rate = refresh_rate
 
-    # Emoji buttons for quick settings
     col1, col2, col3 = st.columns(3)
     with col1:
         if st.button("🐇 Small (1s)"):
@@ -214,6 +347,15 @@ def create_streamlit_interface(initial_username):
     else:
         st.markdown(f"<script>window.setTimeout(() => window.location.reload(), {int((st.session_state.refresh_rate - elapsed) * 1000)});</script>", unsafe_allow_html=True)
 
+    # Sidebar vote stats
+    st.sidebar.subheader("Vote Totals")
+    chat_votes = load_votes(QUOTE_VOTES_FILE)
+    image_votes = load_votes(IMAGE_VOTES_FILE)
+    for item, count in chat_votes.items():
+        st.sidebar.write(f"{item}: {count} votes")
+    for image, count in image_votes.items():
+        st.sidebar.write(f"{image}: {count} votes")
+
 # Main event - the ringmaster kicking off the chat circus! 🎪🤡
 async def main():
     """🎤 Drops the mic and starts the chat party - it’s showtime, folks! 🎉"""
@@ -221,7 +363,6 @@ async def main():
     NODE_NAME, port = get_node_name()
     await start_websocket_server()
 
-    # Grab username from URL or roll the dice! 🎲
     query_params = st.query_params if hasattr(st, 'query_params') else {}
     initial_username = query_params.get("username", random.choice(FUN_USERNAMES)) if query_params else random.choice(FUN_USERNAMES)
     print(f"Welcoming {initial_username} to the chat bash!")