Create app.py

app.py

@@ -0,0 +1,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()