134 lines
5.2 KiB
Python
134 lines
5.2 KiB
Python
# app/app.py
|
|
"""
|
|
Streamlit entry point — uses st.navigation() to control the sidebar.
|
|
Main workflow pages are listed at the top; Settings is separated into
|
|
a "System" section so it doesn't crowd the navigation.
|
|
|
|
Run: streamlit run app/app.py
|
|
bash scripts/manage-ui.sh start
|
|
"""
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
|
|
|
import streamlit as st
|
|
from scripts.db import DEFAULT_DB, init_db, get_active_tasks
|
|
import sqlite3
|
|
|
|
st.set_page_config(
|
|
page_title="Job Seeker",
|
|
page_icon="💼",
|
|
layout="wide",
|
|
)
|
|
|
|
init_db(DEFAULT_DB)
|
|
|
|
# ── Startup cleanup — runs once per server process via cache_resource ──────────
|
|
@st.cache_resource
|
|
def _startup() -> None:
|
|
"""Runs exactly once per server lifetime (st.cache_resource).
|
|
1. Marks zombie tasks as failed.
|
|
2. Auto-queues re-runs for any research generated without SearXNG data,
|
|
if SearXNG is now reachable.
|
|
"""
|
|
conn = sqlite3.connect(DEFAULT_DB)
|
|
conn.execute(
|
|
"UPDATE background_tasks SET status='failed', error='Interrupted by server restart',"
|
|
" finished_at=datetime('now') WHERE status IN ('queued','running')"
|
|
)
|
|
conn.commit()
|
|
|
|
# Auto-recovery: re-run LLM-only research when SearXNG is available
|
|
try:
|
|
import requests as _req
|
|
if _req.get("http://localhost:8888/", timeout=3).status_code == 200:
|
|
from scripts.task_runner import submit_task
|
|
_ACTIVE_STAGES = ("phone_screen", "interviewing", "offer", "hired")
|
|
rows = conn.execute(
|
|
"""SELECT cr.job_id FROM company_research cr
|
|
JOIN jobs j ON j.id = cr.job_id
|
|
WHERE (cr.scrape_used IS NULL OR cr.scrape_used = 0)
|
|
AND j.status IN ({})""".format(",".join("?" * len(_ACTIVE_STAGES))),
|
|
_ACTIVE_STAGES,
|
|
).fetchall()
|
|
for (job_id,) in rows:
|
|
submit_task(str(DEFAULT_DB), "company_research", job_id)
|
|
except Exception:
|
|
pass # never block startup
|
|
|
|
conn.close()
|
|
|
|
_startup()
|
|
|
|
# ── First-run wizard gate ───────────────────────────────────────────────────────
|
|
from scripts.user_profile import UserProfile as _UserProfile
|
|
_USER_YAML = Path(__file__).parent.parent / "config" / "user.yaml"
|
|
|
|
_show_wizard = (
|
|
not _UserProfile.exists(_USER_YAML)
|
|
or not _UserProfile(_USER_YAML).wizard_complete
|
|
)
|
|
if _show_wizard:
|
|
_setup_page = st.Page("pages/0_Setup.py", title="Setup", icon="👋")
|
|
st.navigation({"": [_setup_page]}).run()
|
|
st.stop()
|
|
|
|
# ── Navigation ─────────────────────────────────────────────────────────────────
|
|
# st.navigation() must be called before any sidebar writes so it can establish
|
|
# the navigation structure first; sidebar additions come after.
|
|
pages = {
|
|
"": [
|
|
st.Page("Home.py", title="Home", icon="🏠"),
|
|
st.Page("pages/1_Job_Review.py", title="Job Review", icon="📋"),
|
|
st.Page("pages/4_Apply.py", title="Apply Workspace", icon="🚀"),
|
|
st.Page("pages/5_Interviews.py", title="Interviews", icon="🎯"),
|
|
st.Page("pages/6_Interview_Prep.py", title="Interview Prep", icon="📞"),
|
|
st.Page("pages/7_Survey.py", title="Survey Assistant", icon="📋"),
|
|
],
|
|
"System": [
|
|
st.Page("pages/2_Settings.py", title="Settings", icon="⚙️"),
|
|
],
|
|
}
|
|
|
|
pg = st.navigation(pages)
|
|
|
|
# ── Background task sidebar indicator ─────────────────────────────────────────
|
|
# Fragment polls every 3s so stage labels update live without a full page reload.
|
|
# The sidebar context WRAPS the fragment call — do not write to st.sidebar inside it.
|
|
@st.fragment(run_every=3)
|
|
def _task_indicator():
|
|
tasks = get_active_tasks(DEFAULT_DB)
|
|
if not tasks:
|
|
return
|
|
st.divider()
|
|
st.markdown(f"**⏳ {len(tasks)} task(s) running**")
|
|
for t in tasks:
|
|
icon = "⏳" if t["status"] == "running" else "🕐"
|
|
task_type = t["task_type"]
|
|
if task_type == "cover_letter":
|
|
label = "Cover letter"
|
|
elif task_type == "company_research":
|
|
label = "Research"
|
|
elif task_type == "email_sync":
|
|
label = "Email sync"
|
|
elif task_type == "discovery":
|
|
label = "Discovery"
|
|
elif task_type == "enrich_descriptions":
|
|
label = "Enriching"
|
|
elif task_type == "scrape_url":
|
|
label = "Scraping URL"
|
|
elif task_type == "wizard_generate":
|
|
label = "Wizard generation"
|
|
elif task_type == "enrich_craigslist":
|
|
label = "Enriching listing"
|
|
else:
|
|
label = task_type.replace("_", " ").title()
|
|
stage = t.get("stage") or ""
|
|
detail = f" · {stage}" if stage else (f" — {t.get('company')}" if t.get("company") else "")
|
|
st.caption(f"{icon} {label}{detail}")
|
|
|
|
with st.sidebar:
|
|
_task_indicator()
|
|
|
|
pg.run()
|