Update app.py

app.py

@@ -1,9 +1,88 @@
-app.py
-src/
-  __init__.py
-  utils.py
-  auth.py
-  admin.py
-  chat.py
-  mis.py
-  video_export.py
+# app.py
+import os
+import streamlit as st
+
+st.set_page_config(page_title="Olist Chatbot", layout="wide", page_icon="🛍️")
+
+# Local imports
+from src.utils import get_connection
+from src.auth import (
+    init_auth_schema, bootstrap_admin_from_env, render_login_signup_gate,
+    render_force_change_password, session_active, session_countdown_widget
+)
+from src.chat import render_chat_tab, ensure_chat_schema
+from src.mis import render_mis_admin, render_mis_user
+from src.admin import render_admin_create_user, render_admin_manage_users
+from src.video_export import render_admin_export_video_tab, ensure_video_deps
+
+def main():
+    # Initialize DB schemas
+    init_auth_schema()
+    ensure_chat_schema()
+    ensure_video_deps()  # validates optional deps, fonts etc.
+
+    # Gate by login/session
+    if not session_active():
+        if not render_login_signup_gate():
+            st.stop()
+        if st.session_state.get("_force_change_pw"):
+            if not render_force_change_password():
+                st.stop()
+
+    if st.session_state.get("_force_change_pw"):
+        if not render_force_change_password():
+            st.stop()
+
+    # Sidebar: user info + logout + session timer
+    user = st.session_state["user"]
+    st.sidebar.markdown(f"**Logged in as:** {user['username']}  \n**Role:** {user['role']}")
+    session_countdown_widget()
+    if st.sidebar.button("Logout"):
+        st.session_state.pop("user", None)
+        st.rerun()
+
+    role = user["role"]
+
+    if role == "admin":
+        # Admin: NO Chat tab
+        tab_mis, tab_export_video, tab_create_user, tab_manage_users = st.tabs(
+            ["📊 MIS (Admin)", "📥 Export + 🎬 Video", "👤 Create User", "🔧 Manage Users"]
+        )
+
+        with tab_mis:
+            with get_connection() as conn:
+                render_mis_admin(conn)
+
+        with tab_export_video:
+            with get_connection() as conn:
+                render_admin_export_video_tab(conn)
+
+        with tab_create_user:
+            render_admin_create_user()
+
+        with tab_manage_users:
+            render_admin_manage_users()
+
+    else:
+        # User: Chat, MIS (user), Export (no video)
+        tab_chat, tab_mis_user, tab_export = st.tabs(["💬 Chat", "📊 MIS", "📥 Export"])
+
+        with tab_chat:
+            with get_connection() as conn:
+                render_chat_tab(conn)
+
+        with tab_mis_user:
+            with get_connection() as conn:
+                render_mis_user(conn, user_id=user["id"])
+
+        with tab_export:
+            # simple CSV/TXT export of this user's chats (no video here)
+            from src.chat import render_user_export_tab
+            with get_connection() as conn:
+                render_user_export_tab(conn, user_id=user["id"])
+
+
+if __name__ == "__main__":
+    # One-time admin bootstrap if DB is empty
+    bootstrap_admin_from_env()
+    main()