import streamlit as st
import os
import random

# 🖼️ 1. Display two columns, each with a random image from the directory
def display_images(image_dir):
    """🖼️ Function 1: Displays two random images side by side for voting."""
    col1, col2 = st.columns(2)

    # 🗂️ Load random images from directory
    images = os.listdir(image_dir)
    
    # 📸 Randomly select two images
    image1 = random.choice(images)
    image2 = random.choice(images)
    
    with col1:
        st.image(os.path.join(image_dir, image1))
        if st.button(f"Upvote {image1}"):
            handle_vote(image1)
    
    with col2:
        st.image(os.path.join(image_dir, image2))
        if st.button(f"Upvote {image2}"):
            handle_vote(image2)

# 🎯 2. Handle the voting logic and store the history
def handle_vote(image_name):
    """🎯 Function 2: Handles voting and stores vote history."""
    # ✅ Save the upvote to session state or database
    if 'vote_history' not in st.session_state:
        st.session_state['vote_history'] = {}
    
    if image_name in st.session_state['vote_history']:
        st.session_state['vote_history'][image_name] += 1
    else:
        st.session_state['vote_history'][image_name] = 1

    st.success(f"Upvoted {image_name}! Total votes: {st.session_state['vote_history'][image_name]}")

# 📊 3. Sidebar to show vote history
def show_vote_history():
    """📊 Function 3: Displays the vote history in the sidebar."""
    st.sidebar.title("Vote History")
    if 'vote_history' in st.session_state:
        for image_name, votes in st.session_state['vote_history'].items():
            st.sidebar.write(f"{image_name}: {votes} votes")
    else:
        st.sidebar.write("No votes yet!")

# ⚙️ 4. Main function to structure app execution
def main():
    """⚙️ Function 4: Main function to run the app."""
    st.title("Image Voting App")
    
    # 🎨 Set up the sidebar for vote history
    show_vote_history()
    
    # 🖼️ Display images for voting
    image_dir = './images'  # Set your directory path with images
    display_images(image_dir)

# 🚀 5. Run the app
if __name__ == "__main__":
    main()