Update app.py

app.py

@@ -21,6 +21,11 @@ ADMIN_PASS = os.environ.get("ADMIN_PASS")
 if not ADMIN_PASS:
     raise ValueError("CRITICAL SECURITY ERROR: ADMIN_PASS environment variable is not set.")
 
+# Path to SQLite DB — override via DB_PATH env var if your file has a different name
+DB_PATH = os.environ.get("DB_PATH", "database.db")
+
+
+# ── Helper functions ──────────────────────────────────────────────────────────
 
 def get_faqs():
     data = fetch_all_faq_metadata()
@@ -57,51 +62,59 @@ def run_sync():
     except Exception as e:
         return f"Sync Failed: {e}"
 
+def prepare_download():
+    """Return the SQLite file path for Gradio to serve as a download."""
+    if not os.path.exists(DB_PATH):
+        return None, f"❌ Database file not found at: `{DB_PATH}`. Check your DB_PATH env var."
+    size_mb = os.path.getsize(DB_PATH) / (1024 * 1024)
+    return DB_PATH, f"✅ Ready — `{os.path.basename(DB_PATH)}` ({size_mb:.2f} MB). Click the file below to save it."
+
 def check_login(username, password):
-    """Validate credentials and return visibility updates."""
     if username == ADMIN_USER and password == ADMIN_PASS:
         return (
             gr.update(visible=False),  # Hide login panel
             gr.update(visible=True),   # Show dashboard
-            ""                         # Clear error message
+            ""                         # Clear error
         )
     else:
         return (
-            gr.update(visible=True),   # Keep login panel visible
-            gr.update(visible=False),  # Keep dashboard hidden
+            gr.update(visible=True),
+            gr.update(visible=False),
             "❌ Invalid username or password."
         )
 
 def logout():
-    """Log out and show login panel again."""
     return (
         gr.update(visible=True),   # Show login panel
         gr.update(visible=False),  # Hide dashboard
-        "",                        # Clear username
-        ""                         # Clear password
+        "",                        # Clear username field
+        ""                         # Clear password field
     )
 
 
+# ── UI ────────────────────────────────────────────────────────────────────────
+
 with gr.Blocks(title="Get Scene Admin Dashboard", theme=gr.themes.Base()) as demo:
 
     # ── LOGIN PANEL ──────────────────────────────────────────────
     with gr.Column(visible=True, elem_id="login_panel") as login_panel:
-        gr.Markdown("# Get Scene Admin\nPlease log in to continue.")
+        gr.Markdown("# 🎭 Get Scene Admin\nPlease log in to continue.")
         login_user = gr.Textbox(label="Username", placeholder="Enter username")
         login_pass = gr.Textbox(label="Password", placeholder="Enter password", type="password")
-        login_error = gr.Markdown("", visible=True)
+        login_error = gr.Markdown("")
         login_btn = gr.Button("Log In", variant="primary")
 
     # ── DASHBOARD (hidden until login) ───────────────────────────
     with gr.Column(visible=False) as dashboard:
         with gr.Row():
-            gr.Markdown("# Get Scene Admin Dashboard")
+            gr.Markdown("# 🎭 Get Scene Admin Dashboard")
             logout_btn = gr.Button("Log Out", variant="stop", scale=0)
 
         gr.Markdown("Manage FAQs, Podcasts, and Knowledge Embeddings.")
 
         with gr.Tabs():
-            # Tab 1: FAQs
+
+            # ── Tab 1: FAQs ──────────────────────────────────────
             with gr.TabItem("Manage FAQs"):
                 with gr.Row():
                     faq_df = gr.Dataframe(
@@ -132,7 +145,7 @@ with gr.Blocks(title="Get Scene Admin Dashboard", theme=gr.themes.Base()) as dem
                 add_btn.click(add_and_refresh, [new_q, new_a], [faq_df, new_q, new_a])
                 upload_faq_btn.click(handle_faq_upload, [faq_file], [faq_upload_status])
 
-            # Tab 2: Podcasts
+            # ── Tab 2: Podcasts ──────────────────────────────────
             with gr.TabItem("Podcasts"):
                 pod_df = gr.Dataframe(
                     value=get_podcasts,
@@ -147,15 +160,47 @@ with gr.Blocks(title="Get Scene Admin Dashboard", theme=gr.themes.Base()) as dem
 
                 upload_pod_btn.click(handle_podcast_upload, [pod_file], [pod_upload_status])
 
-            # Tab 3: Sync
+            # ── Tab 3: Sync & Embed ──────────────────────────────
             with gr.TabItem("Sync & Embed"):
                 gr.Markdown("### Recalculate Embeddings")
-                gr.Markdown("When you change text or upload new data, the 'embeddings' (AI understanding) must be recalculated for the chatbot to recognize the new information.")
+                gr.Markdown(
+                    "When you change text or upload new data, the 'embeddings' (AI understanding) "
+                    "must be recalculated for the chatbot to recognize the new information."
+                )
                 sync_btn = gr.Button("🔄 Sync & Recalculate Embeddings", variant="primary", scale=2)
                 sync_status = gr.Textbox(label="Sync Status", interactive=False)
 
                 sync_btn.click(run_sync, None, [sync_status])
 
+            # ── Tab 4: Download Database ─────────────────────────
+            with gr.TabItem("⬇️ Download Database"):
+                gr.Markdown("### Download SQLite Database")
+                gr.Markdown(
+                    "Click the button below to download the raw `database.db` SQLite file. "
+                    "You can open it locally with [DB Browser for SQLite](https://sqlitebrowser.org/) "
+                    "or any other SQLite viewer."
+                )
+                download_btn = gr.Button("⬇️ Download database.db", variant="primary")
+                download_status = gr.Markdown("")
+                db_file_output = gr.File(
+                    label="📦 database.db — click to save",
+                    visible=False,
+                    interactive=False
+                )
+
+                def on_download_click():
+                    path, msg = prepare_download()
+                    if path:
+                        return gr.update(value=path, visible=True), msg
+                    else:
+                        return gr.update(visible=False), msg
+
+                download_btn.click(
+                    on_download_click,
+                    inputs=None,
+                    outputs=[db_file_output, download_status]
+                )
+
     # ── WIRING ────────────────────────────────────────────────────
     login_btn.click(
         check_login,
@@ -163,7 +208,6 @@ with gr.Blocks(title="Get Scene Admin Dashboard", theme=gr.themes.Base()) as dem
         outputs=[login_panel, dashboard, login_error]
     )
 
-    # Allow pressing Enter in password field to log in
     login_pass.submit(
         check_login,
         inputs=[login_user, login_pass],
@@ -178,5 +222,4 @@ with gr.Blocks(title="Get Scene Admin Dashboard", theme=gr.themes.Base()) as dem
 
 
 if __name__ == "__main__":
-    # No auth= parameter — login is handled inside the app
     demo.launch(server_name="0.0.0.0")