Update app.py

app.py

@@ -12,6 +12,8 @@ from PIL import Image
 import glob
 import re
 from urllib.parse import quote
+import base64
+import io
 
 # App Configuration
 Site_Name = '🤖🧠Chat & Quote Node📝🔬'
@@ -26,23 +28,36 @@ st.set_page_config(
     initial_sidebar_state="auto"
 )
 
-# Fun usernames with emojis - the VIP list of quirky characters! 🎉😜
-FUN_USERNAMES = [
-    "CosmicJester 🌌", "PixelPanda 🐼", "QuantumQuack 🦆", "StellarSquirrel 🐿️",
-    "GizmoGuru ⚙️", "NebulaNinja 🌠", "ByteBuster 💾", "GalacticGopher 🌍",
-    "RocketRaccoon 🚀", "EchoElf 🧝", "PhantomFox 🦊", "WittyWizard 🧙",
-    "LunarLlama 🌙", "SolarSloth ☀️", "AstroAlpaca 🦙", "CyberCoyote 🐺",
-    "MysticMoose 🦌", "GlitchGnome 🧚", "VortexViper 🐍", "ChronoChimp 🐒"
-]
+# Fun usernames with emojis and rhyming labels - the VIP list of quirky characters! 🎉😜
+FUN_USERNAMES = {
+    "CosmicJester 🌌": "Jest Best",
+    "PixelPanda 🐼": "Panda Dandy",
+    "QuantumQuack 🦆": "Quack Pack",
+    "StellarSquirrel 🐿️": "Squirrel Whirl",
+    "GizmoGuru ⚙️": "Guru Brew",
+    "NebulaNinja 🌠": "Ninja Zinger",
+    "ByteBuster 💾": "Buster Cluster",
+    "GalacticGopher 🌍": "Gopher Loafer",
+    "RocketRaccoon 🚀": "Raccoon Zoom",
+    "EchoElf 🧝": "Elf Shelf",
+    "PhantomFox 🦊": "Fox Rocks",
+    "WittyWizard 🧙": "Wizard Blizzard",
+    "LunarLlama 🌙": "Llama Drama",
+    "SolarSloth ☀️": "Sloth Growth",
+    "AstroAlpaca 🦙": "Alpaca Plaque",
+    "CyberCoyote 🐺": "Coyote Byte",
+    "MysticMoose 🦌": "Moose Juice",
+    "GlitchGnome 🧚": "Gnome Roam",
+    "VortexViper 🐍": "Viper Hyper",
+    "ChronoChimp 🐒": "Chimp Whimp"
+}
 
-# Directories for chat, votes, and media - the secret vaults where treasures are stashed! 🗄️🔒
+# Directories for chat and votes - the secret vaults where treasures are stashed! 🗄️🔒
 CHAT_DIR = "chat_logs"
 VOTE_DIR = "vote_logs"
-MEDIA_DIR = "media"
 STATE_FILE = "user_state.txt"
 os.makedirs(CHAT_DIR, exist_ok=True)
 os.makedirs(VOTE_DIR, exist_ok=True)
-os.makedirs(MEDIA_DIR, exist_ok=True)
 
 # Persistent files - the grand tomes of chatter and votes! 📖✨
 CHAT_FILE = os.path.join(CHAT_DIR, "global_chat.md")
@@ -122,6 +137,13 @@ def get_user_list(chat_content):
             users.add(user)
     return sorted(list(users))
 
+# Message suggestion loader - the oracle of past chats! 🔮💬
+def get_message_suggestions(chat_content, prefix):
+    """🔮 Pulls past messages from the cosmic archives for autocomplete magic! ✨"""
+    lines = chat_content.split('\n')
+    messages = [line.split(': ', 1)[1] for line in lines if ': ' in line and line.strip()]
+    return [msg for msg in messages if msg.lower().startswith(prefix.lower())][:5]  # Top 5 matches
+
 # 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! 🗣️"""
@@ -218,6 +240,23 @@ def load_username():
             print(f"Failed to load username: {e}")
     return None
 
+# Image paste handler - the cosmic courier delivering pasted pics! 📸🚀
+def save_pasted_image(image_data):
+    """📸 Saves a pasted image to the root vault with a cosmic name! 🌠"""
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    filename = f"paste_{timestamp}.png"
+    filepath = os.path.join('./', filename)  # Root directory
+    try:
+        if ',' in image_data:
+            image_data = image_data.split(',')[1]
+        img_bytes = base64.b64decode(image_data)
+        img = Image.open(io.BytesIO(img_bytes))
+        img.save(filepath, "PNG")
+        return filename
+    except Exception as e:
+        print(f"Failed to save pasted image: {e}")
+        return None
+
 # Media HTML generators - IVA’s multimedia magic! 🎥🎶🖼️
 def get_video_html(video_path, width="100%"):
     """🎬 Rolls out the red carpet for videos - autoplay and loop like a star! 🌟"""
@@ -249,8 +288,7 @@ async def websocket_handler(websocket, path):
         room_id = "chat"  # Single room "Sector 🌌"
         active_connections.setdefault(room_id, {})[client_id] = websocket
         print(f"Client {client_id} joined the Sector party!")
-        # Auto-join message for every connection
-        username = st.session_state.get('username', random.choice(FUN_USERNAMES))
+        username = st.session_state.get('username', random.choice(list(FUN_USERNAMES.keys())))
         save_chat_entry("System 🌟", f"{username} has joined {START_ROOM}!")
 
         async for message in websocket:
@@ -298,7 +336,7 @@ async def run_websocket_server():
 # Chat interface maker - the stage builder for our Sector extravaganza! 🎭🏟️
 def create_streamlit_interface(initial_username):
     """🖌️ Sets up the Sector stage with pulsing timers, voting, and IVA media flair! 🌟"""
-    # Custom CSS for a sleek chat box and timer
+    # Custom CSS and JS for chat box, timer, and paste detection
     st.markdown("""
         <style>
         .chat-box {
@@ -322,11 +360,34 @@ def create_streamlit_interface(initial_username):
             100% { transform: scale(1); }
         }
         </style>
+        <script>
+        document.addEventListener('paste', function(e) {
+            const items = (e.clipboardData || window.clipboardData).items;
+            for (let i = 0; i < items.length; i++) {
+                if (items[i].type.indexOf('image') !== -1) {
+                    const blob = items[i].getAsFile();
+                    const reader = new FileReader();
+                    reader.onload = function(event) {
+                        const imageData = event.target.result;
+                        sessionStorage.setItem('pastedImage', imageData);
+                        document.getElementById('message_input').value = 'PastedImage:' + blob.name;
+                        document.getElementById('send_button').click();
+                    };
+                    reader.readAsDataURL(blob);
+                }
+            }
+        });
+        document.getElementById('message_input')?.addEventListener('keypress', function(e) {
+            if (e.key === 'Enter') {
+                document.getElementById('send_button').click();
+            }
+        });
+        </script>
     """, unsafe_allow_html=True)
 
     # Title and intro
     st.title(f"{Site_Name}")
-    st.markdown(f"Welcome to {START_ROOM} - chat, vote, and enjoy IVA’s media magic! 🎉")
+    st.markdown(f"Welcome to {START_ROOM} - chat, vote, paste images, and enjoy IVA’s media magic! 🎉")
 
     # Load or set username
     saved_username = load_username()
@@ -347,6 +408,8 @@ def create_streamlit_interface(initial_username):
         st.session_state.quote_index = random.randint(0, max(0, len(quotes) - 1)) if quotes else 0
     if 'quote_source' not in st.session_state:
         st.session_state.quote_source = "famous"
+    if 'pasted_image' not in st.session_state:
+        st.session_state.pasted_image = None
 
     # Chat section
     st.subheader(f"{START_ROOM} Chat 💬")
@@ -366,22 +429,44 @@ def create_streamlit_interface(initial_username):
                     st.rerun()
 
     user_list = get_user_list(chat_content)
-    new_username = st.selectbox("Switch User", user_list + [st.session_state.username], index=len(user_list))
+    new_username = st.selectbox("Switch Star", user_list + [st.session_state.username], index=len(user_list))
     if new_username != st.session_state.username:
         save_chat_entry("System 🌟", f"{st.session_state.username} switched to {new_username} in {START_ROOM}!")
         st.session_state.username = new_username
         st.session_state.timer_start = time.time()
         save_username(st.session_state.username)
 
-    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)
+    # Message input with search engine and paste handling
+    agent_icon = st.session_state.username.split()[-1]  # Extract emoji
+    agent_label = FUN_USERNAMES.get(st.session_state.username, "Chatty Brat")
+    message = st.text_input(f"Message Mate {agent_icon} - {agent_label} 🔍📸", key="message_input", placeholder="Type or paste here...")
+    suggestions = get_message_suggestions(chat_content, message) if message else []
+    if suggestions:
+        suggestion = st.selectbox("Chat Snap", [""] + suggestions, index=0)
+        if suggestion:
+            message = suggestion
+            st.session_state.message_input = suggestion
+
+    col_send, _ = st.columns([1, 5])
+    with col_send:
+        send_button = st.button("Send Bend 🚀", key="send_button")
+    if send_button and message.strip():
+        if message.startswith("PastedImage:"):
+            # Handle pasted image
+            image_data = st.session_state.get('pasted_image')
+            if image_data:
+                filename = save_pasted_image(image_data)
+                if filename:
+                    save_chat_entry(st.session_state.username, f"Pasted image: {filename}")
+                    st.session_state.pasted_image = None
+        else:
+            save_chat_entry(st.session_state.username, message)
         st.session_state.timer_start = time.time()
         save_username(st.session_state.username)
         st.rerun()
 
     # Quote section
-    st.subheader("Quote of the Moment 📝")
+    st.subheader("Quote Note 📝")
     quotes = load_quotes(st.session_state.quote_source)
     if quotes:
         st.session_state.quote_index = st.session_state.quote_index % len(quotes)
@@ -390,7 +475,7 @@ def create_streamlit_interface(initial_username):
         with col1:
             st.markdown(quote)
         with col2:
-            if st.button("👍 Upvote", key="quote_vote"):
+            if st.button("👍 Vote Float", key="quote_vote"):
                 user_hash = generate_user_hash()
                 save_vote(QUOTE_VOTES_FILE, quote, user_hash)
                 st.session_state.timer_start = time.time()
@@ -407,15 +492,15 @@ def create_streamlit_interface(initial_username):
     else:
         st.markdown("No quotes available - check back later! 📭")
 
-    # IVA Media Gallery
-    st.subheader("IVA Media Gallery 🎨🎶🎥")
+    # IVA Media Gallery - Root Directory
+    st.subheader("IVA Gallery 🎨🎶🎥")
     media_files = (
-        glob.glob(f"{MEDIA_DIR}/*.png") + glob.glob(f"{MEDIA_DIR}/*.jpg") + glob.glob(f"{MEDIA_DIR}/*.jpeg") +
-        glob.glob(f"{MEDIA_DIR}/*.mp3") + glob.glob(f"{MEDIA_DIR}/*.wav") +
-        glob.glob(f"{MEDIA_DIR}/*.mp4")
+        glob.glob("./*.png") + glob.glob("./*.jpg") + glob.glob("./*.jpeg") +
+        glob.glob("./*.mp3") + glob.glob("./*.wav") +
+        glob.glob("./*.mp4")
     )
     if media_files:
-        media_cols = st.slider("Gallery Columns", min_value=1, max_value=5, value=3)
+        media_cols = st.slider("Gallery Rally", min_value=1, max_value=5, value=3)
         cols = st.columns(media_cols)
         for idx, media_file in enumerate(media_files):
             with cols[idx % media_cols]:
@@ -425,18 +510,18 @@ def create_streamlit_interface(initial_username):
                     st.markdown(get_audio_html(media_file, width="100%"), unsafe_allow_html=True)
                 elif media_file.endswith('.mp4'):
                     st.markdown(get_video_html(media_file, width="100%"), unsafe_allow_html=True)
-                if st.button(f"👍 Upvote {os.path.basename(media_file)}", key=f"media_vote_{idx}"):
+                if st.button(f"👍 Vote Tote {os.path.basename(media_file)}", key=f"media_vote_{idx}"):
                     user_hash = generate_user_hash()
                     save_vote(IMAGE_VOTES_FILE, media_file, user_hash)
                     st.session_state.timer_start = time.time()
                     save_username(st.session_state.username)
                     st.rerun()
     else:
-        st.error("No media files (images, audio, video) found in 'media' directory!")
+        st.error("No media files (images, audio, video) found in root directory!")
 
     # Refresh rate controls with pulsing timer
-    st.subheader("Set Refresh Rate ⏳")
-    refresh_rate = st.slider("Refresh Rate (seconds)", min_value=1, max_value=300, value=st.session_state.refresh_rate, step=1)
+    st.subheader("Refresh Dash ⏳")
+    refresh_rate = st.slider("Refresh Clash", min_value=1, max_value=300, value=st.session_state.refresh_rate, step=1)
     if refresh_rate != st.session_state.refresh_rate:
         st.session_state.refresh_rate = refresh_rate
         st.session_state.timer_start = time.time()
@@ -444,17 +529,17 @@ def create_streamlit_interface(initial_username):
 
     col1, col2, col3 = st.columns(3)
     with col1:
-        if st.button("🐇 Small (1s)"):
+        if st.button("🐇 Small Call (1s)"):
             st.session_state.refresh_rate = 1
             st.session_state.timer_start = time.time()
             save_username(st.session_state.username)
     with col2:
-        if st.button("🐢 Medium (5s)"):
+        if st.button("🐢 Medium Fling (5s)"):
             st.session_state.refresh_rate = 5
             st.session_state.timer_start = time.time()
             save_username(st.session_state.username)
     with col3:
-        if st.button("🐘 Large (5m)"):
+        if st.button("🐘 Large Surge (5m)"):
             st.session_state.refresh_rate = 300
             st.session_state.timer_start = time.time()
             save_username(st.session_state.username)
@@ -474,7 +559,7 @@ def create_streamlit_interface(initial_username):
     st.rerun()
 
     # Sidebar vote stats
-    st.sidebar.subheader("Vote Totals")
+    st.sidebar.subheader("Vote Loads")
     chat_votes = load_votes(QUOTE_VOTES_FILE)
     image_votes = load_votes(IMAGE_VOTES_FILE)
     for item, count in chat_votes.items():
@@ -493,7 +578,7 @@ async def main():
         server_task = asyncio.create_task(run_websocket_server())
 
     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)
+    initial_username = query_params.get("username", random.choice(list(FUN_USERNAMES.keys()))) if query_params else random.choice(list(FUN_USERNAMES.keys()))
     print(f"Welcoming {initial_username} to the Sector bash!")
 
     create_streamlit_interface(initial_username)