Initial commit

routers/transcribe/utterance.py

@@ -0,0 +1,42 @@
+from fastapi import APIRouter, Request
+import os
+import queue
+from dotenv import load_dotenv
+import json
+
+load_dotenv()
+
+router = APIRouter()
+
+@router.post("/handle_whisper_live_eos_utterance/{user_id}")
+async def handle_whisper_live_eos_utterance(user_id: str, request: Request):
+    data = await request.json()
+    utterance = data.get("utterance")
+    print(f"Utterance: {utterance}")
+    start = data.get("start")
+    end = data.get("end")
+    print(f"Start: {start}")
+    print(f"End: {end}")
+    eos = data.get("eos")
+    print(f"Eos: {eos}")
+    user_dir = f"../../data/users/{user_id}/transcripts"
+    if not os.path.exists(user_dir):
+        os.makedirs(user_dir)
+
+    log_file = os.path.join(user_dir, "utterances.log")
+    with open(log_file, "a") as f:
+        f.write(json.dumps(data) + "\n")
+
+    return {"message": "Utterance logged successfully"}
+
+@router.get("/get_utterances/{user_id}")
+async def get_utterances(user_id: str):
+    user_dir = f"../../data/users/{user_id}/transcripts"
+    log_file = os.path.join(user_dir, "utterances.log")
+    if not os.path.exists(log_file):
+        return {"utterances": []}
+
+    with open(log_file, "r") as f:
+        utterances = [json.loads(line) for line in f]
+
+    return {"utterances": utterances}