File size: 2,181 Bytes
1e77d79 |
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 |
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()
|