Update app.py

app.py

@@ -1207,34 +1207,33 @@ def display_save_confirmation(type_saved):
 # Function to update the message counter in a static location
 message_counter_placeholder = st.sidebar.empty()
 
-def initialize_message_limit(user_id):
-    user_data = db.child("users").child(user_id).get().val()
-    if user_data:
-        st.session_state["message_limit"] = user_data.get("message_limit", INITIAL_MESSAGE_LIMIT)
-        st.session_state["used_messages"] = user_data.get("used_messages", 0)
-    else:
-        # Set default limits for new users and store them in Firebase
-        st.session_state["message_limit"] = INITIAL_MESSAGE_LIMIT
-        st.session_state["used_messages"] = 0
-        db.child("users").child(user_id).update({
-            "message_limit": INITIAL_MESSAGE_LIMIT,
-            "used_messages": 0
-        })
-
-def update_message_usage(user_id):
-    st.session_state["used_messages"] += 1
-    db.child("users").child(user_id).update({
-        "used_messages": st.session_state["used_messages"]
-    })
-    update_message_counter()
+
 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']}")
 
-    
+def initialize_user_data(wix_user_id):
+    response = requests.get(f"{backend_url}/check-user/{wix_user_id}")
+    if response.status_code == 200:
+        data = response.json()
+        st.session_state["message_limit"] = data.get("message_limit", 100)  # Default limit is 100
+        st.session_state["used_messages"] = data.get("used_messages", 0)
+        update_message_counter()
+    else:
+        st.error("Error fetching user data.")
+
+# Update message usage in Firebase after each interaction
+def update_message_usage(wix_user_id):
+    response = requests.post(f"{backend_url}/update-message-usage/{wix_user_id}")
+    if response.status_code == 200:
+        st.session_state["used_messages"] += 1
+        update_message_counter()
+    else:
+        st.error("Error updating message usage.")
+
+# Main interface
 if st.session_state.get("wix_user_id") and st.session_state.get("email"):
-    initialize_message_limit(st.session_state["wix_user_id"])
-    retrieve_user_data(st.session_state["wix_user_id"])
+    retrieve_user_data(st.session_state["wix_user_id"])  # Fetch and display saved data for the user
     display_saved_trustbuilders()
 
     user_name = extract_name(st.session_state["email"])
@@ -1242,11 +1241,11 @@ if st.session_state.get("wix_user_id") and st.session_state.get("email"):
     welcome_placeholder.info(f"**Hello, {user_name}!**")
     time.sleep(3)
     welcome_placeholder.empty()
-  
-    
-    update_message_counter()
 
-    # Input for AI interaction
+    # Initialize the user's message limit and usage from Firebase
+    initialize_user_data(st.session_state["wix_user_id"])
+
+    # Input field for chatbot interaction
     prompt = st.chat_input("")
     global combined_text
     if st.session_state["used_messages"] < st.session_state["message_limit"]:
@@ -1254,18 +1253,15 @@ if st.session_state.get("wix_user_id") and st.session_state.get("email"):
             st.session_state.chat_started = True
             st.session_state.chat_history.append({"role": "user", "content": prompt})
 
-            # Check if the prompt is a memory query
+            # Handle save commands based on the user prompt
             memory_response = handle_memory_queries(prompt)
             if memory_response:
                 with st.chat_message("assistant"):
                     st.markdown(memory_response)
-
             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:
@@ -1274,9 +1270,8 @@ if st.session_state.get("wix_user_id") and st.session_state.get("email"):
                     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
+                    # Generate a response with AI for general queries
                     with st.chat_message("user"):
                         st.markdown(prompt)
                     response_placeholder = st.empty()
@@ -1285,16 +1280,12 @@ if st.session_state.get("wix_user_id") and st.session_state.get("email"):
                             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,doesn’t use flowery language and typical ai words .Sources should be accurate.Avoid AI jargons.",
-                             "chat_history": st.session_state.chat_history
+                            "input": f"{prompt} Be in natural tone, avoid AI jargon, and provide accurate sources.",
+                            "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}"
@@ -1302,19 +1293,6 @@ if st.session_state.get("wix_user_id") and st.session_state.get("email"):
                             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:
@@ -1322,8 +1300,8 @@ if st.session_state.get("wix_user_id") and st.session_state.get("email"):
                         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) 
+                    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.")
+    st.warning("Please log in to access the chatbot features.")