Update app.py

app.py

@@ -1,6 +1,6 @@
 import logging
 import os
-import streamlit as st
+import streamlit as st 
 from dotenv import load_dotenv
 import openai
 from langchain_openai import ChatOpenAI
@@ -14,7 +14,7 @@ from langchain_core.messages import AIMessage, HumanMessage
 from langchain_community.document_loaders import TextLoader
 from langchain_text_splitters import CharacterTextSplitter
 import serpapi
-import requests
+import requests 
 import streamlit.components.v1 as components
 import smtplib
 from email.mime.multipart import MIMEMultipart
@@ -27,14 +27,126 @@ import smtplib
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
 from markdownify import markdownify
+import chargebee
+import pyrebase
+import streamlit.components.v1 as components
+import time
+import warnings
+from streamlit.components.v1 import html
+from langchain.docstore.document import Document  
+import firebase_admin
+import uuid
+import json
+import io
+from firebase_admin import credentials, firestore
+import base64
+from pdfminer.high_level import extract_text  # Import for PDF text extraction
+
+st.set_page_config(layout="wide")
+import logging
+import asyncio
+import re 
+# Set up logging to suppress Streamlit warnings about experimental functions
+logging.getLogger('streamlit').setLevel(logging.ERROR)
+
+
+
 
 # Initialize logging and load environment variables
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 load_dotenv()
 
+chargebee.configure(site="mextconsulting", api_key="live_dBLXn8yG5dFcuIgU5Szebj2KfTcdt4hjpf")
+
+# Firebase Configuration
+firebase_config = {
+    "apiKey": "AIzaSyAWiaqrduoG7fzmJxBVnVg9nCC4EoEnwfY",
+    "authDomain": "trustai-3e7a2.firebaseapp.com",
+    "databaseURL": "https://trustai-3e7a2-default-rtdb.firebaseio.com",
+    "projectId": "trustai-3e7a2",
+    "storageBucket": "trustai-3e7a2.appspot.com",
+    "messagingSenderId": "964339831031",
+    "appId": "1:964339831031:web:66d21ceea68ab03f1043f2",
+   "measurementId": "G-ZMLZQZMHK2"
+}
+# Initialize Firebase
+firebase = pyrebase.initialize_app(firebase_config)
+db = firebase.database()
+storage = firebase.storage()
+
+
+def convert_pdf_to_md(file):
+    """
+    Convert a PDF file to Markdown.
+    """
+    try:
+        text = extract_text(file)
+        return f"# PDF Document\n\n{text}"
+    except Exception as e:
+        logger.error(f"Error converting PDF to MD: {e}")
+        return ""
+
+def convert_docx_to_md(file):
+    """
+    Convert a Word (DOCX) file to Markdown.
+    """
+    try:
+        doc = Document(file)
+        full_text = '\n'.join([para.text for para in doc.paragraphs])
+        return f"# Word Document\n\n{full_text}"
+    except Exception as e:
+        logger.error(f"Error converting DOCX to MD: {e}")
+        return ""
+
+def convert_txt_to_md(file):
+    """
+    Convert a TXT file to Markdown.
+    """
+    try:
+        text = file.read().decode("utf-8")
+        return f"# Text Document\n\n{text}"
+    except Exception as e:
+        logger.error(f"Error converting TXT to MD: {e}")
+        return ""
 
+def convert_file_to_md(file):
+    """
+    Detect file type and convert to Markdown accordingly.
+    """
+    if file.type == "application/pdf":
+        return convert_pdf_to_md(file)
+    elif file.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
+        return convert_docx_to_md(file)
+    elif file.type == "text/plain":
+        return convert_txt_to_md(file)
+    else:
+        st.sidebar.warning(f"Unsupported file type: {file.type}")
+        return ""
+
+def merge_markdown_contents(contents):
+    """
+    Merge multiple Markdown contents into a single Markdown string.
+    """
+    merged_content = "\n\n---\n\n".join(contents)
+    return merged_content
 
+def upload_to_firebase(user_id, merged_md_content):
+    """
+    Upload the merged Markdown content to Firebase Storage.
+    Returns the public URL of the uploaded file.
+    """
+    try:
+        file_path = f"knowledge_bases/{user_id}/knowledge_base.md"
+        # Upload the content as bytes
+        storage.child(file_path).put(io.BytesIO(merged_md_content.encode('utf-8')), content_type="text/markdown")
+        # Get the download URL
+        public_url = storage.child(file_path).get_url(None)
+        return public_url
+    except Exception as e:
+        logger.error(f"Error uploading to Firebase Storage: {e}")
+        st.sidebar.error(f"Error uploading files: {e}")
+        return None
 
 # Define and validate API keys
 openai_api_key = os.getenv("OPENAI_API_KEY")
@@ -66,7 +178,11 @@ if 'used_trust_tips' not in st.session_state:
     st.session_state.used_trust_tips = set()
 if 'used_suggestions' not in st.session_state:
     st.session_state.used_suggestions = set()
-    
+
+
+backend_url = "https://backend-web-05122eab4e09.herokuapp.com"
+# Suppress Streamlit deprecation and warning messages
+
 def copy_to_clipboard(text):
     """Creates a button to copy text to clipboard."""
     escaped_text = text.replace('\n', '\\n').replace('"', '\\"')
@@ -124,8 +240,7 @@ def copy_to_clipboard(text):
     </script>
     """
     components.html(copy_icon_html, height=60)
-
-
+ 
 def send_feedback_via_email(name, email, feedback):
     """Sends an email with feedback details."""
     smtp_server = 'smtp.office365.com'  
@@ -203,7 +318,7 @@ def clean_text(text):
     return cleaned_text
     
 
-
+ 
   
 
 def get_trust_tip_and_suggestion():
@@ -213,15 +328,152 @@ def get_trust_tip_and_suggestion():
 
 
 
+def load_main_data_source():
+    try:
+        with open("./data_source/time_to_rethink_trust_book.md", "r") as f:
+            main_content = f.read()
+        
+        # Split main content into chunks
+        text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
+        main_texts = text_splitter.split_text(main_content)
 
+        # Create Document objects for the main data source
+        main_documents = [Document(page_content=text) for text in main_texts]
+        return main_documents
+    except Exception as e:
+        st.error(f"Error loading main data source: {e}")
+        return []
 
-def side():        
-    with st.sidebar.form(key='feedback_form'):
+def combine_data_sources():
+    main_data_source = load_main_data_source()
+    user_data_source = load_user_data_source(st.session_state["wix_user_id"])
+    
+    trustbuilders = [Document(page_content=tb["message"]) for tb in st.session_state.get("trustbuilders", {}).values()]
+    brand_tonalities = [Document(page_content=bt["message"]) for bt in st.session_state.get("brand_tonality", {}).values()]
     
-        st.image( "Trust Logic_Wheel_RGB_Standard.png")
+    return main_data_source + user_data_source + trustbuilders + brand_tonalities
+
+
+def refresh_faiss_index():
+    combined_sources = combine_data_sources()
+    if combined_sources:
+        embeddings = OpenAIEmbeddings()
+        db_faiss = FAISS.from_documents(combined_sources, embeddings)
+        st.session_state["faiss_db"] = db_faiss
+
+def load_user_data_source(user_id):
+    """
+    Load user-uploaded data source from Firebase.
+    """
+    try:
+        file_path = f"knowledge_bases/{user_id}/knowledge_base.md"
+        storage_url = storage.child(file_path).get_url(None)
+        response = requests.get(storage_url)
+
+        if response.status_code == 200:
+            user_content = response.text
+            st.session_state["user_data_source"] = user_content
+            return [Document(page_content=user_content)]
+        else:
+            return []
+    except Exception as e:
+        st.error(f"Error loading user data source: {e}")
+        return []
+
+
+
+
+def store_brand_tonality(user_id, message):
+    try:
+        tonality_id = str(uuid.uuid4())
+        db.child("users").child(user_id).child("brandTonality").child(tonality_id).set({"message": message})
+        
+        # Update st.session_state for instant display in sidebar
+        if "brand_tonality" not in st.session_state:
+            st.session_state["brand_tonality"] = {}
+        st.session_state["brand_tonality"][tonality_id] = {"message": message}
+        display_save_confirmation("Brand Tonality")
+    except Exception as e:
+        st.error(f"Error saving Brand Tonality: {e}")
+
+def store_trustbuilder(user_id, message):
+    try:
+        trustbuilder_id = str(uuid.uuid4())
+        db.child("users").child(user_id).child("trustBuilders").child(trustbuilder_id).set({"message": message})
+        
+        # Update st.session_state for instant display in sidebar
+        if "trustbuilders" not in st.session_state:
+            st.session_state["trustbuilders"] = {}
+        st.session_state["trustbuilders"][trustbuilder_id] = {"message": message}
+        display_save_confirmation("TrustBuilder")
+    except Exception as e:
+        st.error(f"Error saving TrustBuilder: {e}")
+
+def load_user_content(user_id):
+    """Load TrustBuilders and Brand Tonalities from Firebase."""
+    try:
+        trustbuilders = db.child("users").child(user_id).child("trustBuilders").get()
+        brand_tonalities = db.child("users").child(user_id).child("brandTonality").get()
+        st.session_state["TrustBuilders"] = [item.val()["content"] for item in trustbuilders.each()] if trustbuilders.each() else []
+        st.session_state["BrandTonality"] = [item.val()["content"] for item in brand_tonalities.each()] if brand_tonalities.each() else []
+    except Exception as e:
+        st.error(f"Error loading saved content: {e}")
+        
+def save_content(user_id, content, content_type):
+    try:
+        entry_id = str(uuid.uuid4())
+        db.child("users").child(user_id).child(content_type).child(entry_id).set({"message": content})
+        
+        # Update session state directly after saving to avoid duplicates
+        st.session_state[content_type.lower()][entry_id] = {"message": content}
+        display_save_confirmation(content_type)
+    except Exception as e:
+        logger.error(f"Error saving {content_type}: {e}")
+        st.error(f"Could not save {content_type}. Please try again.")
+        
+def display_saved_trustbuilders():
+    """
+    Display saved TrustBuilders and Brand Tonality entries with download and delete options.
+    """
+    st.sidebar.subheader("Saved TrustBuilders")
+    trustbuilders = st.session_state.get("trustbuilders", {})
+    for trustbuilder_id, trustbuilder_data in trustbuilders.items():
+        st.sidebar.markdown(f"- {trustbuilder_data['message'][:50]}...")
+        st.sidebar.markdown(download_link(trustbuilder_data["message"], f"TrustBuilder_{trustbuilder_id}.md"), unsafe_allow_html=True)
+        if st.sidebar.button("❌", key=f"del_tb_{trustbuilder_id}"):
+            delete_entry("trustBuilders", trustbuilder_id, "trustbuilders")
+            st.rerun() 
+
+    st.sidebar.subheader("Saved Brand Tonality")
+    brand_tonality = st.session_state.get("brand_tonality", {})
+    for tonality_id, tonality_data in brand_tonality.items():
+        st.sidebar.markdown(f"- {tonality_data['message'][:50]}...")
+        st.sidebar.markdown(download_link(tonality_data["message"], f"BrandTonality_{tonality_id}.md"), unsafe_allow_html=True)
+        if st.sidebar.button("❌", key=f"del_bt_{tonality_id}"):
+            delete_entry("brandTonality", tonality_id, "brand_tonality")
+            st.rerun()
+
+
+def download_link(content, filename):
+    """
+    Create a download link for content.
+    """
+    b64 = base64.b64encode(content.encode()).decode()
+    return f'<a href="data:text/plain;base64,{b64}" download="{filename}">Download</a>'
+
+
+
+def side():
+    with st.sidebar:
+        st.image("Trust Logic_Wheel_RGB_Standard.png")
         st.header("Let's create something great.")
-        st.markdown("Our minds assess trust through Six Buckets of Trust® and determine their importance and order in a given situation. We then evaluate why we can or can’t trust someone in these Buckets. Trustifier.ai®, trained on 20 years of TrustLogic® application, helps you identify reasons why your audience can trust you in each Bucket and create trust-optimised solutions. It’s copy AI with substance.")
-        st.markdown("""
+
+        with st.sidebar.expander("Explore", expanded=False):
+                st.markdown(
+                    "Our minds assess trust through Six Buckets of Trust® and determine their importance and order in a given situation. We then evaluate why we can or can’t trust someone in these Buckets. Trustifier.ai®, trained on 20 years of TrustLogic® application, helps you identify reasons why your audience can trust you in each Bucket and create trust-optimised solutions. It’s copy AI with substance."
+                )
+                st.markdown(
+                    """
                     <style>
                         .stability { color: rgb(7, 55, 99); font-size: 24px; font-weight: bold; }
                         .development { color: rgb(241, 194, 50); font-size: 24px; font-weight: bold; }
@@ -230,76 +482,80 @@ def side():
                         .vision { color: rgb(255, 153, 0); font-size: 24px; font-weight: bold; }
                         .competence { color: rgb(111, 168, 220); font-size: 24px; font-weight: bold; }
                     </style>
-    
+        
                     <h3 class="stability">Stability Trust:</h3>
                     <p>Why can I trust you to have built a strong and stable foundation?</p>
-    
+        
                     <h3 class="development">Development Trust:</h3>
                     <p>Why can I trust you to develop well in the future?</p>
-    
+        
                     <h3 class="relationship">Relationship Trust:</h3>
                     <p>What appealing relationship qualities can I trust you for?</p>
-    
+        
                     <h3 class="benefit">Benefit Trust:</h3>
                     <p>What benefits can I trust you for?</p>
-    
+        
                     <h3 class="vision">Vision Trust:</h3>
                     <p>What Vision and Values can I trust you for?</p>
-    
+        
                     <h3 class="competence">Competence Trust:</h3>
                     <p>What competencies can I trust you for?</p>
-                    """, unsafe_allow_html=True)
-        st.markdown("For detailed descriptions, visit [Academy](https://www.trustifier.ai/account/academy)")
-    
-        
-    
-    
-    
-        feedback_name = st.text_input("Name")
-        feedback_email_input = st.text_input("Email")
-        feedback_text = st.text_area("Feedback")
-
-        # Submit button within the form
-        submit_button = st.form_submit_button("Submit Feedback")
-        if submit_button:
-            if feedback_name and feedback_email_input and feedback_text:
-                with st.spinner('Sending email'):
-                    send_feedback_via_email(feedback_name, feedback_email_input, feedback_text)
-                    st.success("Thank you for your feedback!")
-            else:
-                st.error("Please fill in all fields.")
+                    """, unsafe_allow_html=True
+                )
+                st.markdown("For detailed descriptions, visit [Academy](https://www.trustifier.ai/account/academy)")
 
+        st.header("Upload Data source")
+        
+        with st.form(key="upload_form"):
+            uploaded_files = st.file_uploader("Upload PDF, DOCX, or TXT files", type=["pdf", "docx", "txt"], accept_multiple_files=True)
+            submit_button = st.form_submit_button("Upload")
+            
+            if submit_button and uploaded_files:
+                md_contents = [convert_file_to_md(file) for file in uploaded_files if convert_file_to_md(file)]
+                if md_contents:
+                    merged_md = merge_markdown_contents(md_contents)
+                    upload_url = upload_to_firebase(st.session_state["wix_user_id"], merged_md)
+                    if upload_url:
+                        st.success("Files uploaded and processed successfully!")
+                        st.markdown(f"[View your uploaded knowledge base here]({upload_url})", unsafe_allow_html=True)
+                        refresh_faiss_index()
+        # Save Custom Content section
+        st.header("Save Custom Content")
+        custom_content = st.text_area("Enter content to save")
+        
+        if st.button("Save as Brand Tonality"):
+            save_content(st.session_state["wix_user_id"], custom_content, "BrandTonality")
+            display_save_confirmation("Brand Tonality")
+        if st.button("Save as TrustBuilder"):
+            save_content(st.session_state["wix_user_id"], custom_content, "TrustBuilder")
+            display_save_confirmation("TrustBuilder")
 
 
 
 side()
 # Load knowledge base
-def load_knowledge_base():
-    try:
-        loader = TextLoader("./data_source/time_to_rethink_trust_book.md")
-        documents = loader.load()
-        text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
-        docs = text_splitter.split_documents(documents)
-        return docs
-    except Exception as e:
-        logger.error(f"Error loading knowledge base: {e}")
-        raise e
 
-knowledge_base = load_knowledge_base()
 
-# Initialize embeddings and FAISS index
-embeddings = OpenAIEmbeddings()
-db = FAISS.from_documents(knowledge_base, embeddings)
+# Initialize embeddings and FAISS index only when wix_user_id is set
+
+if st.session_state.get("wix_user_id") and "faiss_db" not in st.session_state:
+    refresh_faiss_index()
+
 
 # Define search functions
 def search_knowledge_base(query):
-    try:
-        output = db.similarity_search(query)
-        return output
-    except Exception as e:
-        logger.error(f"Error searching knowledge base: {e}")
-        return ["Error occurred during knowledge base search"]
-    
+    """
+    Searches the FAISS index using the provided query.
+    Returns the most relevant documents based on the query.
+    """
+    if "faiss_db" not in st.session_state:
+        st.error("FAISS database is not initialized.")
+        return []
+
+    # Retrieve the top 5 most relevant documents
+    retrieved_docs = st.session_state["faiss_db"].similarity_search(query, k=5)
+    return retrieved_docs
+
 def google_search(query):
     try:
         search_client = serpapi.Client(api_key=serper_api_key)
@@ -347,6 +603,7 @@ tools = [knowledge_base_tool, google_search_tool]
 
 prompt_message = f"""
 **You are an expert copywriter specializing in creating highly creative marketing content in natural tone that integrates **Trust Builders** into various content formats for any organization. Your goal is to produce compelling, factual,and well-structured material that is concise and based on the knowledgebase, adhering to the following guidelines.The content should be written in an active voice, avoiding third-person perspective .Interconnect the trust-building elements contextually and creatively to enhance the reading flow, making it clear what the impact is on audience. While prompts may mention terms like Stability, Development, Competence, Relationship, Benefit, or Vision, do not include words like "trust","cutting-edge","beacon,"beacon of hope"and "realm"  in any of the output. Trust word can be used like here are these eg Development trust builders but in the copy .**
+
 ### MANDATORY VERIFICATION CHECKLIST:
 Before submitting **any content**, ensure that each piece includes:
 1. **Specific Details**:
@@ -363,11 +620,13 @@ Before submitting **any content**, ensure that each piece includes:
      - "This [specific benefit] for [specific audience]"
      - **Example**: "This reduces wait times by 47% for patients seeking emergency care."
 3. *Give [sources] next to each trust building point and heuristics and creative techniques with every copy application*.    
+
 *SOURCE LINK REQUIREMENTS*
 1. **Each source link must**:
    - Be **working and clickable**.
    - Be **no older than 2 years** unless historically significant.
 2. Refer knowledge base for description, guiding principles, question to consider and examples for relevant trustbucket then *search on google* and then give relevant trustbuilders.
+
 ##SPECIFICITY ENFORCEMENT
 Replace vague phrases with specific details:
 - ❌ "many" → ✅ exact number.
@@ -377,6 +636,7 @@ Replace vague phrases with specific details:
 - ��� "industry leader" → ✅ "ranked #1 in customer satisfaction by J.D. Power in 2023".
 - ❌ "significant impact" → ✅ "47% reduction in processing time".
 - ❌ "team of experts" → ✅ "157 certified professionals led by Dr. Sarah Chen".
+
 ###ERROR CHECKING
 Before submitting, verify and fix:
 **When query is to find trust builders never mention heuristic and creative techniques**
@@ -387,6 +647,7 @@ Before submitting, verify and fix:
 5. **No missing specific details** from the mandatory checklist.
 6. **No prohibited terms included in the copy. For searching it can be used.**.
 7. ** DONOT INCLUDE heuristic and  creative techniques in Trustbuilder queries**
+
 ###CRITICAL MANDATORY INSTRUCTIONS
 - **Avoid Prohibited Terms**:
   - Do **not** mention "trust," "trust buckets," or any category names like "Development," "Stability," "Competence," "Relationship," "Vision" in the final copy.
@@ -407,7 +668,9 @@ Before submitting, verify and fix:
      -Limit to 3-5 items in each category.
     Note: When including heuristics and creative techniques, use the  single line structure “Heuristics: examples” and “Creative Techniques: examples” with no extra details.
 - If the query is about trust builders , do not include heuristics or creative techniques,List of TrustBuilders Used.
+
 ### CONTENT TYPES AND FORMATS
+
 #### 1. Annual report article / article 
 - **Introduction**: Start with "Here is a draft of your [Annual Report/Article]. Feel free to suggest further refinements."
 - **Structure**:
@@ -428,6 +691,7 @@ Before submitting, verify and fix:
 - **Important Notes**:
   - **Strictly search and provide accurate source links always**.
   - Include heuristics and creative techniques if requested.
+
 #### 2. Social Media Posts
 - **Introduction Line**: Start with "Here is a draft of your social media post. Feel free to suggest further refinements."
 - **Content**:
@@ -481,6 +745,7 @@ Before submitting, verify and fix:
      -Limit to 3-5 items in each category.
 Note: When including heuristics and creative techniques, use the structure “Heuristics: examples” and “Creative Techniques: examples” with no extra details.
 - **Word Count**: Follow any specified word count for the main body. Do not count sub-heading sections in the word count limit.
+
 ### 5.Trust-Based Queries:**
  Be over specific with numbers,names,dollars, programs ,awards and action. 
 - When a query seeks a specific number of trust builders (e.g., "5 trust builders"), the AI should:
@@ -502,9 +767,11 @@ Note: When including heuristics and creative techniques, use the structure “He
           - **People** (5 points)
           - **Offers/Services** (5 points)
       - **Important Specificity:** Always include **names**, **numbers** (e.g., $ amounts and years), **programs**, **strategies**, **places**, **awards**, and **actions** by searching on google to add credibility and depth to the content. Ensure that trust-building points are detailed and specific.
+
 - **For Specific Categories:**
   - When a query asks for a specific category (e.g., "Development trust builders"), find 15 trust-building points that are specific with relevant names, numbers like $ amounts and years, programs, strategies, places, awards, and actions specifically for that category.
   - Categorize these points into Organization, People, and Offers/Services (with 5 points for each category).
+
 - **Format:**
   - **Introduction Line:** Start with "Here are TrustBuilders® for [Selected Categories] at [Organization Name]. Let me know if you want to refine the results or find more."
   - **Categories:**
@@ -528,6 +795,7 @@ Note: When including heuristics and creative techniques, use the structure “He
       - [Trust-Building Point 15] - [Source](#)
   - Ensure each selected category contains 15 trust-building points, categorized as specified.
   - Provide bullet points under each section with relevant accurate source link.
+
 **Important Notes:**
   - Strictly search and provide accurate source links always.
   - **No Subheadings or Labels:** Under each main category, list the trust-building points directly as bullet points or numbered lists **without any additional subheadings, labels, descriptors, phrases, or words before the points**.
@@ -544,18 +812,17 @@ Note: When including heuristics and creative techniques, use the structure “He
 - **Example of Correct Format**:
   **Organization**
   - In **2023**, World Vision invested **$150 million** in sustainable agriculture programs across **35 countries**, impacting over **2 million** farmers.This improves food security for vulnerable communities.- [Source](#)
+
 ### 6. LinkedIn Profile
 - If requested, generate a LinkedIn profile in a professional manner.
 - **Avoid prohibited terms** and **flowery language**.
+
 ### GENERAL QUERIES
 - Donot use knowledge base for non-trust content.Give quick output according to above given guidance.
 - **Audience Impact**: Always clarify what the information means to  the audience.
 
-"""
-
-
-
 
+"""
 
 prompt_template = ChatPromptTemplate.from_messages([
     ("system", prompt_message),
@@ -565,7 +832,7 @@ prompt_template = ChatPromptTemplate.from_messages([
 ])
 
 # Create Langchain Agent
-llm = ChatOpenAI(model="gpt-4o", temperature=0.5)
+llm = ChatOpenAI(model="gpt-4o", temperature=0.6)
 llm_with_tools = llm.bind_tools(tools)
 
 # Define the agent pipeline
@@ -684,7 +951,69 @@ if not st.session_state.get("chat_started", False):
     </div>
     <div style="height: 50px;"></div> <!-- Adds a gap of 50px after the section containers -->
     """, unsafe_allow_html=True)
-  
+    
+hide_specific_warning = """
+<script>
+document.addEventListener('DOMContentLoaded', function() {
+    const alerts = window.parent.document.querySelectorAll('div[data-testid="stAlert"]');
+    alerts.forEach(function(alert) {
+        if (alert.innerText.includes('Please replace st.experimental_get_query_params with st.query_params')) {
+            alert.style.display = 'none';          // Hide the warning
+            alert.style.visibility = 'hidden';     // Make it invisible
+            alert.style.height = '0px';            // Set height to zero to remove space
+            alert.style.margin = '0px';            // Set margin to zero
+            alert.style.padding = '0px';           // Set padding to zero
+        }
+    });
+});
+</script>
+"""
+
+
+# Embed the JavaScript in your Streamlit app
+components.html(hide_specific_warning, height=0, scrolling=False)
+
+query_params = st.experimental_get_query_params()
+wix_user_id = query_params.get('wix_user_id', [None])[0]
+email = query_params.get('email', [None])[0]
+
+# Session state to track user login and message usage
+if "wix_user_id" not in st.session_state:
+    st.session_state["wix_user_id"] = wix_user_id
+if "email" not in st.session_state:
+    st.session_state["email"] = email
+if "message_limit" not in st.session_state:
+    st.session_state["message_limit"] = 0
+if "used_messages" not in st.session_state:
+    st.session_state["used_messages"] = 0
+
+
+def receive_wix_message():
+    components.html(
+        """
+        <script>
+        window.addEventListener('message', function(event) {
+            const data = event.data;
+            if (data.wixUserId && data.email) {
+                window.parent.postMessage({
+                    'wix_user_id': data.wixUserId,
+                    'email': data.email
+                }, "*");
+
+                // Send message back to Streamlit
+                window.parent.postMessage({
+                    wix_user_id: data.wixUserId,
+                    email: data.email
+                }, "*");
+            }
+        });
+        </script>
+        """,
+        height=0
+    )
+
+# Calling  this function to initialize listening for Wix messages
+receive_wix_message()
 
 trust_tips = [
     "What I don’t know I can’t trust you for. Make sure you know all your great TrustBuilders® and use them over time.",
@@ -742,6 +1071,7 @@ def add_dot_typing_animation():
             display: flex;
             align-items: center;
         }
+
         .dot {
             height: 10px;
             width: 10px;
@@ -751,12 +1081,15 @@ def add_dot_typing_animation():
             display: inline-block;
             animation: dot-blink 1.5s infinite ease-in-out;
         }
+
         .dot:nth-child(2) {
             animation-delay: 0.2s;
         }
+
         .dot:nth-child(3) {
             animation-delay: 0.4s;
         }
+
         @keyframes dot-blink {
             0% {
                 opacity: 0.3;
@@ -782,76 +1115,210 @@ def display_typing_indicator():
         <span class="dot"></span>
     </div>
     """
+ 
     st.markdown(dot_typing_html, unsafe_allow_html=True)
+def display_save_confirmation(type_saved):
+    st.info(f"Content successfully saved as **{type_saved}**!")
 
-prompt = st.chat_input("")
-if prompt :
-    
-        st.session_state["chat_started"] = True
-        # Add user message to chat history
-        st.session_state.chat_history.append({"role": "user", "content": prompt})
-        
 
-        # Display user message
-        with st.chat_message("user"):
-            st.markdown(prompt)
-        response_placeholder = st.empty()
-    
-        # Generate AI response
-        with response_placeholder:
-            with st.chat_message("assistant"):
-                add_dot_typing_animation()  # Adds the CSS for the typing animation
-                display_typing_indicator()  
-            
+def extract_name(email):
+    return email.split('@')[0].capitalize()
+
+if "trustbuilders" not in st.session_state:
+    st.session_state["trustbuilders"] = {}
+if "brand_tonality" not in st.session_state:
+    st.session_state["brand_tonality"] = {}
+
+# Load saved entries upon user login
+def retrieve_user_data(user_id):
+    try:
+        trustbuilders = db.child("users").child(user_id).child("trustBuilders").get().val() or {}
+        brand_tonality = db.child("users").child(user_id).child("brandTonality").get().val() or {}
         
-        full_response = ""
-            
-        try:
-                # Generate response using the agent executor
-                output = agent_executor.invoke({
+        # Set retrieved data into session state
+        st.session_state["trustbuilders"] = trustbuilders
+        st.session_state["brand_tonality"] = brand_tonality
+    except Exception as e:
+        st.error("Error loading saved entries. Please try again later.")
 
-                                "input": f" {prompt} Be in natural tone.Sources should be accurate.Avoid AI jargons.DONOT USE use words that end with '-ing' for headlines and subheadlines when asked to write a article,blog,newsletter.",
-    
-    
-                    "chat_history": st.session_state.chat_history
-                })
-                full_response = output["output"]
-                #full_response= replace_terms(full_response)
-
-                cleaned_text = clean_text(full_response)
-                cleaned_text = re.sub(r'<[^>]*>', '', cleaned_text)
-
-
-                
-                #cleaned_text = re.sub(r'</span>', '', cleaned_text) 
-                #cleaned_text = re.sub(r'<span[^>]*>', '', cleaned_text)
-                # Display the response
-                
-                trust_tip, suggestion = get_trust_tip_and_suggestion()
-
-                #combined_text = f"{cleaned_text}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
-                combined_text = f"{cleaned_text}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
-
-                #st.markdown(combined_text,unsafe_allow_html=True)
-                
-                #seprtor= st.markdown("---")  # Add a separator
-                #t_tip= st.markdown(f"**Trust Tip**: {trust_tip}")
-                #s_sug- st.markdown(f"**Suggestion**: {suggestion}")
-                with response_placeholder:
-                    with st.chat_message("assistant"):
-                        st.markdown(combined_text, unsafe_allow_html=True)
 
+
+
+
+
+
+# Function to handle queries about saved TrustBuilders or Brand Tonality
+def handle_memory_queries(prompt):
+    prompt = prompt.lower().strip()
     
+    # Save as TrustBuilder, removing the phrase from the content
+    if "save this as trustbuilder" in prompt:
+        content_to_save = prompt.replace("save this as trustbuilder", "").strip()
+        if content_to_save:  # Only save if there's actual content
+            store_trustbuilder(st.session_state["wix_user_id"], content_to_save)
+            return "TrustBuilder saved successfully!"
+        else:
+            return "Please provide content to save as TrustBuilder."
+
+    # Save as Brand Tonality, removing the phrase from the content
+    elif "save this as brand tonality" in prompt:
+        content_to_save = prompt.replace("save this as brand tonality", "").strip()
+        if content_to_save:  # Only save if there's actual content
+            store_brand_tonality(st.session_state["wix_user_id"], content_to_save)
+            return "Brand Tonality saved successfully!"
+        else:
+            return "Please provide content to save as Brand Tonality."
+
+    # Retrieve saved TrustBuilders
+    elif "show my saved trustbuilders" in prompt or "find my saved trustbuilders" in prompt:
+        if st.session_state.get("trustbuilders"):
+            return "\n".join([f"- {entry['message']}" for entry in st.session_state["trustbuilders"].values()])
+        else:
+            return "You haven't saved any TrustBuilders yet."
+
+    # Retrieve saved Brand Tonality
+    elif "show my saved brand tonality" in prompt or "find my saved brand tonality" in prompt:
+        if st.session_state.get("brand_tonality"):
+            return "\n".join([f"- {entry['message']}" for entry in st.session_state["brand_tonality"].values()])
+        else:
+            return "You haven't saved any Brand Tonality entries yet."
     
-    
-    
-    
-        except Exception as e:
-                logger.error(f"Error generating response: {e}")
-                full_response = "I apologize, but an error occurred while generating the response. Please try again."
-                st.write(full_response)
-    
-        # Add AI response to chat history 
+    return None  # If no recognized command, proceed with general handling
+
+
+def delete_entry(category, entry_id, session_key):
+    try:
+        user_id = st.session_state["wix_user_id"]
+        db.child("users").child(user_id).child(category).child(entry_id).remove()
+        del st.session_state[session_key][entry_id]  # Remove from session
+        st.experimental_rerun()  # Refresh sidebar
+    except Exception as e:
+        st.error(f"Error deleting entry: {e}")
+
+# Function to download TrustBuilder as a .md file
+def download_trustbuilder_as_md(content, trustbuilder_id):
+    b64_content = base64.b64encode(content.encode()).decode()
+    download_link = f'<a href="data:text/markdown;base64,{b64_content}" download="TrustBuilder_{trustbuilder_id}.md">Download</a>'
+    st.sidebar.markdown(download_link, unsafe_allow_html=True)
+
+
+
+
+
+                        
+# Function to update the message counter in a static location
+message_counter_placeholder = st.sidebar.empty()
+
+def update_message_counter():
+    remaining_messages = st.session_state["message_limit"] - st.session_state["used_messages"]
+    message_counter_placeholder.markdown(f"**Messages Left**: {remaining_messages} / {st.session_state['message_limit']}")
+
+if st.session_state.get("wix_user_id") and st.session_state.get("email"):
+    retrieve_user_data(st.session_state["wix_user_id"])
+    display_saved_trustbuilders()
+
+    user_name = extract_name(st.session_state["email"])
+    welcome_placeholder = st.empty()
+    welcome_placeholder.info(f"**Hello, {user_name}!**")
+    time.sleep(3)
+    welcome_placeholder.empty()
+    def check_user_subscription(wix_user_id):
+        response = requests.get(f"{backend_url}/check-subscription/{wix_user_id}")
+        if response.status_code == 200:
+            data = response.json()
+            st.session_state["message_limit"] = data.get("message_limit", 0)
+            st.session_state["used_messages"] = data.get("used_messages", 0)
+            update_message_counter()
+        else:
+            st.error("Error fetching user subscription details.")
+
+    def update_message_usage(wix_user_id):
+        response = requests.post(f"{backend_url}/update-message-usage/{st.session_state['wix_user_id']}")
+        if response.status_code == 200:
+            st.session_state["used_messages"] += 1
+            update_message_counter()
+        else:
+            st.error("Error updating message usage.")
+    check_user_subscription(st.session_state["wix_user_id"])
+
+    # Input for AI interaction
+    prompt = st.chat_input("")
+    global combined_text
+    if st.session_state["used_messages"] < st.session_state["message_limit"]:
+        if prompt:
+            st.session_state.chat_started = True
+            st.session_state.chat_history.append({"role": "user", "content": prompt})
+
+            # Check if the prompt is a memory query
+            memory_response = handle_memory_queries(prompt)
+            if memory_response:
+                with st.chat_message("assistant"):
+                    st.markdown(memory_response)
 
-        st.session_state.chat_history.append({"role": "assistant", "content": cleaned_text})
-        copy_to_clipboard(cleaned_text) 
+            else:
+                # Detect if the user wants to save content based on chatbox prompts
+                save_as_trustbuilder = re.search(r"\b(save|add|store)\s*(this)?\s*(as)?\s*(trust\s*builder|trustbuilder)\b", prompt, re.IGNORECASE)
+                save_as_tonality = re.search(r"\b(save|add|store)\s*(this)?\s*(as)?\s*(brand\s*tonality|tonality)\b", prompt, re.IGNORECASE)
+
+
+                if save_as_trustbuilder or save_as_tonality:
+                    user_id = st.session_state["wix_user_id"]
+                    if save_as_trustbuilder:
+                        store_trustbuilder(user_id, prompt)
+                        display_save_confirmation("TrustBuilder")
+                    elif save_as_tonality:
+                        store_brand_tonality(user_id, prompt)
+                        display_save_confirmation("Brand Tonality")
+
+                else:
+                    # Generate a response with AI for other types of queries
+                    with st.chat_message("user"):
+                        st.markdown(prompt)
+                    response_placeholder = st.empty()
+                    with response_placeholder:
+                        with st.chat_message("assistant"):
+                            add_dot_typing_animation()
+                            display_typing_indicator()
+                    cleaned_text = ""
+                    # Specialized responses if keywords detected
+                    try:
+                        #if "trustbuilder" in prompt.lower() or "trust" in prompt.lower():
+                        output = agent_executor.invoke({
+                             "input": f"{prompt}Be in natural tone.Sources should be accurate.Avoid AI jargons.DONOT USE use words that end with '-ing' for headlines and subheadlines when asked to write a article,blog,newsletter.",
+                             "chat_history": st.session_state.chat_history
+                        })
+                        full_response = output["output"]
+                        #full_response= replace_terms(full_response)
+        
+                        cleaned_text = clean_text(full_response)
+                        trust_tip, suggestion = get_trust_tip_and_suggestion()
+                        combined_text = f"{cleaned_text}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
+                        with response_placeholder:
+                            with st.chat_message("assistant"):
+                                st.markdown(combined_text, unsafe_allow_html=True)
+
+                        #else: 
+                        #    llm = ChatOpenAI(model="gpt-4o", temperature=0.8)
+                        #    output = llm.invoke(prompt)
+                        #    full_response = output["output"]
+                            #full_response= replace_terms(full_response)
+            
+                        #    cleaned_text = clean_text(full_response)
+                         #   trust_tip, suggestion = get_trust_tip_and_suggestion()
+                         #   combined_text = f"{cleaned_text}\n\n---\n\n**Trust Tip**: {trust_tip}\n\n**Suggestion**: {suggestion}"
+                         #   with response_placeholder:
+                        #        with st.chat_message("assistant"):
+                        #           st.markdown(combined_text, unsafe_allow_html=True)
+
+                        update_message_usage(st.session_state["wix_user_id"])
+
+                    except Exception as e:
+                        logger.error(f"Error generating response: {e}")
+                        st.error("An error occurred while generating the response. Please try again.")
+                    
+                    st.session_state.chat_history.append({"role": "assistant", "content": cleaned_text})
+                    copy_to_clipboard(cleaned_text) 
+    else:
+        st.warning("You have reached your message limit. Please upgrade your plan to continue.")
+else:
+    st.warning("Please log in to access the chatbot features.")