Update pages.py

pages.py

@@ -239,51 +239,18 @@ def render_brainstorm_page():
             st.info("No products yet. Create one to get started!")
 
 
-def render_chat():
+# Update the render_chat function in pages.py
+def render_chat(chatbot_manager):
     st.header("💬 AI Business Mentor")
     
-    # Initialize chatbot manager
-    chatbot = ChatbotManager()
-    chatbot.initialize_chat()
-    
     # Sidebar options
-    st.sidebar.subheader("Chat Options")
-    if st.sidebar.button("Clear Chat History"):
-        chatbot.clear_chat()
-        st.rerun()
-    
-    # Display chat history
-    st.subheader("🤖 Conversation")
-    
-    # Chat container
-    chat_container = st.container()
-    
-    with chat_container:
-        # Display all messages
-        for message in chatbot.get_chat_history():
-            if message["role"] == "user":
-                with st.chat_message("user"):
-                    st.write(message["content"])
-            else:
-                with st.chat_message("assistant"):
-                    st.write(message["content"])
-    
-    # Chat input
-    user_input = st.chat_input("Ask me anything about business strategy, marketing, products, or operations...")
+    with st.sidebar:
+        if st.button("Clear Chat History"):
+            chatbot_manager.clear_chat()
+            st.rerun()
     
-    if user_input:
-        # Add user message
-        chatbot.add_message("user", user_input)
-        
-        # Generate response
-        with st.spinner("Thinking..."):
-            response = chatbot.generate_business_response(user_input)
-        
-        # Add assistant response
-        chatbot.add_message("assistant", response)
-        
-        # Rerun to update the display
-        st.rerun()
+    # Render the chat interface using the manager
+    chatbot_manager.render_chat_interface()
     
     # Additional helpful sections
     st.markdown("---")
@@ -293,23 +260,23 @@ def render_chat():
     
     with col1:
         if st.button("📊 Business Strategy"):
-            chatbot.add_message("user", "I need help with business strategy")
-            response = chatbot.generate_business_response("I need help with business strategy")
-            chatbot.add_message("assistant", response)
+            chatbot_manager.add_message("user", "I need help with business strategy")
+            response = chatbot_manager.generate_response("I need help with business strategy")
+            chatbot_manager.add_message("assistant", response)
             st.rerun()
     
     with col2:
         if st.button("📈 Marketing Tips"):
-            chatbot.add_message("user", "Give me marketing advice")
-            response = chatbot.generate_business_response("Give me marketing advice")
-            chatbot.add_message("assistant", response)
+            chatbot_manager.add_message("user", "Give me marketing advice")
+            response = chatbot_manager.generate_response("Give me marketing advice")
+            chatbot_manager.add_message("assistant", response)
             st.rerun()
     
     with col3:
         if st.button("💰 Financial Planning"):
-            chatbot.add_message("user", "Help with financial planning")
-            response = chatbot.generate_business_response("Help with financial planning")
-            chatbot.add_message("assistant", response)
+            chatbot_manager.add_message("user", "Help with financial planning")
+            response = chatbot_manager.generate_response("Help with financial planning")
+            chatbot_manager.add_message("assistant", response)
             st.rerun()
     
     # Optional: Keep the iframe as alternative