"""Streamlit UI for the GitHub Repository Analyzer."""
import json
import os
import sys
from pathlib import Path
# Load .env from project root before anything else
APP_ROOT = Path(__file__).resolve().parent
PROJECT_ROOT = APP_ROOT.parent
if str(PROJECT_ROOT) not in sys.path:
sys.path.insert(0, str(PROJECT_ROOT))
_env = PROJECT_ROOT / ".env"
if _env.exists():
for _line in _env.read_text(encoding="utf-8").splitlines():
_line = _line.strip()
if _line and not _line.startswith("#") and "=" in _line:
_k, _, _v = _line.partition("=")
os.environ.setdefault(_k.strip(), _v.strip())
import streamlit as st
from src.repo_analyzer import Config, analyze_full
from src.repo_analyzer.claims import claims_for_reporting
from src.repo_analyzer.pdf_parser import parse_pdf_to_sections
from src.repo_analyzer.pdf_parser_marker import (
is_marker_available,
parse_pdf_to_sections_marker,
)
from src.repo_analyzer.pdf_report import build_pdf_report
from src.repo_analyzer.report import build_text_report
LEADERBOARD_ROOT = PROJECT_ROOT / "leaderboard"
LEADERBOARD_MANIFEST = LEADERBOARD_ROOT / "manifest.json"
CHECK_LABELS = {
"readme": "README",
"license": "License",
"commits": "Commits (>5)",
"execution_files": "Entry-point files",
"requirements": "Requirements file",
"tests": "Tests",
"data_files": "Data files",
"experiment_scripts": "Experiment scripts",
}
def load_leaderboard_manifest() -> list[dict]:
"""Return the lightweight saved-report index."""
if not LEADERBOARD_MANIFEST.exists():
return []
data = json.loads(LEADERBOARD_MANIFEST.read_text(encoding="utf-8"))
if not isinstance(data, list):
raise ValueError("leaderboard/manifest.json must contain a JSON array")
return sorted(data, key=lambda entry: (entry.get("language", ""), entry.get("rank", 0)))
def filter_leaderboard_manifest(manifest: list[dict], language: str) -> list[dict]:
"""Return saved reports for one language sorted by rank."""
entries = [entry for entry in manifest if entry.get("language") == language]
return sorted(entries, key=lambda entry: entry.get("rank", 0))
def resolve_leaderboard_path(path_text: str) -> Path:
"""Resolve a manifest path and keep it inside the leaderboard folder."""
path = Path(path_text)
candidate = path if path.is_absolute() else PROJECT_ROOT / path
candidate = candidate.resolve()
leaderboard_root = LEADERBOARD_ROOT.resolve()
if not candidate.is_relative_to(leaderboard_root):
raise ValueError(f"Refusing to load path outside leaderboard: {path_text}")
return candidate
def load_saved_report(entry: dict) -> dict:
"""Load a saved report JSON from the manifest entry selected by the user."""
report_path = resolve_leaderboard_path(entry["json_path"])
return json.loads(report_path.read_text(encoding="utf-8"))
def clear_pdf_cache(prefix: str) -> None:
for suffix in ("pdf_en", "pdf_ru"):
st.session_state.pop(f"{prefix}_{suffix}", None)
def render_report(
report: dict,
*,
paper_name: str | None = None,
repo_url: str | None = None,
pdf_cache_prefix: str,
openrouter_key: str | None = None,
model: str | None = None,
file_stem: str = "report",
show_back_button: bool = False,
back_button_key: str = "saved_report_back",
back_state_prefix: str = "leaderboard",
) -> None:
"""Render the report UI used by both fresh and saved analyses."""
summary = report["summary"]
score = summary["score"]
breakdown = summary["score_breakdown"]
repo_type = summary["repo_type"]
display_repo_url = repo_url or report.get("repo_url", "")
# ── Сontrols ───────────────────────────────────────────────────────────────
col_dl0, col_dl1, col_dl2, col_dl3, col_dl4 = st.columns([1, 0.5, 0.5, 1, 2])
with col_dl0:
if show_back_button:
if st.button(":material/arrow_back: Back to saved reports", key=back_button_key, type="primary"):
st.session_state.pop(f"{back_state_prefix}_selected", None)
st.session_state.pop(f"{back_state_prefix}_report", None)
st.session_state.pop(f"{back_state_prefix}_report_path", None)
st.session_state.pop(f"{back_state_prefix}_loading", None)
clear_pdf_cache(pdf_cache_prefix)
st.rerun()
with col_dl1:
st.download_button(
":material/download: Download JSON report",
data=json.dumps(report, ensure_ascii=False, indent=2),
file_name=f"{file_stem}.json",
mime="application/json",
key=f"{pdf_cache_prefix}_download_json",
)
with col_dl2:
pdf_en_key = f"{pdf_cache_prefix}_pdf_en"
if pdf_en_key not in st.session_state:
st.session_state[pdf_en_key] = build_pdf_report(report, lang="en", paper_name=paper_name)
st.download_button(
":material/file_save: Download PDF (EN)",
data=st.session_state[pdf_en_key],
file_name=f"{file_stem}_en.pdf",
mime="application/pdf",
key=f"{pdf_cache_prefix}_download_pdf_en",
)
with col_dl3:
pdf_ru_key = f"{pdf_cache_prefix}_pdf_ru"
if pdf_ru_key not in st.session_state:
st.session_state[pdf_ru_key] = build_pdf_report(
report,
lang="ru",
openrouter_key=openrouter_key,
model=model,
paper_name=paper_name,
)
st.download_button(
":material/file_save: Download PDF (RU)",
data=st.session_state[pdf_ru_key],
file_name=f"{file_stem}_ru.pdf",
mime="application/pdf",
key=f"{pdf_cache_prefix}_download_pdf_ru",
)
with col_dl4:
with st.expander("Text report"):
st.code(build_text_report(report), language=None)
st.divider()
# ── Score ─────────────────────────────────────────────────────────────────
col_score, col_meta = st.columns([1, 3])
with col_score:
color = "var(--app-success)" if score >= 70 else "var(--app-warning)" if score >= 40 else "var(--app-danger)"
st.markdown(
f""
f"
Sanity Score:
"
f"
{score}"
f"
/100"
f"
"
f"
Implementation rate
"
f"
{rate_pct}%"
f"