Update app.py

app.py

@@ -1463,77 +1463,72 @@ def update_message_usage(wix_user_id):
         st.error("Error updating message usage.")
 
 # Main interface
-if st.session_state.get("wix_user_id") and st.session_state.get("email"):
-    initialize_user_session(st.session_state["wix_user_id"], st.session_state["email"])
+initialize_user_session(st.session_state["wix_user_id"], st.session_state["email"])
 
-    retrieve_user_data(st.session_state["wix_user_id"])  # Fetch and display saved data for the user
-    display_saved_trustbuilders()
+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"])
-    welcome_placeholder = st.empty()
-    welcome_placeholder.info(f"**Hello, {user_name}!**")
-    time.sleep(3)
-    welcome_placeholder.empty()
+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()
 
 
 
-    # Input field for chatbot 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})
+# Input field for chatbot 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})
 
-            # 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)
+        # 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:
+            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:
+                    handle_save_trustbuilder_or_tonality(user_id, prompt, "trustbuilder")
+                elif save_as_tonality:
+                    handle_save_trustbuilder_or_tonality(user_id, prompt, "brandtonality")
+
+               
             else:
-                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:
-                        handle_save_trustbuilder_or_tonality(user_id, prompt, "trustbuilder")
-                    elif save_as_tonality:
-                        handle_save_trustbuilder_or_tonality(user_id, prompt, "brandtonality")
-
-                   
-                else:
-                    # Generate a response with AI for general queries
-                    with st.chat_message("user"):
-                        st.markdown(prompt)
-                    response_placeholder = st.empty()
+                # Generate a response with AI for general 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 = ""
+                try:
+                    output = agent_executor.invoke({
+                        "input": f"{prompt} Be in natural tone,doesn’t use flowery language and typical ai words .Sources should be accurate search on google for that.Avoid AI jargons.Give headings with each paragraph in content of article,blogs,writeup and newsletter keeping mentioned format.",
+                        "chat_history": st.session_state.chat_history
+                    })
+                    full_response = output["output"]
+                    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"):
-                            add_dot_typing_animation()
-                            display_typing_indicator()
-                    cleaned_text = ""
-                    try:
-                        output = agent_executor.invoke({
-                            "input": f"{prompt} Be in natural tone,doesn’t use flowery language and typical ai words .Sources should be accurate search on google for that.Avoid AI jargons.Give headings with each paragraph in content of article,blogs,writeup and newsletter keeping mentioned format.",
-                            "chat_history": st.session_state.chat_history
-                        })
-                        full_response = output["output"]
-                        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.")
+                            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)