added

app.py

@@ -1,39 +1,56 @@
 import streamlit as st
-import os, json
-
-IMG_DIR = "images"
-VOTE_FILE = "votes.json"
-
-# Ensure votes.json exists with image names
-if not os.path.exists(VOTE_FILE):
-    votes = {}
-
-if os.path.exists(VOTE_FILE):
-    with open(VOTE_FILE, 'r') as f:
-        votes = json.load(f)
-
-# Initialize votes for new images
-for img in os.listdir(IMG_DIR):
-    if img not in votes:
-        votes[img] = 0
-
-# Save updated votes.json
-with open(VOTE_FILE, 'w') as f:
-    json.dump(votes, f, indent=2)
-
-st.title("Vote for the Best Logo")
-
-# Display images and vote buttons
-for img in sorted(votes.keys()):
-    st.image(os.path.join(IMG_DIR, img), width=200)
-    if st.button(f"Vote for {img}"):
-        votes[img] += 1
-        with open(VOTE_FILE, 'w') as f:
-            json.dump(votes, f, indent=2)
-        st.success(f"Voted for {img}")
-        st.experimental_rerun()
-
-# Display current vote tally
-st.header("Live Results")
-for img, count in sorted(votes.items(), key=lambda x: -x[1]):
-    st.markdown(f"**{img}**: {count} votes")
+from PIL import Image
+import os
+
+# Title of the app
+st.title("Logo Voter App")
+
+# Path to the images folder
+images_folder = "images"
+
+# Check if the images folder exists
+if not os.path.exists(images_folder):
+    st.error("The 'images' folder is missing!")
+    st.stop()
+
+# Get a list of all image files in the folder
+image_files = [f for f in os.listdir(images_folder) if f.endswith(('png', 'jpg', 'jpeg'))]
+
+# If no images found, show a message
+if not image_files:
+    st.error("No images found in the 'images' folder.")
+    st.stop()
+
+# Display images in a grid layout
+st.write("Uploaded Images:")
+
+# Define the number of columns for the grid
+columns = st.columns(3)  # 3 columns for layout
+
+# Initialize a dictionary to store votes for each image
+votes = {image_file: 0 for image_file in image_files}
+
+# Loop through the images and display them in the columns
+for idx, image_file in enumerate(image_files):
+    with columns[idx % 3]:  # Loop through columns
+        # Open the image
+        img_path = os.path.join(images_folder, image_file)
+        img = Image.open(img_path)
+        
+        # Display the image
+        st.image(img, use_column_width=True, caption=image_file)
+        
+        # Voting buttons
+        col1, col2 = st.columns([1, 1])
+        with col1:
+            if st.button(f"Like {image_file}", key=f"like_{idx}"):
+                votes[image_file] += 1
+        with col2:
+            if st.button(f"Dislike {image_file}", key=f"dislike_{idx}"):
+                votes[image_file] -= 1
+
+# Show voting results
+st.write("### Voting Results:")
+for image_file, vote_count in votes.items():
+    st.write(f"**{image_file}:** {vote_count} votes")
+

requirements.txt

@@ -1 +1,2 @@
-streamlit
+streamlit
+pillow