fix tts replay, self-rating button, phase0 prefill, tts cache

Author: srijabal

backend/app/agents/screening_logic.py

@@ -298,14 +298,14 @@ def choose_opening_question(worker_name: str, assignment: str, locale: str = "en
 PHASE0_FIELDS = ["name", "age", "sex", "address", "tradeRaw", "yearsExp"]
 
 
-def init_phase0_profile(worker_name: str, trade_raw: str) -> dict:
+def init_phase0_profile(worker_name: str, trade_raw: str, experience_years: int | None = None) -> dict:
     return {
         "name": worker_name.strip() if worker_name else "",
         "age": None,
         "sex": None,
         "address": None,
         "tradeRaw": trade_raw.strip() if trade_raw else "",
-        "yearsExp": None,
+        "yearsExp": experience_years if experience_years is not None else None,
     }
 
 

backend/app/api/routes/sessions.py

@@ -337,7 +337,7 @@ async def start_session(
         assignment = ASSIGNMENT_TEMPLATES.get(domain, DEFAULT_ASSIGNMENT)
 
     first_question = choose_opening_question(worker.name, assignment, locale, domain)
-    phase0_profile = init_phase0_profile(worker.name, worker.specialization)
+    phase0_profile = init_phase0_profile(worker.name, worker.specialization, worker.experience_years)
     phase0_completed = len(phase0_missing_fields(phase0_profile)) == 0
 
     session = Session(

backend/app/api/routes/speech.py

@@ -1,4 +1,5 @@
 import base64
+import hashlib
 
 import httpx
 from fastapi import APIRouter, File, HTTPException, UploadFile
@@ -9,6 +10,9 @@
 
 router = APIRouter(tags=["speech"])
 
+# In-memory TTS cache: (text, language) hash -> WAV bytes
+_tts_cache: dict[str, bytes] = {}
+
 @router.post("/speech/stt")
 async def speech_to_text(file: UploadFile = File(...)):
     audio_bytes = await file.read()
@@ -27,6 +31,10 @@ class TtsRequest(BaseModel):
 
 @router.post("/speech/tts")
 async def text_to_speech(payload: TtsRequest):
+    cache_key = hashlib.md5(f"{payload.text}|{payload.language}".encode()).hexdigest()
+    if cache_key in _tts_cache:
+        return Response(content=_tts_cache[cache_key], media_type="audio/wav")
+
     headers = sarvam_headers()
 
     async with httpx.AsyncClient(timeout=30) as client:
@@ -51,4 +59,5 @@ async def text_to_speech(payload: TtsRequest):
         raise HTTPException(status_code=502, detail="No audio returned from TTS")
 
     audio_bytes = base64.b64decode(audios[0])
+    _tts_cache[cache_key] = audio_bytes
     return Response(content=audio_bytes, media_type="audio/wav")

frontend/src/pages/ScreeningRoomPage.jsx

@@ -1616,6 +1616,15 @@ export default function ScreeningRoomPage() {
                 <p style={{ fontSize: "clamp(13px,1.6vw,15px)", color: "#71675d", lineHeight: 1.75, margin: 0, minHeight: 42 }}>
                   {currentQuestion || (session ? copy.waiting_ai : copy.setup_to_begin)}
                 </p>
+                {currentQuestion && !aiSpeaking && (
+                  <button
+                    onClick={() => speakAi(currentQuestion)}
+                    title="Replay question"
+                    style={{ marginTop: 6, background: "none", border: "none", cursor: "pointer", color: "rgba(59,130,246,0.7)", fontSize: 18, padding: "2px 6px" }}
+                  >
+                    🔊
+                  </button>
+                )}
                 {session && (
                   <p style={{ margin: "8px 0 0", fontSize: 11, color: "#64748b" }}>
                     {session.phase0_completed ? "Phase 0 intake complete" : "Phase 0 intake in progress"}
@@ -1685,7 +1694,7 @@ export default function ScreeningRoomPage() {
                   ))}
                 </div>
                 <div style={{ display: "flex", justifyContent: "flex-end", marginTop: 8 }}>
-                  <button onClick={saveSelfRatings} style={{ ...iconBtn(false, "#3b82f6"), padding: "6px 10px" }}>
+                  <button onClick={saveSelfRatings} style={{ background: "rgba(59,130,246,0.12)", color: "#1e3a8a", border: "1px solid rgba(59,130,246,0.25)", borderRadius: 8, padding: "6px 14px", fontSize: 12, fontWeight: 700, cursor: "pointer" }}>
                     {copy.self_rate_send}
                   </button>
                 </div>