Upload app.py

app.py

@@ -1,4 +1,7 @@
-# app.py — Diag Studio AI Frontend (TR/EN, retries, warm-up, artifacts download, audio preview)
+# app.py — Diag Studio AI Frontend
+# Compatible with backend v1.8.0 (FAL open-world cue-rich → GPT-4o hard-discriminator labeling)
+# TR/EN UI • retries • warm-up • audio preview • artifact downloads
+
 import os
 import json
 import time
@@ -7,12 +10,12 @@ import streamlit as st
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
 
-# ---- container-safe defaults to avoid '/.streamlit' permission errors
+# ---- container-safe defaults
 os.environ.setdefault("HOME", "/tmp")
 os.environ.setdefault("STREAMLIT_BROWSER_GATHER_USAGE_STATS", "false")
 os.environ.setdefault("STREAMLIT_SERVER_HEADLESS", "true")
 
-# ================= Theme (no extra files) =================
+# ================= Theme =================
 try:
     st._config.set_option("theme.base", "dark")
     st._config.set_option("theme.primaryColor", "#ff7a1a")
@@ -26,6 +29,7 @@ except Exception:
 def _env(k: str) -> str:
     return (os.getenv(k) or "").strip().strip("'\"")
 
+# Allow reusing existing envs from your other apps
 API_BASE = (
     _env("CARDIAG_AI_API") or _env("AI_LIGHTBOX_API") or _env("LUXFIT_API")
 ).rstrip("/") if (_env("CARDIAG_AI_API") or _env("AI_LIGHTBOX_API") or _env("LUXFIT_API")) else ""
@@ -37,7 +41,7 @@ HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 L = {
     "en": {
         "title": "Diag Studio AI — Engine Sound Diagnosis",
-        "about": "Upload an engine sound (mp3/wav/ogg) or paste an audio URL. Backend runs fal.ai audio-understanding with a minimal prompt, then GPT-4o maps to labels and summarizes.",
+        "about": "Upload an engine sound (mp3/wav/ogg) or paste an audio URL. Backend extracts cue-rich acoustics with fal.ai, then GPT-4o applies HARD discriminators to avoid collapsing similar timbres.",
         "backend_url": "Backend URL",
         "lang": "Language",
         "upload": "Upload audio file",
@@ -46,7 +50,7 @@ L = {
         "adv": "Advanced",
         "prompt": "Override FAL prompt (optional)",
         "use_data_uri": "Send as data:URI (safer for private backends)",
-        "model": "OpenAI model for labeling/commentary",
+        "model": "OpenAI model (labeling/commentary)",
         "analyze": "Analyze",
         "reset": "Reset",
         "result": "Result",
@@ -60,23 +64,24 @@ L = {
         "health_ok": "Backend is reachable.",
         "health_fail": "Backend not reachable.",
         "upload_hint": "Max ~200 MB per Streamlit; use mp3/wav/ogg.",
-        "footer": "Tip: You can also paste a public audio URL if you don’t want to upload the file.",
+        "footer": "Tip: You can paste a public audio URL instead of uploading.",
         "auto_retry": "Auto-retry on 429/500/502/503/504",
         "warmup": "Warm up backend before analysis",
         "max_wait": "Max total wait (sec) for retries",
         "warming": "Warming up backend…",
         "analyzing": "Analyzing…",
-        "lang_help": "Output language for GPT-4o commentary",
-        "ok_false": "Request returned a controlled error from backend.",
-        "try_again": "Backend might be waking up or rate-limited. Try Analyze again or increase ‘Max total wait’.",
+        "lang_help": "Commentary language",
+        "ok_false": "Backend returned a controlled error.",
+        "try_again": "Service may be waking or rate-limited. Try again or increase ‘Max total wait’.",
         "preview": "Audio preview",
         "raw_text": "Raw model text",
         "prompt_used": "Prompt used (first 240 chars)",
-        "api_used": "Backend audio URL (truncated)"
+        "api_used": "Backend audio URL (truncated)",
+        "hard_note": "HARD discriminators active: half-crank vs full-crank, load vs RPM, cold-start confinement, tonal vs broadband, deep vs light."
     },
     "tr": {
         "title": "Diag Studio AI — Motor Sesi Teşhis",
-        "about": "Motor sesini (mp3/wav/ogg) yükleyin veya bir ses URL’si girin. Backend minimal bir prompt ile fal.ai ‘audio-understanding’ çalıştırır; GPT-4o etiketlere map eder ve özetler.",
+        "about": "Motor sesini (mp3/wav/ogg) yükleyin veya bir ses URL’si girin. Backend fal.ai ile ipucu-zengin akustik özet çıkarır; GPT-4o benzer tınıları karıştırmamak için SERT ayraçlar uygular.",
         "backend_url": "Backend URL",
         "lang": "Dil",
         "upload": "Ses dosyası yükle",
@@ -85,7 +90,7 @@ L = {
         "adv": "Gelişmiş",
         "prompt": "FAL prompt’u geçersiz kıl (opsiyonel)",
         "use_data_uri": "data:URI olarak gönder (özel backend için güvenli)",
-        "model": "OpenAI modeli (labeling/yorum)",
+        "model": "OpenAI modeli (label/yorum)",
         "analyze": "Analiz Et",
         "reset": "Sıfırla",
         "result": "Sonuç",
@@ -98,20 +103,21 @@ L = {
         "no_backend": "Bir backend URL girin (örn: http://127.0.0.1:7860)",
         "health_ok": "Backend erişilebilir.",
         "health_fail": "Backend’e erişilemedi.",
-        "upload_hint": "Streamlit varsayılan yükleme sınırı ~200 MB; mp3/wav/ogg kullanın.",
-        "footer": "İpucu: Dosya yüklemek istemezseniz herkese açık bir ses URL’si de girebilirsiniz.",
+        "upload_hint": "Streamlit yükleme sınırı ~200 MB; mp3/wav/ogg kullanın.",
+        "footer": "İpucu: Dosya yüklemek istemezseniz herkese açık bir ses URL’si girebilirsiniz.",
         "auto_retry": "429/500/502/503/504 için otomatik yeniden dene",
         "warmup": "Analizden önce backend’i ısıt",
         "max_wait": "Retry için azami bekleme (sn)",
         "warming": "Backend ısıtılıyor…",
         "analyzing": "Analiz ediliyor…",
-        "lang_help": "GPT-4o yorum çıktısının dili",
+        "lang_help": "Yorum dilini seçin",
         "ok_false": "Backend kontrollü bir hata döndürdü.",
-        "try_again": "Backend uyanıyor veya sınırlandı. Yeniden Dene’ye basın ya da ‘Azami bekleme’ değerini artırın.",
+        "try_again": "Servis uyanıyor veya sınırlandı. Tekrar deneyin veya ‘Azami bekleme’ süresini artırın.",
         "preview": "Ses önizleme",
         "raw_text": "Ham model metni",
         "prompt_used": "Kullanılan prompt (ilk 240 karakter)",
-        "api_used": "Backend ses URL’si (kısaltılmış)"
+        "api_used": "Backend ses URL’si (kısaltılmış)",
+        "hard_note": "SERT ayraçlar aktif: yarım-krank vs tam-krank, yük vs RPM, soğuk-başlangıç sınırı, tonal vs geniş bant, derin vs hafif."
     }
 }
 
@@ -122,7 +128,7 @@ if "last_response" not in st.session_state:
     st.session_state.last_response = None
 
 # ================= Header =================
-col1, col2 = st.columns([0.6, 0.4])
+col1, col2 = st.columns([0.65, 0.35])
 with col1:
     st.title(L[st.session_state.lang]["title"])
 with col2:
@@ -135,6 +141,7 @@ with col2:
 
 t = L[st.session_state.lang]
 st.caption(t["about"])
+st.info(t["hard_note"])
 
 backend_default = API_BASE or "http://127.0.0.1:7860"
 api_base = st.text_input(t["backend_url"], value=backend_default)
@@ -143,10 +150,10 @@ api_base = st.text_input(t["backend_url"], value=backend_default)
 with st.expander(t["adv"], expanded=False):
     use_data_uri = st.checkbox(t["use_data_uri"], value=True)
     openai_model = st.text_input(t["model"], value="gpt-4o")
-    prompt_override = st.text_area(t["prompt"], value="", height=120)
+    prompt_override = st.text_area(t["prompt"], value="", height=140, help="Leave empty to use backend’s cue-rich default")
     auto_retry = st.checkbox(t["auto_retry"], value=True)
     warmup = st.checkbox(t["warmup"], value=True)
-    max_wait = st.slider(t["max_wait"], 10, 180, 60, step=10)
+    max_wait = st.slider(t["max_wait"], 10, 240, 60, step=10)
     st.caption(t["footer"])
 
 st.markdown("---")
@@ -176,9 +183,7 @@ def build_session(_auto_retry: bool) -> requests.Session:
     s.headers.update(HEADERS)
     if _auto_retry:
         retry = Retry(
-            total=6,
-            connect=6,
-            read=6,
+            total=6, connect=6, read=6,
             backoff_factor=0.8,
             status_forcelist=[429, 500, 502, 503, 504],
             allowed_methods=frozenset(["GET", "POST"]),
@@ -215,7 +220,7 @@ def post_analyze(sess: requests.Session, base: str, _file, _audio_url: str, lang
         "use_data_uri": "true" if use_data_uri else "false",
         "openai_model": openai_model or "gpt-4o",
         "lang": lang,  # "tr" or "en"
-        # backend pins detailed_analysis=True internally; no need to send.
+        # backend pins detailed_analysis=True
     }
     if prompt_override.strip():
         data["prompt"] = prompt_override.strip()
@@ -231,7 +236,8 @@ def post_analyze(sess: requests.Session, base: str, _file, _audio_url: str, lang
         return {"ok": False, "error": "No file or URL provided."}
 
     deadline = time.time() + max_wait
-    attempts, last_err = 0, None
+    attempts = 0
+    last_err = None
     while True:
         attempts += 1
         try:
@@ -241,12 +247,7 @@ def post_analyze(sess: requests.Session, base: str, _file, _audio_url: str, lang
         except Exception as e:
             last_err = e
             if not auto_retry or time.time() > deadline:
-                return {
-                    "ok": False,
-                    "error": f"All attempts failed. Last error: {e}",
-                    "attempts": attempts,
-                    "api_base": base,
-                }
+                return {"ok": False, "error": f"All attempts failed. Last error: {e}", "attempts": attempts, "api_base": base}
             time.sleep(1.2)
 
 # ================= Actions =================
@@ -256,10 +257,9 @@ if health_btn:
     try:
         r = sess.get(f"{api_base.rstrip('/')}/health", timeout=10)
         st.success(t["health_ok"] if r.ok else t["health_fail"])
-        if r.ok:
+        if r.ok and r.headers.get("content-type","").startswith("application/json"):
             payload = r.json()
-            if isinstance(payload, dict):
-                st.code(json.dumps(payload, ensure_ascii=False, indent=2), language="json")
+            st.code(json.dumps(payload, ensure_ascii=False, indent=2), language="json")
     except Exception:
         st.error(t["health_fail"])
 
@@ -291,19 +291,19 @@ if res:
     else:
         st.subheader(t["result"])
 
-        # driver-friendly commentary
+        # Commentary
         comm = res.get("commentary") or ""
         if comm:
             st.markdown(f"**{t['commentary']}**")
             st.write(comm)
 
-        # parsed JSON
+        # Parsed JSON
         parsed = res.get("fal_parsed")
         if parsed:
             st.markdown(f"**{t['json']}**")
             st.json(parsed)
 
-        # prompt used and fal audio url
+        # Prompt used + fal audio url
         extras_col1, extras_col2 = st.columns(2)
         if res.get("prompt_used"):
             with extras_col1:
@@ -314,7 +314,7 @@ if res:
                 st.caption(t["api_used"])
                 st.code(res["fal_audio_url"], language="text")
 
-        # backend artifacts
+        # Backend artifacts
         st.markdown(f"**{t['backend_artifacts']}**")
         dls = res.get("downloads") or {}
         col_a, col_b, col_c = st.columns(3)
@@ -341,4 +341,4 @@ if res:
                 col_c.error("raw.txt ✖")
 
 # ===== Footer =====
-st.caption("© 2025 Diag Studio AI — Streamlit frontend (TR/EN, retries, warm-up, downloads)")
+st.caption("© 2025 Diag Studio AI — Streamlit frontend (TR/EN, retries, warm-up, cue-rich + hard-discriminator flow)")