File size: 3,594 Bytes
324b6d5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
import streamlit as st
import os
import random
import hashlib
import json
from datetime import datetime
# Function to generate a short user hash
def generate_user_hash():
if 'user_hash' not in st.session_state:
# Generate a unique identifier for the session
session_id = str(random.getrandbits(128))
# Create a hash of the session ID
hash_object = hashlib.md5(session_id.encode())
st.session_state['user_hash'] = hash_object.hexdigest()[:8] # Use first 8 characters of the hash
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 {}
# 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 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:
vote_history[image_name] = {'votes': 0, 'users': []}
if user_hash not in vote_history[image_name]['users']:
vote_history[image_name]['votes'] += 1
vote_history[image_name]['users'].append(user_hash)
save_vote_history(vote_history)
update_history_md(image_name, user_hash)
st.success(f"Upvoted {image_name}! Total votes: {vote_history[image_name]['votes']}")
else:
st.warning("You've already voted for this image.")
# Function to show vote history in sidebar
def show_vote_history():
st.sidebar.title("Vote History")
vote_history = load_vote_history()
# Sort images by vote count in descending order
sorted_images = sorted(vote_history.items(), key=lambda x: x[1]['votes'], reverse=True)
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)
# Main function
def main():
st.title("Image Voting App")
# Set up the sidebar for vote history
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.write(f"Your User ID: {generate_user_hash()}")
# Run the app
if __name__ == "__main__":
main() |