Update app.py

app.py

@@ -546,66 +546,6 @@ def delete_document(user_id, doc_id):
     except Exception as e:
         st.error(f"Error deleting document: {e}")
 
-def display_popup():
-    # Custom CSS for the pop-up message
-    popup_css = """
-    <style>
-    .custom-popup {
-        background-color: #f9f9f9;
-        color: #333;
-        padding: 15px;
-        border-radius: 8px;
-        font-family: Arial, sans-serif;
-        font-size: 14px;
-        margin-top: 10px;
-        box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);
-    }
-    .custom-popup code {
-        background-color: #e1e1e1;
-        padding: 2px 4px;
-        border-radius: 4px;
-        color: #333;
-        font-family: monospace;
-    }
-    .custom-popup b {
-        color: #333;
-    }
-    .custom-popup i {
-        color: #333;
-    }
-    </style>
-    """
-    # Inject custom CSS
-    st.markdown(popup_css, unsafe_allow_html=True)
-    # Display the pop-up message
-    st.markdown("""
-    <div class="custom-popup">
-        <b>Here’s how you can save your TrustBuilder® using Trustifier AI:</b><br><br>
-        1. Type your TrustBuilder® in the chat.<br><br>
-        2. If you're unsure of the TrustBucket®, ask the AI:<br>
-        <i>"Hey, which TrustBucket does this TrustBuilder® belong to?"</i><br><br>
-        3. Save it using the following format:<br>
-        <code>Save this as a TrustBuilder. [BucketName]. [TrustBuilder Text]</code><br><br>
-        <b>Example:</b><br>
-        <code>Save this as a TrustBuilder. Stability. We focus on keeping and nurturing our team.</code>
-    </div>
-
-
-    """, unsafe_allow_html=True)
-
-# Initialize session state for button click
-if 'letai_allocate_clicked' not in st.session_state:
-    st.session_state['letai_allocate_clicked'] = False
-
-
-
-
-
-
-
-
-
-
 
 
 
@@ -902,58 +842,80 @@ def side():
                             st.warning("Please fill all fields")
     
         with col2:
-                custom_button_css = """
+            custom_css = """
+            <style>
+            button {
+                background-color: rgb(82, 129, 134) !important;
+                color: white !important;
+                border: none !important;
+                padding: 8px 16px !important;
+                font-size: 14px !important;
+                border-radius: 5px !important;
+                cursor: pointer !important;
+                font-family: Arial, sans-serif !important;
+            }
+            button:hover {
+                background-color: rgb(70, 115, 119) !important;
+            }
+            </style>
+            """
+            st.markdown(custom_css, unsafe_allow_html=True)
+            letai_allocate_clicked = st.button("LetAI Allocate", key="letai_allocate_button")
+        
+            if letai_allocate_clicked:
+                # Display the pop-up message
+                popup_css = """
                 <style>
-                .custom-button {
+                .custom-popup {
                     background-color: rgb(82, 129, 134);
                     color: white;
-                    border: none;
-                    padding: 8px 16px;
-                    font-size: 14px;
-                    border-radius: 5px;
-                    cursor: pointer;
+                    padding: 15px;
+                    border-radius: 8px;
                     font-family: Arial, sans-serif;
+                    font-size: 14px;
+                    margin-top: 10px;
                 }
-                .custom-button:hover {
-                    background-color: rgb(70, 115, 119);
+                .custom-popup code {
+                    background-color: rgba(255, 255, 255, 0.2);
+                    padding: 2px 4px;
+                    border-radius: 4px;
+                    color: white;
+                    font-family: monospace;
+                }
+                .custom-popup b {
+                    color: #FFD700; /* Gold color for emphasis */
+                }
+                .custom-popup i {
+                    color: #FFFFFF;
                 }
                 </style>
                 """
-                st.markdown(custom_button_css, unsafe_allow_html=True)
-            
-                # Create a unique ID for the button
-                import uuid
-                button_id = 'custom_button_' + str(uuid.uuid4()).replace('-', '')
-            
-                # HTML and JavaScript for the custom button
-                custom_button_html = f"""
-                <button class="custom-button" id="{button_id}">LetAI Allocate</button>
-                <script>
-                const btn = document.getElementById("{button_id}");
-                btn.addEventListener("click", function() {{
-                    // Send a message to Streamlit to set a session state variable
-                    fetch('/_stcore/streamlit/report/session_state', {{
-                        method: 'POST',
-                        headers: {{
-                            'Content-Type': 'application/json'
-                        }},
-                        body: JSON.stringify({{ 'letai_allocate_clicked': true }})
-                    }}).then((response) => {{
-                        window.location.reload();
-                    }});
-                }});
-                </script>
-                """
-                st.markdown(custom_button_html, unsafe_allow_html=True)
-            
-            # Check if the custom button was clicked
-        if st.session_state['letai_allocate_clicked']:
-                display_popup()
-                # Reset the click state
-                st.session_state['letai_allocate_clicked'] = False
+                st.markdown(popup_css, unsafe_allow_html=True)
+                st.markdown("""
+                <div class="custom-popup">
+                    <b>Here’s how you can save your TrustBuilder® using Trustifier AI:</b><br><br>
+                    1. Type your TrustBuilder® in the chat.<br><br>
+                    2. If you're unsure of the TrustBucket®, ask the AI:<br>
+                    <i>"Hey, which TrustBucket does this TrustBuilder® belong to?"</i><br><br>
+                    3. Save it using the following format:<br>
+                    <code>Save this as a TrustBuilder. [BucketName]. [TrustBuilder Text]</code><br><br>
+                    <b>Example:</b><br>
+                    <code>Save this as a TrustBuilder. Stability. We focus on keeping and nurturing our team.</code>
+                </div>
+                """, unsafe_allow_html=True)
+        
+        
+        
+        
+        
+        
+        
+
+
+
+
+
             
-                
-side()
 if st.session_state.get("wix_user_id") and "faiss_db" not in st.session_state:
     refresh_faiss_index()