Upload 2 files

Dockerfile

@@ -1,21 +1,33 @@
-# ---------- Public HF Space: Streamlit frontend ----------
+# ---------- Public HF Space: Streamlit frontend (DiagStudio AI) ----------
 FROM python:3.11-slim
 
+# System deps
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    ca-certificates \
- && rm -rf /var/lib/apt/lists/*
+    ca-certificates curl && \
+    rm -rf /var/lib/apt/lists/*
 
+# Prevent Python from writing .pyc; flush stdout
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1
+
+# Streamlit needs a writable HOME; disable usage stats; headless mode
+ENV HOME=/tmp \
+    STREAMLIT_BROWSER_GATHER_USAGE_STATS=false \
+    STREAMLIT_SERVER_HEADLESS=true
+
+# Workdir
 WORKDIR /app
 
+# Install deps first (better layer caching)
 COPY requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy frontend app
+# Copy app
 COPY app.py /app/app.py
 
-# Expose default Spaces port
+# Spaces Docker expects the app to listen on 7860
 EXPOSE 7860
 
-# Streamlit must bind to 0.0.0.0 and the provided port
-# Use ${PORT} if set by the platform, otherwise fall back to 7860.
+# Bind to 0.0.0.0 and honor $PORT if provided by platform
+# Using sh -c form to expand ${PORT}; recommended for passing env to CMD
 CMD ["sh", "-c", "streamlit run app.py --server.address=0.0.0.0 --server.port=${PORT:-7860}"]

app.py

@@ -1,13 +1,17 @@
-# app.py — DiagStudio AI (Streamlit frontend for public HF Space)
-# - Talks to your private FastAPI backend:
-#     POST {API_BASE}/analyze
-#     POST {API_BASE}/analyze_zip
-# - Friendly UI for non-technical users; no DSP jargon on the surface.
+# app.py — DiagStudio AI (Streamlit frontend, fixed)
+# Root-cause fix: prevent Streamlit from writing to '/.streamlit'
+# by defining a writable HOME and disabling usage stats before import.
 
 import os
+# ----- MUST be set BEFORE importing streamlit -----
+os.environ.setdefault("HOME", "/tmp")
+os.environ.setdefault("STREAMLIT_BROWSER_GATHER_USAGE_STATS", "false")
+os.environ.setdefault("STREAMLIT_SERVER_HEADLESS", "true")
+
 import io
 import json
 import time
+import base64
 import requests
 import streamlit as st
 
@@ -22,11 +26,11 @@ except Exception:
     pass
 
 # ================= Env & API =================
-def _env(k): 
+def _env(k):
     return (os.getenv(k) or "").strip().strip("'\"")
 
 API_BASE = (
-    _env("CARDIAG_AI_API") 
+    _env("CARDIAG_AI_API")
     or _env("AI_LIGHTBOX_API")
     or _env("LUXFIT_API")
 ).rstrip("/") if (
@@ -41,25 +45,19 @@ st.set_page_config(page_title="DiagStudio AI", layout="wide")
 st.title("DiagStudio AI")
 st.caption("Upload a short engine recording and get a clear, visual, and textual diagnosis report.")
 
-# Simple validation for API base
 if not API_BASE:
     st.error("Backend URL is not set. Define CARDIAG_AI_API (or AI_LIGHTBOX_API / LUXFIT_API).")
     st.stop()
 
-# Health check
+# ================= Helpers =================
 @st.cache_data(show_spinner=False, ttl=30)
-def check_health():
+def check_health(url: str, headers: dict):
     try:
-        r = requests.get(f"{API_BASE}/health", headers=HEADERS, timeout=10)
+        r = requests.get(f"{url}/health", headers=headers, timeout=10)
         return r.status_code == 200
     except Exception:
         return False
 
-ok = check_health()
-if not ok:
-    st.warning("Could not reach the backend. Please verify your private Space is running and accessible.")
-
-# ================= Helpers =================
 def _nice_conf(p):
     try:
         v = float(p)
@@ -83,8 +81,7 @@ def _post_analyze(audio_file, rpm_file, sr_target, n_fft, hop, env_band, run_llm
         "extra_text": notes or "",
     }
 
-    # Multipart form-data via requests.files -> backend expects UploadFile/File. :contentReference[oaicite:1]{index=1}
-    r = requests.post(f"{API_BASE}/analyze", headers=HEADERS, files=files, data=data, timeout=120)
+    r = requests.post(f"{API_BASE}/analyze", headers=HEADERS, files=files, data=data, timeout=180)
     r.raise_for_status()
     return r.json()
 
@@ -104,23 +101,19 @@ def _post_analyze_zip(audio_file, rpm_file, sr_target, n_fft, hop, env_band, run
         "extra_text": notes or "",
     }
 
-    r = requests.post(f"{API_BASE}/analyze_zip", headers=HEADERS, files=files, data=data, timeout=180, stream=True)
+    r = requests.post(f"{API_BASE}/analyze_zip", headers=HEADERS, files=files, data=data, timeout=240, stream=True)
     r.raise_for_status()
-    return r.content  # bytes for st.download_button. :contentReference[oaicite:2]{index=2}
+    return r.content
 
-def _dataurl_to_bytes(data_url: str) -> bytes:
+def dataurl_to_bytes(data_url: str) -> bytes:
     # "data:image/png;base64,...."
     b64 = data_url.split(",", 1)[1]
-    return base64_to_bytes(b64)
-
-def base64_to_bytes(b64: str) -> bytes:
-    import base64
     return base64.b64decode(b64)
 
 # ================= Sidebar =================
 with st.sidebar:
     st.header("Upload")
-    audio_up = st.file_uploader("Engine recording (WAV/MP3/M4A)", type=["wav", "mp3", "m4a"])  # st.file_uploader returns UploadedFile. :contentReference[oaicite:3]{index=3}
+    audio_up = st.file_uploader("Engine recording (WAV/MP3/M4A)", type=["wav", "mp3", "m4a"])
     rpm_up = st.file_uploader("Optional RPM CSV (time_sec,rpm)", type=["csv"])
 
     st.header("Options")
@@ -133,14 +126,18 @@ with st.sidebar:
         hop = st.select_slider("Hop length", options=[128, 256, 512, 1024], value=512)
         env_band = st.text_input("Impact band (Hz low,high)", value="1000,8000")
 
-    disabled = audio_up is None or not ok
+    ok = check_health(API_BASE, HEADERS)
+    if not ok:
+        st.warning("Could not reach the backend. Verify your private Space is running and reachable.")
+
+    disabled = (audio_up is None) or (not ok)
     analyze_btn = st.button("Analyze now", type="primary", disabled=disabled)
 
-# ================= Main UI =================
+# ================= Main =================
 status_ph = st.empty()
 
 if analyze_btn and audio_up is not None:
-    status_ph.info("Analyzing… This usually takes a few seconds.")
+    status_ph.info("Analyzing…")
     t0 = time.time()
     try:
         data = _post_analyze(
@@ -156,14 +153,11 @@ if analyze_btn and audio_up is not None:
     except Exception as e:
         status_ph.error(f"Request failed: {e}")
 
-# Show results if present
 res = st.session_state.get("last_result")
 if res:
-    # -------- Summary cards (user-friendly text) --------
     st.subheader("Result")
     col1, col2 = st.columns([1,1])
 
-    # AI diagnosis (JSON). We tolerate either dict or string.
     llm = res.get("llm_report") or {}
     if isinstance(llm, str):
         try:
@@ -197,7 +191,6 @@ if res:
             st.markdown("**Suggested next checks**")
             st.write(next_tests)
 
-    # -------- Visuals in tabs (friendly labels, no DSP jargon) --------
     st.subheader("Visual checks")
     tabs = st.tabs([
         "Energy over time",
@@ -206,7 +199,7 @@ if res:
         "Signal envelope",
         "Detail view",
         "Order view"
-    ])  # st.tabs API. :contentReference[oaicite:4]{index=4}
+    ])
 
     imgs = res.get("images", {})
 
@@ -238,7 +231,6 @@ if res:
 
     st.divider()
 
-    # -------- Download ZIP --------
     st.subheader("Download")
     st.caption("Get all visuals and the diagnosis as a single ZIP.")
     if st.button("Prepare ZIP"):
@@ -253,9 +245,8 @@ if res:
                 data=zbytes,
                 file_name="diagstudio_results.zip",
                 mime="application/zip"
-            )  # st.download_button for binary files. :contentReference[oaicite:5]{index=5}
+            )
         except Exception as e:
             st.error(f"ZIP build failed: {e}")
-
 else:
     st.info("Upload an engine recording, optionally add RPM CSV, then click Analyze.")