Update app.py

app.py

@@ -801,41 +801,14 @@ def side():
                 st.write("No TrustBuilders found.")
         st.markdown("</div>", unsafe_allow_html=True)
         
-      st.write("Choose the Trust Bucket® by clicking an image:")
-    st.markdown('<div class="image-container">', unsafe_allow_html=True)
-    for bucket, img_path in image_files:
-        try:
-            if os.path.exists(img_path):
-                # Display clickable image as a button
-                if st.button(label="", key=f"filter_{bucket}", help=f"Filter by {bucket}"):
-                    st.write(f"You clicked on {bucket} bucket!")
-                st.markdown(
-                    f"""
-                    <div class="image-wrapper">
-                        <img src="{img_path}" alt="{bucket}" title="{bucket}">
-                    </div>
-                    """,
-                    unsafe_allow_html=True,
-                )
-            else:
-                st.warning(f"Image not found: {img_path}")
-        except Exception as e:
-            st.error(f"Error loading image {img_path}: {e}")
-    st.markdown("</div>", unsafe_allow_html=True)
-        
-        # Save TrustBuilders Section
-        st.subheader("Save TrustBuilders®")
-        brand = st.text_input("Brand/Product/Person", key="brand_input_save")
-        trust_builder_text = st.text_area("Type/paste Trust Builder®", height=100)
-        
-        if trust_builder_text.strip():
-            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:
-                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.write("Choose the Trust Bucket® by clicking an image:")
+        st.markdown('<div class="image-container">', unsafe_allow_html=True)
+        for bucket, img_path in image_files:
+            try:
+                if os.path.exists(img_path):
+                    # Display clickable image as a button
+                    if st.button(label="", key=f"filter_{bucket}", help=f"Filter by {bucket}"):
+                        st.write(f"You clicked on {bucket} bucket!")
                     st.markdown(
                         f"""
                         <div class="image-wrapper">
@@ -844,11 +817,38 @@ def side():
                         """,
                         unsafe_allow_html=True,
                     )
-                except FileNotFoundError:
+                else:
                     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.")
+            except Exception as e:
+                st.error(f"Error loading image {img_path}: {e}")
+        st.markdown("</div>", unsafe_allow_html=True)
+            
+            # Save TrustBuilders Section
+            st.subheader("Save TrustBuilders®")
+            brand = st.text_input("Brand/Product/Person", key="brand_input_save")
+            trust_builder_text = st.text_area("Type/paste Trust Builder®", height=100)
+            
+            if trust_builder_text.strip():
+                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:
+                    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.")