Update app.py

app.py

@@ -12,19 +12,20 @@ def main():
     st.header("Meme Template Gallery")
     meme_templates = [f for f in os.listdir("meme_templates") if f.endswith(('.png', '.jpg', '.jpeg'))]
     
-    # Create a 4-column layout
-    cols = st.columns(4)
-    
     # Use session state to store the selected template
     if 'selected_template' not in st.session_state:
         st.session_state.selected_template = None
 
+    # Create a 4-column layout
+    cols = st.columns(4)
+    
     for idx, template in enumerate(meme_templates):
         col = cols[idx % 4]
         image_path = os.path.join("meme_templates", template)
         with Image.open(image_path) as img:
-            if col.image(img, caption=template, use_column_width=True):
-                st.session_state.selected_template = template
+            col.image(img, use_column_width=True)
+        if col.button(f"Select {template}", key=f"btn_{template}"):
+            st.session_state.selected_template = template
 
     # Display selected template and generate meme
     if st.session_state.selected_template: