Create app.py

app.py

@@ -0,0 +1,169 @@
+import streamlit as st
+import os
+import random
+import hashlib
+import json
+from datetime import datetime
+from collections import Counter
+import re
+import time
+
+# 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']}")
+    
+    # Reset the timer when a vote is cast
+    st.session_state.last_interaction = time.time()
+
+# 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")
+
+# Function to add refresh rate slider
+def add_refresh_rate_slider():
+    st.sidebar.subheader("Refresh Settings")
+    refresh_rate = st.sidebar.slider("Refresh rate (seconds)", 0, 120, 30, 
+                                     help="Set to 0 for no auto-refresh")
+    return refresh_rate
+
+# Main function
+def main():
+    st.title("Image Voting App")
+    
+    # Initialize last interaction time if not set
+    if 'last_interaction' not in st.session_state:
+        st.session_state.last_interaction = time.time()
+    
+    # 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())
+    
+    # Add refresh rate slider
+    refresh_rate = add_refresh_rate_slider()
+    
+    # Check if it's time to refresh
+    if refresh_rate > 0:
+        time_since_last_interaction = time.time() - st.session_state.last_interaction
+        if time_since_last_interaction >= refresh_rate:
+            st.session_state.last_interaction = time.time()
+            st.rerun()
+    
+    # Update last interaction time when the page is interacted with
+    if st.button("Refresh Now"):
+        st.session_state.last_interaction = time.time()
+        st.rerun()
+
+# Run the app
+if __name__ == "__main__":
+    main()