Update app.py

app.py

@@ -820,23 +820,20 @@ def side():
         st.write("Choose the Trust Bucket® by clicking on an image:")
         st.markdown('<div class="image-container">', unsafe_allow_html=True)
         for bucket, img_path in image_files:
-            with st.container():
-                try:
-                    # Display clickable image as a button
-                    if st.button(f"", key=f"filter_{bucket}", help=f"Filter by {bucket}"):
-                        search_query = bucket
-                    st.markdown(
-                        f"""
-                        <div class="image-wrapper">
-                            <button>
-                                <img src="{img_path}" alt="{bucket}">
-                            </button>
-                        </div>
-                        """,
-                        unsafe_allow_html=True,
-                    )
-                except FileNotFoundError:
-                    st.warning(f"Image not found: {img_path}")
+            try:
+                # Display clickable image as a button
+                if st.button(label="", key=f"filter_{bucket}", help=f"Filter by {bucket}"):
+                    search_query = bucket
+                st.markdown(
+                    f"""
+                    <div class="image-wrapper">
+                        <img src="{img_path}" alt="{bucket}" title="{bucket}">
+                    </div>
+                    """,
+                    unsafe_allow_html=True,
+                )
+            except FileNotFoundError:
+                st.warning(f"Image not found: {img_path}")
         st.markdown("</div>", unsafe_allow_html=True)
         
         # Save TrustBuilders Section
@@ -848,25 +845,23 @@ def side():
             st.write("Choose the Trust Bucket® by clicking on an image to save:")
             st.markdown('<div class="image-container">', unsafe_allow_html=True)
             for bucket, img_path in image_files:
-                with st.container():
-                    try:
-                        # Display clickable image for saving
-                        if st.button(f"", key=f"save_{bucket}", help=f"Save to {bucket}"):
-                            save_trustbuilder(user_id, trust_builder_text.strip(), bucket)
-                        st.markdown(
-                            f"""
-                            <div class="image-wrapper">
-                                <button>
-                                    <img src="{img_path}" alt="{bucket}">
-                                </button>
-                            </div>
-                            """,
-                            unsafe_allow_html=True,
-                        )
-                    except FileNotFoundError:
-                        st.warning(f"Image not found: {img_path}")
+                try:
+                    # Display clickable image for saving
+                    if st.button(label="", key=f"save_{bucket}", help=f"Save to {bucket}"):
+                        save_trustbuilder(user_id, trust_builder_text.strip(), bucket)
+                    st.markdown(
+                        f"""
+                        <div class="image-wrapper">
+                            <img src="{img_path}" alt="{bucket}" title="{bucket}">
+                        </div>
+                        """,
+                        unsafe_allow_html=True,
+                    )
+                except FileNotFoundError:
+                    st.warning(f"Image not found: {img_path}")
             st.markdown("</div>", unsafe_allow_html=True)
-
+        else:
+            st.warning("Please enter a Trust Builder text to save.")