Spaces:
Running
Running
import streamlit as st | |
import os | |
import random | |
import hashlib | |
import json | |
from datetime import datetime | |
from collections import Counter | |
import re | |
# Function to generate a short user hash | |
def generate_user_hash(): | |
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'] | |
# Function to load vote history from file | |
def load_vote_history(): | |
try: | |
with open('vote_history.json', 'r') as f: | |
return json.load(f) | |
except FileNotFoundError: | |
return {'images': {}, 'users': {}, 'words': {}} | |
# Function to save vote history to file | |
def save_vote_history(history): | |
with open('vote_history.json', 'w') as f: | |
json.dump(history, f) | |
# Function to update vote history in Markdown file | |
def update_history_md(image_name, user_hash): | |
with open('history.md', 'a') as f: | |
f.write(f"- {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - User {user_hash} voted for {image_name}\n") | |
# Function to extract words from file name | |
def extract_words(filename): | |
words = re.findall(r'\w+', filename.lower()) | |
return [word for word in words if len(word) > 2] | |
# Function to update word counts | |
def update_word_counts(history, image_name): | |
words = extract_words(image_name) | |
for word in words: | |
if word not in history['words']: | |
history['words'][word] = 0 | |
history['words'][word] += 1 | |
# Function to display images | |
def display_images(image_dir): | |
col1, col2 = st.columns(2) | |
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: | |
st.error("Not enough images in the directory.") | |
return | |
image1, image2 = random.sample(images, 2) | |
with col1: | |
st.image(os.path.join(image_dir, image1)) | |
if st.button(f"Upvote {image1}", key=f"upvote_{image1}"): | |
handle_vote(image1) | |
with col2: | |
st.image(os.path.join(image_dir, image2)) | |
if st.button(f"Upvote {image2}", key=f"upvote_{image2}"): | |
handle_vote(image2) | |
# Function to handle voting | |
def handle_vote(image_name): | |
user_hash = generate_user_hash() | |
vote_history = load_vote_history() | |
if image_name not in vote_history['images']: | |
vote_history['images'][image_name] = {'votes': 0, 'users': {}} | |
if user_hash not in vote_history['users']: | |
vote_history['users'][user_hash] = 0 | |
vote_history['images'][image_name]['votes'] += 1 | |
if user_hash not in vote_history['images'][image_name]['users']: | |
vote_history['images'][image_name]['users'][user_hash] = 0 | |
vote_history['images'][image_name]['users'][user_hash] += 1 | |
vote_history['users'][user_hash] += 1 | |
update_word_counts(vote_history, image_name) | |
save_vote_history(vote_history) | |
update_history_md(image_name, user_hash) | |
st.success(f"Upvoted {image_name}! Total votes: {vote_history['images'][image_name]['votes']}") | |
# Function to show vote history and user stats in sidebar | |
def show_vote_history(): | |
st.sidebar.title("Vote History") | |
vote_history = load_vote_history() | |
# Sort users by total votes | |
sorted_users = sorted(vote_history['users'].items(), key=lambda x: x[1], reverse=True) | |
st.sidebar.subheader("User Vote Totals") | |
for user_hash, votes in sorted_users: | |
st.sidebar.write(f"User {user_hash}: {votes} votes") | |
# Sort images by vote count | |
sorted_images = sorted(vote_history['images'].items(), key=lambda x: x[1]['votes'], reverse=True) | |
st.sidebar.subheader("Image Vote Totals") | |
for i, (image_name, data) in enumerate(sorted_images): | |
votes = data['votes'] | |
st.sidebar.write(f"{image_name}: {votes} votes") | |
# Display top 3 images | |
if i < 3: | |
st.sidebar.image(os.path.join('.', image_name), caption=f"#{i+1}: {image_name}", width=150) | |
# Display top 3 words | |
st.sidebar.subheader("Top 3 Words in Voted Images") | |
top_words = sorted(vote_history['words'].items(), key=lambda x: x[1], reverse=True)[:3] | |
for word, count in top_words: | |
st.sidebar.write(f"{word}: {count} occurrences") | |
# Main function | |
def main(): | |
st.title("Image Voting App") | |
# Set up the sidebar for vote history and user stats | |
show_vote_history() | |
# Display images for voting | |
image_dir = '.' # Current directory where the app is running | |
display_images(image_dir) | |
# Display user hash | |
st.sidebar.subheader("Your User ID") | |
st.sidebar.write(generate_user_hash()) | |
# Run the app | |
if __name__ == "__main__": | |
main() |