Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,15 +58,15 @@ UNICODE_FONTS = [
|
|
| 58 |
server_running = False
|
| 59 |
server_task = None
|
| 60 |
|
| 61 |
-
|
| 62 |
"""π - Naming Node with Code - Spins a name, oh so bold!"""
|
| 63 |
action = "π - Naming Node with Code - Spins a name, oh so bold!"
|
| 64 |
username = st.session_state.get('username', 'System π')
|
| 65 |
-
|
| 66 |
parser = argparse.ArgumentParser(description='Start a chat node with a specific name')
|
| 67 |
parser.add_argument('--node-name', type=str, default=None)
|
| 68 |
parser.add_argument('--port', type=int, default=8501)
|
| 69 |
-
args =
|
| 70 |
return args.node_name or f"node-{uuid.uuid4().hex[:8]}", args.port
|
| 71 |
|
| 72 |
def log_action(username, action):
|
|
@@ -99,13 +99,13 @@ async def load_chat():
|
|
| 99 |
await asyncio.to_thread(lambda: open(CHAT_FILE, 'w').write(f"# {START_ROOM} Chat\n\nWelcome to the cosmic hub - start chatting! π€\n"))
|
| 100 |
with open(CHAT_FILE, 'r') as f:
|
| 101 |
content = await asyncio.to_thread(f.read)
|
| 102 |
-
return content
|
| 103 |
|
| 104 |
-
|
| 105 |
"""π₯ - Crew Clue Brew - Spots whoβs who in the crew!"""
|
| 106 |
action = "π₯ - Crew Clue Brew - Spots whoβs who in the crew!"
|
| 107 |
username = st.session_state.get('username', 'System π')
|
| 108 |
-
|
| 109 |
users = set()
|
| 110 |
for line in chat_content.split('\n'):
|
| 111 |
if line.strip() and ': ' in line:
|
|
@@ -269,11 +269,11 @@ async def run_websocket_server():
|
|
| 269 |
server_running = True
|
| 270 |
await server.wait_closed()
|
| 271 |
|
| 272 |
-
|
| 273 |
"""π¨ - UI Brew Crew - Builds the view, so new!"""
|
| 274 |
action = "π¨ - UI Brew Crew - Builds the view, so new!"
|
| 275 |
username = st.session_state.get('username', 'System π')
|
| 276 |
-
|
| 277 |
|
| 278 |
# Dynamic title based on username
|
| 279 |
if 'username' in st.session_state:
|
|
@@ -298,7 +298,7 @@ async def create_streamlit_interface():
|
|
| 298 |
if 'client_id' not in st.session_state:
|
| 299 |
st.session_state.client_id = str(uuid.uuid4())
|
| 300 |
if 'username' not in st.session_state:
|
| 301 |
-
chat_content =
|
| 302 |
available_names = [name for name in FUN_USERNAMES if not any(f"{name} has joined" in line for line in chat_content.split('\n'))]
|
| 303 |
st.session_state.username = random.choice(available_names) if available_names else random.choice(FUN_USERNAMES)
|
| 304 |
|
|
@@ -308,7 +308,7 @@ async def create_streamlit_interface():
|
|
| 308 |
if 'timer_start' not in st.session_state:
|
| 309 |
st.session_state.timer_start = time.time()
|
| 310 |
if 'quote_index' not in st.session_state:
|
| 311 |
-
quotes =
|
| 312 |
st.session_state.quote_index = random.randint(0, max(0, len(quotes) - 1)) if quotes else 0
|
| 313 |
if 'quote_source' not in st.session_state:
|
| 314 |
st.session_state.quote_source = "famous"
|
|
@@ -349,17 +349,17 @@ async def create_streamlit_interface():
|
|
| 349 |
# Handle pasted image data
|
| 350 |
pasted_image = st.session_state.get('pasted_image_data')
|
| 351 |
if pasted_image:
|
| 352 |
-
filename =
|
| 353 |
if filename:
|
| 354 |
-
|
| 355 |
st.session_state.pasted_image_data = None
|
| 356 |
st.rerun()
|
| 357 |
|
| 358 |
# Chat section with quoting
|
| 359 |
st.subheader(f"{START_ROOM} Chat π¬")
|
| 360 |
-
chat_content =
|
| 361 |
chat_lines = chat_content.split('\n')
|
| 362 |
-
chat_votes =
|
| 363 |
for i, line in enumerate(chat_lines):
|
| 364 |
if line.strip() and ': ' in line:
|
| 365 |
col1, col2, col3 = st.columns([4, 1, 1])
|
|
@@ -369,11 +369,11 @@ async def create_streamlit_interface():
|
|
| 369 |
vote_count = chat_votes.get(line.split('. ')[1] if '. ' in line else line, 0)
|
| 370 |
if st.button(f"π {vote_count}", key=f"chat_vote_{i}"):
|
| 371 |
comment = st.session_state.message_text
|
| 372 |
-
|
| 373 |
if st.session_state.pasted_image_data:
|
| 374 |
-
filename =
|
| 375 |
if filename:
|
| 376 |
-
|
| 377 |
st.session_state.pasted_image_data = None
|
| 378 |
st.session_state.message_text = ''
|
| 379 |
st.rerun()
|
|
@@ -387,18 +387,17 @@ async def create_streamlit_interface():
|
|
| 387 |
st.markdown(f"### Quoting: {st.session_state.quote_line}")
|
| 388 |
quote_response = st.text_area("Add your response", key="quote_response")
|
| 389 |
if st.button("Send Quote π", key="send_quote"):
|
| 390 |
-
|
| 391 |
-
"""π’ - Quote Float Boat - Echoes chat, we gloat!"""
|
| 392 |
action = "π’ - Quote Float Boat - Echoes chat, we gloat!"
|
| 393 |
-
|
| 394 |
markdown_response = f"### Quote Response\n- **Original**: {st.session_state.quote_line}\n- **{st.session_state.username} Replies**: {quote_response}"
|
| 395 |
if st.session_state.pasted_image_data:
|
| 396 |
-
filename =
|
| 397 |
if filename:
|
| 398 |
markdown_response += f"\n- **Image**: "
|
| 399 |
st.session_state.pasted_image_data = None
|
| 400 |
-
|
| 401 |
-
|
| 402 |
del st.session_state.quote_line
|
| 403 |
st.session_state.message_text = ''
|
| 404 |
st.rerun()
|
|
@@ -406,18 +405,18 @@ async def create_streamlit_interface():
|
|
| 406 |
# Username change dropdown
|
| 407 |
new_username = st.selectbox("Change Name", [""] + FUN_USERNAMES, index=0)
|
| 408 |
if new_username and new_username != st.session_state.username:
|
| 409 |
-
|
| 410 |
st.session_state.username = new_username
|
| 411 |
st.rerun()
|
| 412 |
|
| 413 |
# Message input
|
| 414 |
message = st.text_input(f"Message as {st.session_state.username}", key="message_input", value=st.session_state.message_text, on_change=lambda: st.session_state.update(message_text=st.session_state.message_input))
|
| 415 |
if st.button("Send π", key="send_button") and message.strip():
|
| 416 |
-
|
| 417 |
if st.session_state.pasted_image_data:
|
| 418 |
-
filename =
|
| 419 |
if filename:
|
| 420 |
-
|
| 421 |
st.session_state.pasted_image_data = None
|
| 422 |
st.session_state.message_text = ''
|
| 423 |
st.rerun()
|
|
@@ -427,13 +426,14 @@ async def create_streamlit_interface():
|
|
| 427 |
uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp3', 'mp4'])
|
| 428 |
if uploaded_file:
|
| 429 |
file_path = os.path.join('./', uploaded_file.name)
|
| 430 |
-
|
|
|
|
| 431 |
st.success(f"Uploaded {uploaded_file.name}")
|
| 432 |
|
| 433 |
media_files = glob.glob("./*.png") + glob.glob("./*.jpg") + glob.glob("./*.mp3") + glob.glob("./*.mp4")
|
| 434 |
if media_files:
|
| 435 |
cols = st.columns(3)
|
| 436 |
-
media_votes =
|
| 437 |
for idx, media_file in enumerate(media_files):
|
| 438 |
vote_count = media_votes.get(media_file, 0)
|
| 439 |
if vote_count > 0:
|
|
@@ -441,24 +441,24 @@ async def create_streamlit_interface():
|
|
| 441 |
if media_file.endswith(('.png', '.jpg')):
|
| 442 |
st.image(media_file, use_container_width=True)
|
| 443 |
elif media_file.endswith('.mp3'):
|
| 444 |
-
st.markdown(
|
| 445 |
elif media_file.endswith('.mp4'):
|
| 446 |
-
st.markdown(
|
| 447 |
col1, col2 = st.columns(2)
|
| 448 |
with col1:
|
| 449 |
if st.button(f"π {vote_count}", key=f"media_vote_{idx}"):
|
| 450 |
comment = st.session_state.message_text
|
| 451 |
-
|
| 452 |
if st.session_state.pasted_image_data:
|
| 453 |
-
filename =
|
| 454 |
if filename:
|
| 455 |
-
|
| 456 |
st.session_state.pasted_image_data = None
|
| 457 |
st.session_state.message_text = ''
|
| 458 |
st.rerun()
|
| 459 |
with col2:
|
| 460 |
if st.button("ποΈ", key=f"media_delete_{idx}"):
|
| 461 |
-
|
| 462 |
st.rerun()
|
| 463 |
|
| 464 |
# Refresh timer
|
|
@@ -470,13 +470,13 @@ async def create_streamlit_interface():
|
|
| 470 |
font_name, font_func = random.choice(UNICODE_FONTS)
|
| 471 |
countdown_str = "".join(UNICODE_DIGITS[int(d)] for d in str(i)) if i < 10 else font_func(str(i))
|
| 472 |
timer_placeholder.markdown(f"<p class='timer'>β³ {font_func('Refresh in:')} {countdown_str}</p>", unsafe_allow_html=True)
|
| 473 |
-
|
| 474 |
st.rerun()
|
| 475 |
|
| 476 |
# Sidebar vote stats
|
| 477 |
st.sidebar.subheader("Vote Counts")
|
| 478 |
-
chat_votes =
|
| 479 |
-
media_votes =
|
| 480 |
for item, count in {**chat_votes, **media_votes}.items():
|
| 481 |
if count > 0:
|
| 482 |
st.sidebar.write(f"{item}: {count} votes")
|
|
@@ -487,10 +487,10 @@ async def main():
|
|
| 487 |
username = st.session_state.get('username', 'System π')
|
| 488 |
await asyncio.to_thread(log_action, username, action)
|
| 489 |
global NODE_NAME, server_task
|
| 490 |
-
NODE_NAME, port =
|
| 491 |
if server_task is None:
|
| 492 |
server_task = asyncio.create_task(run_websocket_server())
|
| 493 |
-
|
| 494 |
|
| 495 |
if __name__ == "__main__":
|
| 496 |
asyncio.run(main())
|
|
|
|
| 58 |
server_running = False
|
| 59 |
server_task = None
|
| 60 |
|
| 61 |
+
def get_node_name():
|
| 62 |
"""π - Naming Node with Code - Spins a name, oh so bold!"""
|
| 63 |
action = "π - Naming Node with Code - Spins a name, oh so bold!"
|
| 64 |
username = st.session_state.get('username', 'System π')
|
| 65 |
+
log_action(username, action)
|
| 66 |
parser = argparse.ArgumentParser(description='Start a chat node with a specific name')
|
| 67 |
parser.add_argument('--node-name', type=str, default=None)
|
| 68 |
parser.add_argument('--port', type=int, default=8501)
|
| 69 |
+
args = parser.parse_args()
|
| 70 |
return args.node_name or f"node-{uuid.uuid4().hex[:8]}", args.port
|
| 71 |
|
| 72 |
def log_action(username, action):
|
|
|
|
| 99 |
await asyncio.to_thread(lambda: open(CHAT_FILE, 'w').write(f"# {START_ROOM} Chat\n\nWelcome to the cosmic hub - start chatting! π€\n"))
|
| 100 |
with open(CHAT_FILE, 'r') as f:
|
| 101 |
content = await asyncio.to_thread(f.read)
|
| 102 |
+
return content
|
| 103 |
|
| 104 |
+
def get_user_list(chat_content):
|
| 105 |
"""π₯ - Crew Clue Brew - Spots whoβs who in the crew!"""
|
| 106 |
action = "π₯ - Crew Clue Brew - Spots whoβs who in the crew!"
|
| 107 |
username = st.session_state.get('username', 'System π')
|
| 108 |
+
log_action(username, action)
|
| 109 |
users = set()
|
| 110 |
for line in chat_content.split('\n'):
|
| 111 |
if line.strip() and ': ' in line:
|
|
|
|
| 269 |
server_running = True
|
| 270 |
await server.wait_closed()
|
| 271 |
|
| 272 |
+
def create_streamlit_interface():
|
| 273 |
"""π¨ - UI Brew Crew - Builds the view, so new!"""
|
| 274 |
action = "π¨ - UI Brew Crew - Builds the view, so new!"
|
| 275 |
username = st.session_state.get('username', 'System π')
|
| 276 |
+
log_action(username, action) # Sync call here since we're in Streamlit context
|
| 277 |
|
| 278 |
# Dynamic title based on username
|
| 279 |
if 'username' in st.session_state:
|
|
|
|
| 298 |
if 'client_id' not in st.session_state:
|
| 299 |
st.session_state.client_id = str(uuid.uuid4())
|
| 300 |
if 'username' not in st.session_state:
|
| 301 |
+
chat_content = asyncio.run(load_chat()) # Run async function synchronously
|
| 302 |
available_names = [name for name in FUN_USERNAMES if not any(f"{name} has joined" in line for line in chat_content.split('\n'))]
|
| 303 |
st.session_state.username = random.choice(available_names) if available_names else random.choice(FUN_USERNAMES)
|
| 304 |
|
|
|
|
| 308 |
if 'timer_start' not in st.session_state:
|
| 309 |
st.session_state.timer_start = time.time()
|
| 310 |
if 'quote_index' not in st.session_state:
|
| 311 |
+
quotes = asyncio.run(load_quotes("famous"))
|
| 312 |
st.session_state.quote_index = random.randint(0, max(0, len(quotes) - 1)) if quotes else 0
|
| 313 |
if 'quote_source' not in st.session_state:
|
| 314 |
st.session_state.quote_source = "famous"
|
|
|
|
| 349 |
# Handle pasted image data
|
| 350 |
pasted_image = st.session_state.get('pasted_image_data')
|
| 351 |
if pasted_image:
|
| 352 |
+
filename = asyncio.run(save_pasted_image(pasted_image))
|
| 353 |
if filename:
|
| 354 |
+
asyncio.run(save_chat_entry(st.session_state.username, f"Pasted image: {filename}"))
|
| 355 |
st.session_state.pasted_image_data = None
|
| 356 |
st.rerun()
|
| 357 |
|
| 358 |
# Chat section with quoting
|
| 359 |
st.subheader(f"{START_ROOM} Chat π¬")
|
| 360 |
+
chat_content = asyncio.run(load_chat())
|
| 361 |
chat_lines = chat_content.split('\n')
|
| 362 |
+
chat_votes = asyncio.run(load_votes(QUOTE_VOTES_FILE))
|
| 363 |
for i, line in enumerate(chat_lines):
|
| 364 |
if line.strip() and ': ' in line:
|
| 365 |
col1, col2, col3 = st.columns([4, 1, 1])
|
|
|
|
| 369 |
vote_count = chat_votes.get(line.split('. ')[1] if '. ' in line else line, 0)
|
| 370 |
if st.button(f"π {vote_count}", key=f"chat_vote_{i}"):
|
| 371 |
comment = st.session_state.message_text
|
| 372 |
+
asyncio.run(save_vote(QUOTE_VOTES_FILE, line.split('. ')[1] if '. ' in line else line, asyncio.run(generate_user_hash()), st.session_state.username, comment))
|
| 373 |
if st.session_state.pasted_image_data:
|
| 374 |
+
filename = asyncio.run(save_pasted_image(st.session_state.pasted_image_data))
|
| 375 |
if filename:
|
| 376 |
+
asyncio.run(save_chat_entry(st.session_state.username, f"Pasted image: {filename}"))
|
| 377 |
st.session_state.pasted_image_data = None
|
| 378 |
st.session_state.message_text = ''
|
| 379 |
st.rerun()
|
|
|
|
| 387 |
st.markdown(f"### Quoting: {st.session_state.quote_line}")
|
| 388 |
quote_response = st.text_area("Add your response", key="quote_response")
|
| 389 |
if st.button("Send Quote π", key="send_quote"):
|
| 390 |
+
def process_quote():
|
|
|
|
| 391 |
action = "π’ - Quote Float Boat - Echoes chat, we gloat!"
|
| 392 |
+
log_action(st.session_state.username, action)
|
| 393 |
markdown_response = f"### Quote Response\n- **Original**: {st.session_state.quote_line}\n- **{st.session_state.username} Replies**: {quote_response}"
|
| 394 |
if st.session_state.pasted_image_data:
|
| 395 |
+
filename = asyncio.run(save_pasted_image(st.session_state.pasted_image_data))
|
| 396 |
if filename:
|
| 397 |
markdown_response += f"\n- **Image**: "
|
| 398 |
st.session_state.pasted_image_data = None
|
| 399 |
+
asyncio.run(save_chat_entry(st.session_state.username, markdown_response))
|
| 400 |
+
process_quote()
|
| 401 |
del st.session_state.quote_line
|
| 402 |
st.session_state.message_text = ''
|
| 403 |
st.rerun()
|
|
|
|
| 405 |
# Username change dropdown
|
| 406 |
new_username = st.selectbox("Change Name", [""] + FUN_USERNAMES, index=0)
|
| 407 |
if new_username and new_username != st.session_state.username:
|
| 408 |
+
asyncio.run(save_chat_entry("System π", f"{st.session_state.username} changed name to {new_username}"))
|
| 409 |
st.session_state.username = new_username
|
| 410 |
st.rerun()
|
| 411 |
|
| 412 |
# Message input
|
| 413 |
message = st.text_input(f"Message as {st.session_state.username}", key="message_input", value=st.session_state.message_text, on_change=lambda: st.session_state.update(message_text=st.session_state.message_input))
|
| 414 |
if st.button("Send π", key="send_button") and message.strip():
|
| 415 |
+
asyncio.run(save_chat_entry(st.session_state.username, message))
|
| 416 |
if st.session_state.pasted_image_data:
|
| 417 |
+
filename = asyncio.run(save_pasted_image(st.session_state.pasted_image_data))
|
| 418 |
if filename:
|
| 419 |
+
asyncio.run(save_chat_entry(st.session_state.username, f"Pasted image: {filename}"))
|
| 420 |
st.session_state.pasted_image_data = None
|
| 421 |
st.session_state.message_text = ''
|
| 422 |
st.rerun()
|
|
|
|
| 426 |
uploaded_file = st.file_uploader("Upload Media", type=['png', 'jpg', 'mp3', 'mp4'])
|
| 427 |
if uploaded_file:
|
| 428 |
file_path = os.path.join('./', uploaded_file.name)
|
| 429 |
+
with open(file_path, 'wb') as f:
|
| 430 |
+
f.write(uploaded_file.getbuffer())
|
| 431 |
st.success(f"Uploaded {uploaded_file.name}")
|
| 432 |
|
| 433 |
media_files = glob.glob("./*.png") + glob.glob("./*.jpg") + glob.glob("./*.mp3") + glob.glob("./*.mp4")
|
| 434 |
if media_files:
|
| 435 |
cols = st.columns(3)
|
| 436 |
+
media_votes = asyncio.run(load_votes(MEDIA_VOTES_FILE))
|
| 437 |
for idx, media_file in enumerate(media_files):
|
| 438 |
vote_count = media_votes.get(media_file, 0)
|
| 439 |
if vote_count > 0:
|
|
|
|
| 441 |
if media_file.endswith(('.png', '.jpg')):
|
| 442 |
st.image(media_file, use_container_width=True)
|
| 443 |
elif media_file.endswith('.mp3'):
|
| 444 |
+
st.markdown(asyncio.run(get_audio_html(media_file)), unsafe_allow_html=True)
|
| 445 |
elif media_file.endswith('.mp4'):
|
| 446 |
+
st.markdown(asyncio.run(get_video_html(media_file)), unsafe_allow_html=True)
|
| 447 |
col1, col2 = st.columns(2)
|
| 448 |
with col1:
|
| 449 |
if st.button(f"π {vote_count}", key=f"media_vote_{idx}"):
|
| 450 |
comment = st.session_state.message_text
|
| 451 |
+
asyncio.run(save_vote(MEDIA_VOTES_FILE, media_file, asyncio.run(generate_user_hash()), st.session_state.username, comment))
|
| 452 |
if st.session_state.pasted_image_data:
|
| 453 |
+
filename = asyncio.run(save_pasted_image(st.session_state.pasted_image_data))
|
| 454 |
if filename:
|
| 455 |
+
asyncio.run(save_chat_entry(st.session_state.username, f"Pasted image: {filename}"))
|
| 456 |
st.session_state.pasted_image_data = None
|
| 457 |
st.session_state.message_text = ''
|
| 458 |
st.rerun()
|
| 459 |
with col2:
|
| 460 |
if st.button("ποΈ", key=f"media_delete_{idx}"):
|
| 461 |
+
os.remove(media_file)
|
| 462 |
st.rerun()
|
| 463 |
|
| 464 |
# Refresh timer
|
|
|
|
| 470 |
font_name, font_func = random.choice(UNICODE_FONTS)
|
| 471 |
countdown_str = "".join(UNICODE_DIGITS[int(d)] for d in str(i)) if i < 10 else font_func(str(i))
|
| 472 |
timer_placeholder.markdown(f"<p class='timer'>β³ {font_func('Refresh in:')} {countdown_str}</p>", unsafe_allow_html=True)
|
| 473 |
+
time.sleep(1) # Sync sleep here since it's UI-driven
|
| 474 |
st.rerun()
|
| 475 |
|
| 476 |
# Sidebar vote stats
|
| 477 |
st.sidebar.subheader("Vote Counts")
|
| 478 |
+
chat_votes = asyncio.run(load_votes(QUOTE_VOTES_FILE))
|
| 479 |
+
media_votes = asyncio.run(load_votes(MEDIA_VOTES_FILE))
|
| 480 |
for item, count in {**chat_votes, **media_votes}.items():
|
| 481 |
if count > 0:
|
| 482 |
st.sidebar.write(f"{item}: {count} votes")
|
|
|
|
| 487 |
username = st.session_state.get('username', 'System π')
|
| 488 |
await asyncio.to_thread(log_action, username, action)
|
| 489 |
global NODE_NAME, server_task
|
| 490 |
+
NODE_NAME, port = get_node_name() # Sync call since it's CLI parsing
|
| 491 |
if server_task is None:
|
| 492 |
server_task = asyncio.create_task(run_websocket_server())
|
| 493 |
+
create_streamlit_interface() # Call synchronously
|
| 494 |
|
| 495 |
if __name__ == "__main__":
|
| 496 |
asyncio.run(main())
|