Update App.py

App.py

@@ -1,506 +1,158 @@
-import os, re, json, shutil, signal, subprocess, threading, time, uuid, zipfile, html
-from pathlib import Path
-from flask import Flask, request, abort, send_file, jsonify
-
-app = Flask(__name__)
-app.config["MAX_CONTENT_LENGTH"] = 500 * 1024 * 1024
-
-JOBS = Path(os.environ.get("JOBS_DIR", "/tmp/jobs")); JOBS.mkdir(parents=True, exist_ok=True)
-GRADLE = os.environ.get("GRADLE_BIN", "gradle")
-SDK_DIR = os.environ.get("ANDROID_HOME", "/opt/android-sdk")
-ACCESS_KEY = os.environ.get("ACCESS_KEY", "")      # optional; empty = no password
-PORT = int(os.environ.get("PORT", "7860"))
-
-# ---- signing key: asked on EVERY build, never stored ----
-# The .jks + password come with the upload form. They live only in memory
-# (KEYS dict) and as a temp file inside the job folder while Gradle runs, and
-# are deleted right after. Nothing about the key is written to state.json/logs.
-KEYS = {}   # job id -> {"path": Path, "password": str, "alias": str}
-
-MODES = {  # form value -> (gradle task, output glob, file name, label)
-    "debug":   ("assembleDebug",   "build/outputs/apk/debug/*.apk",      "app-debug.apk",   "Debug APK"),
-    "release": ("assembleRelease", "build/outputs/apk/release/*.apk",    "app-release.apk", "Release APK (signed, for installing)"),
-    "bundle":  ("bundleRelease",   "build/outputs/bundle/release/*.aab", "app-release.aab", "Release AAB (for Play Store)"),
-}
-
-def key_fingerprints(path, alias, password):
-    """SHA-1 / SHA-256 of the uploaded key (public values, safe to show)."""
-    try:
-        out = subprocess.run(["keytool", "-list", "-v", "-keystore", str(path), "-alias", alias,
-                              "-storepass", password], capture_output=True, text=True, timeout=30)
-        sha1 = re.search(r"SHA1:\s*([0-9A-F:]+)", out.stdout)
-        sha256 = re.search(r"SHA256:\s*([0-9A-F:]+)", out.stdout)
-        if not sha1:
-            return None
-        return {"sha1": sha1.group(1), "sha256": sha256.group(1) if sha256 else ""}
-    except Exception:
-        return None
-
-build_lock = threading.Lock()   # one build at a time (RAM)
-JOB, PROCS = {}, {}
-ID_RE = re.compile(r"^[0-9a-f]{10}$")
-ACTIVE = ("extracting", "queued", "preparing", "building")
-
-
-# ---------------------------------------------------------------- state
-def save(j):
+import os, re, hmac, threading, traceback
+import numpy as np
+import torch
+from fastapi import FastAPI, Request
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse, StreamingResponse
+from pydantic import BaseModel
+
+torch.set_num_threads(int(os.environ.get("TORCH_THREADS", "2")))
+
+HERE = os.path.dirname(os.path.abspath(__file__))
+SECRET = os.environ.get("TTS_SECRET_KEY", "")          # Space Settings -> Secrets me daalo
+CONFIG = os.environ.get("TTS_CONFIG", "hf://saryps-labs/pocket-tts-hindi/config.yaml")
+VOICE_FILE = os.path.join(HERE, "voice.wav")
+FALLBACK_VOICE = "hf://kyutai/tts-voices/alba-mackenna/casual.wav"
+QUANTIZE = os.environ.get("TTS_QUANTIZE", "0") == "1"   # int8 (thoda tez), default band
+MAX_CHARS = int(os.environ.get("MAX_CHARS", "6000"))
+GAP_SECONDS = 0.22                                       # vakyon ke beech ka pause
+
+model = None
+voice_state = None
+SR = 24000
+state_msg = "loading"
+gen_lock = threading.Lock()
+
+
+def load_everything():
+    global model, voice_state, SR, state_msg
     try:
-        (JOBS / j["id"] / "state.json").write_text(json.dumps(j))
-    except Exception:
-        pass
-
-def load_old():
-    for d in JOBS.iterdir():
-        if not d.is_dir(): continue
-        f = d / "state.json"
-        if f.exists():
+        from pocket_tts import TTSModel
+        try:
+            m = TTSModel.load_model(CONFIG, temp=0.3, lsd_decode_steps=1, eos_threshold=-1.0)
+        except TypeError:
+            m = TTSModel.load_model(CONFIG)
+        if QUANTIZE:
             try:
-                j = json.loads(f.read_text())
-                if j["status"] in ACTIVE:
-                    j["status"], j["error"] = "failed", "Server restarted while building."
-                    j["finished"] = j.get("finished") or time.time()
-                JOB[j["id"]] = j
-            except Exception:
-                pass
-load_old()
-
-DEFAULT_TASKS = 60   # only a last-resort guess; real count comes from a gradle --dry-run
-def load_totals():
-    try: return json.loads((JOBS / "_totals.json").read_text())
-    except Exception: return {}
-def save_totals(t):
-    try: (JOBS / "_totals.json").write_text(json.dumps(t))
-    except Exception: pass
-
-def get_job(jid):
-    if not ID_RE.match(jid) or jid not in JOB:
-        abort(404)
-    return JOB[jid]
-
-def public(j):
-    now = time.time()
-    t0 = j.get("started") or j["created"]
-    end = j.get("finished") or now
-    out = {k: j.get(k) for k in ("id", "name", "status", "total", "done", "current", "error", "apk_mb")}
-    out["elapsed"] = max(0, end - t0)
-    out["has_apk"] = (JOBS / j["id"] / j.get("out", "app-debug.apk")).exists()
-    out["out"] = j.get("out", "app-debug.apk")
-    out["mode"] = j.get("mode", "debug")
-    out["fp"] = j.get("fp")
+                m = torch.quantization.quantize_dynamic(m, {torch.nn.Linear}, dtype=torch.qint8)
+                print("int8 quantization ON")
+            except Exception as e:
+                print("quantize fail, fp32 chal raha he:", e)
+        voice_path = VOICE_FILE if os.path.exists(VOICE_FILE) else FALLBACK_VOICE
+        print("Voice prompt:", voice_path)
+        voice_state = m.get_state_for_audio_prompt(voice_path)
+        SR = int(getattr(m, "sample_rate", 24000))
+        model = m
+        state_msg = "ready"
+        print("Model ready, sample rate", SR)
+    except Exception:
+        state_msg = "error"
+        traceback.print_exc()
+
+
+threading.Thread(target=load_everything, daemon=True).start()
+
+app = FastAPI()
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+    expose_headers=["X-Sample-Rate"],
+)
+
+
+def authorized(request: Request) -> bool:
+    if not SECRET:
+        return False
+    cands = [
+        request.headers.get("x-secret-key", ""),
+        request.query_params.get("key", ""),
+    ]
+    auth = request.headers.get("authorization", "")
+    if auth.startswith("Bearer "):
+        cands.append(auth[7:])
+    return any(hmac.compare_digest(c.encode(), SECRET.encode()) for c in cands if c)
+
+
+def split_clean(text: str):
+    """Text ko vakyon me todo aur punctuation/English hata do (model punctuation nahi samajhta)."""
+    text = text.replace("\r", "\n")
+    parts = re.split(r"[\u0964\u0965\.\?\!\n;:]+", text)
+    out = []
+    for p in parts:
+        # sirf Devanagari (danda chhodkar), digits aur space rakho
+        p = re.sub(r"[^\u0900-\u0963\u0966-\u097F0-9\s]", " ", p)
+        p = re.sub(r"\s+", " ", p).strip()
+        if len(p) < 3:
+            continue
+        # bahut lambe vakya ko ~200 chars ke tukdon me todo
+        while len(p) > 220:
+            cut = p.rfind(" ", 0, 200)
+            cut = cut if cut > 50 else 200
+            out.append(p[:cut].strip())
+            p = p[cut:].strip()
+        if p:
+            out.append(p)
     return out
 
 
-# ---------------------------------------------------------------- build
-def find_root(src: Path):
-    cands = [src] + sorted([p for p in src.rglob("*") if p.is_dir()], key=lambda p: len(p.parts))
-    for p in cands:
-        if (p / "settings.gradle.kts").exists() or (p / "settings.gradle").exists():
-            return p
-    return None
-
-def summarize_error(text):
-    lines = text.splitlines()
-    out = [l for l in lines if l.startswith("e: ")][:15]
-    for i, l in enumerate(lines):
-        if l.startswith("* What went wrong:"):
-            out += lines[i:i + 6]
-            break
-    if not out:
-        out = lines[-25:]
-    return "\n".join(out)
-
-def run_build(jid):
-    j = JOB[jid]; d = JOBS / jid; logp = d / "build.log"
-    lf = open(logp, "ab", buffering=0)
-    def w(s): lf.write(s.encode("utf-8", "replace"))
-    def setj(**kw): j.update(kw); save(j)
-    t0 = time.time()
-    def el():
-        n = int(time.time() - t0); return f"+{n // 60}:{n % 60:02d}"
-    def finish(status, error=""):
-        KEYS.pop(jid, None)
-        try: (d / "upload.jks").unlink(missing_ok=True)
-        except Exception: pass
-        w(f"\n[{el()}] {status.upper()}\n")
-        setj(status=status, error=error, finished=time.time(), current="")
-        lf.close()
+def iter_audio(text: str):
+    if hasattr(model, "generate_audio_stream"):
+        for a in model.generate_audio_stream(voice_state, text):
+            yield a
+    else:
+        yield model.generate_audio(voice_state, text)
 
-    def gradle(args, on_line):
-        p = subprocess.Popen([GRADLE, *args], cwd=j["root"], stdout=subprocess.PIPE,
-                             stderr=subprocess.STDOUT, text=True, bufsize=1, errors="replace",
-                             start_new_session=True, env=genv)
-        PROCS[jid] = p
-        for line in p.stdout:
-            w(line); on_line(line.rstrip("\n"))
-        rc = p.wait(); PROCS.pop(jid, None)
-        return rc
 
-    task, out_glob, out_name, _label = MODES.get(j.get("mode", "debug"), MODES["debug"])
-    genv = os.environ.copy()
-    genv.pop("ACCESS_KEY", None)
-    key = KEYS.get(jid)
-    keyfile = key["path"] if key else (d / "upload.jks")
+def to_pcm16(a) -> bytes:
+    arr = a.detach().cpu().numpy() if hasattr(a, "detach") else np.asarray(a)
+    arr = np.clip(arr.astype(np.float32).reshape(-1), -1.0, 1.0)
+    return (arr * 32767.0).astype("<i2").tobytes()
 
-    def drop_key():
-        KEYS.pop(jid, None)
-        try: keyfile.unlink(missing_ok=True)
-        except Exception: pass
 
-    try:
-        # 0. signing key (from the upload form)
-        if key:
-            genv.update(KEYSTORE_PATH=str(keyfile), STORE_PASSWORD=key["password"],
-                        KEY_PASSWORD=key["password"], KEY_ALIAS=key["alias"])
-            w("Signing with the uploaded key (deleted after the build).\n")
-        elif task != "assembleDebug":
-            return finish("failed", "Release builds need your signing key: choose the .jks file and enter its password.")
-        else:
-            w("No key given: signed with a temporary debug key (SHA-1 will differ from your real key).\n")
-
-        # 1. extract
-        setj(status="extracting", current="Unzipping project…")
-        w("== Extracting zip ==\n")
-        with zipfile.ZipFile(d / "in.zip") as z:
-            z.extractall(d / "src")
-        root = find_root(d / "src")
-        if not root:
-            return finish("failed", "settings.gradle(.kts) not found in the zip.")
-        j["root"] = str(root)
-        (root / "local.properties").write_text(f"sdk.dir={SDK_DIR}\n")
-        w(f"Project root: {root.relative_to(d / 'src')}\n")
-
-        # 2. wait for slot
-        if not build_lock.acquire(blocking=False):
-            setj(status="queued", current="Waiting for another build to finish…")
-            w("== Queued (another build is running) ==\n")
-            while not build_lock.acquire(timeout=1):
-                if j.get("cancel"):
-                    return finish("cancelled")
-        try:
-            if j.get("cancel"):
-                return finish("cancelled")
-            setj(started=time.time())
-
-            # 3a. count the REAL number of tasks with a dry-run (no tasks are executed),
-            #     so the progress bar is accurate from the very first build.
-            setj(status="preparing", current="Counting build tasks…", total=0, done=0)
-            w(f"\n[{el()}] == Counting tasks (dry-run) ==\n")
-            counted = [0]
-            def on_dry(line):
-                if re.match(r"^:\S+ SKIPPED", line):
-                    counted[0] += 1
-            rc0 = gradle([task, "--dry-run", "--console=plain",
-                          "--no-configuration-cache", "-Dorg.gradle.jvmargs=-Xmx4g"], on_dry)
-            if j.get("cancel"):
-                return finish("cancelled")
-            if rc0 == 0 and counted[0] > 0:
-                total, exact = counted[0], True
-                w(f"[{el()}] Tasks in graph: {total}\n")
-            else:
-                total, exact = load_totals().get(j["name"], DEFAULT_TASKS), False
-                w(f"[{el()}] Dry-run failed, estimating {total} tasks\n")
-
-            # 3b. build. Progress = tasks done / tasks counted above
-            setj(total=total, done=0, status="building", current="Starting Gradle…")
-            w(f"\n[{el()}] == Building ({task}) ==\n")
-            def on_line(line):
-                m = re.match(r"^> Task (\S+)", line)
-                if m:
-                    j["done"] += 1; j["current"] = m.group(1)
-                    if exact:
-                        if j["done"] >= j["total"]:          # more tasks than counted: never exceed 100%
-                            j["total"] = j["done"] + 1
-                    elif j["done"] >= j["total"] - 3:        # guessed total was too small: grow it
-                        j["total"] = j["done"] + 10
-                    if j["done"] % 10 == 0: save(j)
-                elif line.startswith("Download "):
-                    j["current"] = line[:90]
-            rc = gradle([task, "--console=plain", "--warning-mode=summary",
-                         "--build-cache", "--no-configuration-cache",
-                         "-Dorg.gradle.jvmargs=-Xmx4g"], on_line)
-            if j.get("cancel"):
-                return finish("cancelled")
-            apks = list(Path(j["root"]).rglob(out_glob))
-            if rc == 0 and apks:
-                shutil.copy(apks[0], d / out_name)
-                j["apk_mb"] = round((d / out_name).stat().st_size / 1048576, 1)
-                j["total"] = j["done"]
-                t = load_totals(); t[j["name"]] = j["done"]; save_totals(t)
-                w(f"\n[{el()}] BUILD OK\n")
-                return finish("done")
-            return finish("failed", summarize_error(logp.read_text(errors="replace")))
-        finally:
-            build_lock.release()
-            shutil.rmtree(d / "src", ignore_errors=True)   # save disk
-            drop_key()                                     # key never stays on disk
-    except Exception as e:
-        drop_key()
-        try: finish("failed", f"Internal error: {e}")
-        except Exception: pass
-
-
-# ---------------------------------------------------------------- pages
-CSS = r"""
-:root{--bg:#0b1020;--card:#151b2e;--bd:#262e48;--tx:#e8ecf8;--mu:#8b95b5;--ac:#5b8cff;--ok:#22c55e;--er:#ef4444;--wa:#f59e0b}
-*{box-sizing:border-box}body{margin:0;background:var(--bg);color:var(--tx);font:15px/1.45 system-ui,sans-serif}
-.w{max-width:760px;margin:0 auto;padding:14px}
-h1{font-size:20px;margin:6px 0 14px}.card{background:var(--card);border:1px solid var(--bd);border-radius:14px;padding:14px;margin-bottom:12px}
-button,.btn{font:inherit;border:0;border-radius:10px;padding:11px 14px;background:#27304d;color:var(--tx);cursor:pointer;text-decoration:none;display:inline-block;text-align:center}
-button.p,.btn.p{background:var(--ac);color:#fff}.btn.g,button.g{background:var(--ok);color:#04210f;font-weight:600}
-button.r{background:#4a1d24;color:#ffb4b4}button:disabled{opacity:.5}
-.row{display:flex;gap:8px;flex-wrap:wrap}.row>*{flex:1 1 auto}
-input[type=file],input[type=password]{width:100%;padding:12px;border-radius:10px;border:1px dashed var(--bd);background:#0f1426;color:var(--tx);margin-bottom:10px}
-.bar{height:16px;background:#0f1426;border-radius:99px;overflow:hidden;border:1px solid var(--bd)}
-.bar>i{display:block;height:100%;width:0;background:linear-gradient(90deg,#5b8cff,#7c5bff);transition:width .4s}
-.bar.ind>i{width:35%;animation:ind 1.2s infinite ease-in-out}.bar.ok>i{background:var(--ok)}.bar.er>i{background:var(--er)}
-@keyframes ind{0%{margin-left:-35%}100%{margin-left:100%}}
-.mu{color:var(--mu);font-size:13px}.big{font-size:28px;font-weight:700}
-.steps{display:flex;gap:6px;margin:12px 0}.steps span{flex:1;text-align:center;font-size:12px;padding:6px 2px;border-radius:8px;background:#0f1426;color:var(--mu);border:1px solid var(--bd)}
-.steps span.on{color:#fff;border-color:var(--ac);background:#1b2750}.steps span.dn{color:var(--ok);border-color:#1d5b36}
-pre{margin:0;background:#050810;color:#b6f5c4;border:1px solid var(--bd);border-radius:10px;padding:10px;font:12px/1.4 ui-monospace,monospace;max-height:50vh;overflow:auto;white-space:pre}
-pre.wrap{white-space:pre-wrap;word-break:break-all}
-.err{border-color:#5b2430;background:#1d0f14}.err pre{color:#ffb4b4;max-height:30vh}
-#toast{position:fixed;left:50%;bottom:22px;transform:translateX(-50%);background:#fff;color:#000;padding:9px 16px;border-radius:99px;opacity:0;transition:.25s;pointer-events:none}
-#toast.s{opacity:1}a{color:var(--ac)}.li{display:flex;justify-content:space-between;gap:8px;padding:8px 0;border-top:1px solid var(--bd)}
-.li:first-child{border-top:0}label.c{display:inline-flex;gap:6px;align-items:center;font-size:13px;color:var(--mu)}
-"""
-TOAST_JS = r"""
-function toast(m){const t=document.getElementById('toast');t.textContent=m;t.classList.add('s');setTimeout(()=>t.classList.remove('s'),1500)}
-async function copyText(t){try{await navigator.clipboard.writeText(t)}catch(e){const a=document.createElement('textarea');a.value=t;a.style.cssText='position:fixed;opacity:0';document.body.appendChild(a);a.select();document.execCommand('copy');a.remove()}toast('Copied ✓')}
-"""
+def pcm_stream(sentences):
+    silence = np.zeros(int(GAP_SECONDS * SR), dtype="<i2").tobytes()
+    with gen_lock:                      # ek samay me ek hi generation (CPU bachane ke liye)
+        for s in sentences:
+            try:
+                for a in iter_audio(s):
+                    yield to_pcm16(a)
+                yield silence
+            except Exception:
+                print("Sentence fail:", s[:60])
+                traceback.print_exc()
 
-def shell(title, body, js=""):
-    return ("<!doctype html><html><head><meta charset='utf-8'><meta name='viewport' content='width=device-width,initial-scale=1'>"
-            f"<title>{title}</title><style>{CSS}</style></head><body><div class='w'>{body}</div><div id='toast'></div>"
-            f"<script>{TOAST_JS}{js}</script></body></html>")
 
-ICON = {"extracting": "📂", "queued": "⏳", "preparing": "🧮", "building": "🔨",
-        "done": "✅", "failed": "❌", "cancelled": "🚫"}
+class TTSReq(BaseModel):
+    text: str
 
-HOME_JS = r"""
-const f=document.getElementById('f'),bar=document.getElementById('ub'),st=document.getElementById('us'),btn=document.getElementById('go');
-f.addEventListener('submit',e=>{e.preventDefault();
- const fd=new FormData(f);btn.disabled=true;bar.parentElement.style.display='block';
- const x=new XMLHttpRequest();x.open('POST','/build');
- x.upload.onprogress=ev=>{if(ev.lengthComputable){const p=Math.round(ev.loaded/ev.total*100);bar.style.width=p+'%';st.textContent='Uploading '+p+'% ('+(ev.loaded/1048576).toFixed(1)+' / '+(ev.total/1048576).toFixed(1)+' MB)'}};
- x.onload=()=>{if(x.status==200){location.href='/job/'+JSON.parse(x.responseText).id}else{let m='';try{m=JSON.parse(x.responseText).error||''}catch(e){}st.textContent='Error '+x.status+(x.status==403?' – wrong access key':'')+(m?' – '+m:'');btn.disabled=false}};
- x.onerror=()=>{st.textContent='Upload failed';btn.disabled=false};x.send(fd)});
-"""
 
 @app.get("/")
-def index():
-    key = "<input name='key' type='password' placeholder='Access key' required>" if ACCESS_KEY else ""
-    jobs = sorted(JOB.values(), key=lambda j: -j["created"])[:10]
-    rows = "".join(
-        f"<div class='li'><a href='/job/{j['id']}'>{ICON[j['status']]} {html.escape(j['name'])}</a>"
-        f"<span class='mu'>{time.strftime('%d %b %H:%M', time.localtime(j['created']))}</span></div>" for j in jobs
-    ) or "<div class='mu'>No builds yet.</div>"
-    opts = "".join(f"<option value='{k}'>{html.escape(v[3])}</option>" for k, v in MODES.items())
-    body = f"""<h1>📦 APK Builder</h1>
-<div class='card'><form id='f'><input type='file' name='zip' accept='.zip' required>{key}
-<select name='mode' style='width:100%;padding:12px;border-radius:10px;margin-bottom:10px;background:#0f1426;color:#e8ecf8;border:1px solid #262e48'>{opts}</select>
-<div class='mu' style='margin:4px 0 6px'>🔑 Signing key — needed for Release; for Debug it keeps the same SHA-1 as your real key. Not saved anywhere.</div>
-<input type='file' name='keystore' accept='.jks,.keystore,.p12'>
-<input type='password' name='storepass' placeholder='Keystore password' autocomplete='off'>
-<input type='text' name='alias' value='upload' placeholder='Key alias' style='width:100%;padding:12px;border-radius:10px;border:1px dashed #262e48;background:#0f1426;color:#e8ecf8;margin-bottom:10px'>
-<button class='p' id='go' style='width:100%'>Upload &amp; build</button></form>
-<div class='bar' style='display:none;margin-top:12px'><i id='ub'></i></div><div class='mu' id='us' style='margin-top:6px'></div></div>
-<div class='card'><b>Recent builds</b><div style='margin-top:6px'>{rows}</div></div>
-<div class='mu'>JDK 17 · Gradle 9.3.1</div>"""
-    return shell("APK Builder", body, HOME_JS)
-
-@app.post("/build")
-def build():
-    if ACCESS_KEY and request.form.get("key") != ACCESS_KEY:
-        abort(403)
-    f = request.files.get("zip")
-    if not f:
-        abort(400)
-    jid = uuid.uuid4().hex[:10]
-    d = JOBS / jid; d.mkdir(parents=True)
-    f.save(d / "in.zip")
-    if not zipfile.is_zipfile(d / "in.zip"):
-        shutil.rmtree(d); abort(400)
-    mode = request.form.get("mode", "debug")
-    if mode not in MODES:
-        mode = "debug"
-    kf = request.files.get("keystore")
-    password = request.form.get("storepass", "")
-    alias = (request.form.get("alias") or "upload").strip() or "upload"
-    fp = None
-    if kf and kf.filename:
-        kpath = d / "upload.jks"
-        kf.save(kpath)
-        os.chmod(kpath, 0o600)
-        fp = key_fingerprints(kpath, alias, password)
-        if not fp:
-            shutil.rmtree(d, ignore_errors=True)
-            return jsonify(error="Wrong keystore password or alias."), 400
-        KEYS[jid] = {"path": kpath, "password": password, "alias": alias}
-    elif mode != "debug":
-        shutil.rmtree(d, ignore_errors=True)
-        return jsonify(error="Release builds need your .jks key file and password."), 400
-    JOB[jid] = {"id": jid, "name": (f.filename or "project.zip")[:60], "status": "extracting",
-                "mode": mode, "out": MODES[mode][2], "fp": fp,
-                "created": time.time(), "started": None, "finished": None,
-                "total": 0, "done": 0, "current": "", "error": "", "apk_mb": None}
-    save(JOB[jid])
-    threading.Thread(target=run_build, args=(jid,), daemon=True).start()
-    return jsonify(id=jid)
-
-JOB_HTML = r"""
-<a href='/' class='mu'>← New build</a>
-<h1 id='title'>__NAME__</h1>
-<div class='card'>
- <div class='row' style='align-items:baseline'><div class='big' id='pct'>0%</div><div class='mu' id='meta' style='text-align:right'></div></div>
- <div class='bar ind' id='bar' style='margin-top:6px'><i></i></div>
- <div class='steps' id='steps'><span>Unzip</span><span>Queue</span><span>Build</span><span>APK</span></div>
- <div class='mu' id='cur' style='word-break:break-all'></div>
- <div class='row' style='margin-top:12px' id='acts'></div>
-</div>
-<div class='card' id='fpbox' style='display:none'>
- <b>🔑 Signed with</b> <span class='mu'>(add these in Firebase → Android app → Add fingerprint)</span>
- <div class='li'><span>SHA-1</span><button id='cpsha1'>Copy</button></div><pre class='wrap' id='sha1'></pre>
- <div class='li'><span>SHA-256</span><button id='cpsha256'>Copy</button></div><pre class='wrap' id='sha256'></pre>
-</div>
-<div class='card err' id='errbox' style='display:none'>
- <div class='row' style='align-items:center'><b>❌ What went wrong</b><button id='cperr' style='flex:0 0 auto'>Copy error</button></div>
- <pre class='wrap' id='errtxt' style='margin-top:8px'></pre>
-</div>
-<div class='card'>
- <div class='row' style='align-items:center;margin-bottom:8px'>
-  <b style='flex:0 0 auto'>Log</b>
-  <label class='c'><input type='checkbox' id='auto' checked> auto-scroll</label>
-  <label class='c'><input type='checkbox' id='wrap'> wrap</label>
- </div>
- <pre id='log'>waiting…</pre>
- <div class='row' style='margin-top:10px'>
-  <button id='cplog'>📋 Copy full log</button>
-  <a class='btn' href='/job/__ID__/log.txt' download='build-__ID__.log'>⬇ Download log</a>
- </div>
-</div>
-"""
-
-JOB_JS = r"""
-const ID='__ID__', END=['done','failed','cancelled'];
-const $=id=>document.getElementById(id);
-let offset=0, full='', S=null, fetchedAt=0, timer=null, lastEmitted='';
-const fmt=s=>{s=Math.floor(s);const m=Math.floor(s/60);return m+'m '+String(s%60).padStart(2,'0')+'s'};
-function stepIdx(s){return {extracting:0,queued:1,preparing:2,building:2,done:4,failed:-1,cancelled:-1}[s]}
-function render(){
- if(!S)return;
- const st=S.status, active=!END.includes(st);
- const total=S.total||0, done=S.done||0;
- let pct=null;
- if(st==='done')pct=100; else if(st==='building'&&total>0)pct=Math.min(99,Math.floor(done/total*100));
- else if(st==='failed'||st==='cancelled')pct=total>0?Math.min(99,Math.floor(done/total*100)):0;
- const bar=$('bar'); bar.className='bar'+(pct===null?' ind':'')+(st==='done'?' ok':'')+(st==='failed'?' er':'');
- if(pct!==null)bar.firstElementChild.style.width=pct+'%';
- $('pct').textContent=pct===null?(st==='queued'?'Queued':'…'):pct+'%';
- const el=S.elapsed+(active?(Date.now()-fetchedAt)/1000:0);
- let meta=fmt(el);
- if(st==='building'&&total>0){meta=done+'/'+total+' tasks · '+meta; if(pct>=8&&pct<100)meta+=' · ~'+fmt(el*(100-pct)/pct)+' left'}
- if(st==='done'&&S.apk_mb)meta+=' · APK '+S.apk_mb+' MB';
- $('meta').textContent=meta;
- $('cur').textContent=active?(S.current||''):({done:'Build finished 🎉',failed:'Build failed',cancelled:'Cancelled'}[st]);
- const idx=stepIdx(st);
- [...$('steps').children].forEach((e,i)=>e.className=(idx===4||i<idx)?'dn':(i===idx?'on':''));
- // actions
- const a=$('acts'), key=st+(S.has_apk?'1':'0'); if(key!==lastEmitted){lastEmitted=key; a.innerHTML='';
-  if(st==='done'&&S.has_apk){
-   a.insertAdjacentHTML('beforeend',"<a class='btn g' style='flex:2 1 100%;font-size:17px' href='/job/"+ID+"/apk' download>⬇ Download "+(S.out.endsWith('.aab')?'AAB':'APK')+"</a>");
-   a.insertAdjacentHTML('beforeend',"<button id='cplink'>🔗 Copy APK link</button>");
-   $('cplink').onclick=()=>copyText(location.origin+'/job/'+ID+'/apk');
-  }
-  if(active){a.insertAdjacentHTML('beforeend',"<button class='r' id='cancel'>✖ Cancel build</button>");
-   $('cancel').onclick=async()=>{if(confirm('Cancel this build?')){await fetch('/api/job/'+ID+'/cancel',{method:'POST'});toast('Cancelling…')}}}
-  if(!active){a.insertAdjacentHTML('beforeend',"<a class='btn p' href='/'>🔄 New build</a>")}
- }
- if(st==='failed'&&S.error){$('errbox').style.display='block';$('errtxt').textContent=S.error}
- if(S.fp&&S.fp.sha1){$('fpbox').style.display='block';$('sha1').textContent=S.fp.sha1;$('sha256').textContent=S.fp.sha256||''}
- document.title=(pct!==null?pct+'% · ':'')+S.name;
-}
-function showLog(){
- const lines=full.split('\n'); const p=$('log');
- p.textContent=lines.slice(-400).join('\n');
- if($('auto').checked)p.scrollTop=p.scrollHeight;
-}
-async function pullLog(){
- for(let i=0;i<20;i++){
-  const r=await (await fetch('/api/job/'+ID+'/log?offset='+offset)).json();
-  if(!r.text)break; full+=r.text; offset=r.offset;
-  if(!r.more)break;
- } showLog();
-}
-async function tick(){
- try{
-  S=await (await fetch('/api/job/'+ID)).json(); fetchedAt=Date.now(); render(); await pullLog();
-  if(END.includes(S.status)){await pullLog();render();return}
- }catch(e){}
- setTimeout(tick,1500);
-}
-setInterval(()=>{if(S&&!END.includes(S.status))render()},1000);
-$('cplog').onclick=()=>copyText(full);
-$('cpsha1').onclick=()=>copyText(S&&S.fp&&S.fp.sha1||'');
-$('cpsha256').onclick=()=>copyText(S&&S.fp&&S.fp.sha256||'');
-$('cperr').onclick=()=>copyText(S&&S.error||'');
-$('wrap').onchange=e=>$('log').classList.toggle('wrap',e.target.checked);
-tick();
-"""
-
-@app.get("/job/<jid>")
-def job_page(jid):
-    j = get_job(jid)
-    body = JOB_HTML.replace("__ID__", jid).replace("__NAME__", html.escape(j["name"]))
-    return shell("Build " + jid, body, JOB_JS.replace("__ID__", jid))
-
-@app.get("/api/job/<jid>")
-def api_job(jid):
-    return jsonify(public(get_job(jid)))
-
-@app.get("/api/job/<jid>/log")
-def api_log(jid):
-    get_job(jid)
-    p = JOBS / jid / "build.log"
-    off = max(0, request.args.get("offset", 0, type=int))
-    if not p.exists():
-        return jsonify(text="", offset=off, more=False)
-    with open(p, "rb") as f:
-        f.seek(off); data = f.read(200_000)
-    # don't cut a UTF-8 char in half
-    while data and (data[-1] & 0xC0) == 0x80: data = data[:-1]
-    if data and data[-1] >= 0xC0: data = data[:-1]
-    more = (off + len(data)) < p.stat().st_size
-    return jsonify(text=data.decode("utf-8", "replace"), offset=off + len(data), more=more)
-
-@app.get("/job/<jid>/log.txt")
-def log_txt(jid):
-    get_job(jid)
-    p = JOBS / jid / "build.log"
-    if not p.exists(): abort(404)
-    return send_file(p, mimetype="text/plain", as_attachment=True, download_name=f"build-{jid}.log")
-
-@app.get("/job/<jid>/apk")
-def apk(jid):
-    j = get_job(jid)
-    name = j.get("out", "app-debug.apk")
-    p = JOBS / jid / name
-    if not p.exists(): abort(404)
-    mime = "application/octet-stream" if name.endswith(".aab") else "application/vnd.android.package-archive"
-    return send_file(p, as_attachment=True, download_name=name, mimetype=mime)
-
-@app.post("/api/job/<jid>/cancel")
-def cancel(jid):
-    j = get_job(jid)
-    if j["status"] not in ACTIVE:
-        return jsonify(ok=False)
-    j["cancel"] = True
-    p = PROCS.get(jid)
-    if p:
-        try: os.killpg(p.pid, signal.SIGTERM)
-        except Exception: pass
-        subprocess.Popen([GRADLE, "--stop"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-    return jsonify(ok=True)
-
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=PORT, threaded=True)
+def root():
+    return {"status": "ok"}
+
+
+@app.get("/health")
+def health():
+    return {"state": state_msg}
+
+
+@app.post("/tts")
+def tts(req: TTSReq, request: Request):
+    if not authorized(request):
+        return JSONResponse({"error": "unauthorized"}, status_code=401)
+    if state_msg == "loading":
+        return JSONResponse({"error": "model abhi load ho raha he, 1 minute baad try karo"}, status_code=503)
+    if state_msg == "error" or model is None:
+        return JSONResponse({"error": "model load fail hua, Space ke logs dekho"}, status_code=500)
+    if len(req.text) > MAX_CHARS:
+        return JSONResponse({"error": f"text bahut lamba he (max {MAX_CHARS} characters)"}, status_code=413)
+    sentences = split_clean(req.text)
+    if not sentences:
+        return JSONResponse({"error": "text me Hindi (देवनागरी) shabd nahi mile"}, status_code=400)
+    return StreamingResponse(
+        pcm_stream(sentences),
+        media_type="application/octet-stream",
+        headers={"X-Sample-Rate": str(SR), "Cache-Control": "no-store"},
+    )