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()