Update app.py

app.py

@@ -6,6 +6,7 @@ from pages import (
     render_brainstorm_page,
     render_chat
 )
+from chatbot import ChatbotManager  # Import the new ChatbotManager
 
 
 def main():
@@ -16,9 +17,24 @@ def main():
         initial_sidebar_state="expanded"
     )
 
-    # Create a selection box to choose between pages
-    page = st.sidebar.radio("Select a page", ["Home", "Dashboard", "Analytics", "Brainstorm", "Chat"])
+    # Initialize the chatbot manager (will load the model)
+    chatbot_manager = ChatbotManager()
 
+    # Sidebar configuration
+    with st.sidebar:
+        st.image("https://via.placeholder.com/150x50?text=Prospira", width=150)
+        st.title("Navigation")
+        
+        # Create a selection box to choose between pages
+        page = st.radio("Select a page", 
+                       ["Home", "Dashboard", "Analytics", "Brainstorm", "Chat"],
+                       label_visibility="collapsed")
+        
+        st.markdown("---")
+        st.caption(f"Model: Blenderbot-400M")
+        st.caption(f"Running on: {'GPU' if torch.cuda.is_available() else 'CPU'}")
+
+    # Page routing
     if page == "Home":
         render_home()
     elif page == "Dashboard":
@@ -28,8 +44,11 @@ def main():
     elif page == "Brainstorm":
         render_brainstorm_page()
     elif page == "Chat":
-        render_chat()
+        # Pass the chatbot manager to the chat page
+        render_chat(chatbot_manager)
 
 
 if __name__ == "__main__":
+    # Add torch import at runtime if not using it elsewhere
+    import torch
     main()