| import os |
| import json |
| import re |
|
|
| def annotate_fillerwords(session_id, base_dir="session_data"): |
| session_dir = os.path.join(base_dir, session_id) |
| json_file = os.path.join(session_dir, f"{session_id}_transcriptionCW.json") |
|
|
| if not os.path.exists(json_file): |
| print(f"[Error] File not found: {json_file}") |
| return |
|
|
| with open(json_file, "r", encoding="utf-8") as f: |
| data = json.load(f) |
|
|
| for segment in data.get("segments", []): |
| words = segment.get("words", []) |
| fillerwords = [] |
| for w in words: |
| word_content = w.get("word", "").strip() |
| if re.fullmatch(r"\[.*?\]", word_content): |
| fillerwords.append({ |
| "start": w.get("start"), |
| "end": w.get("end"), |
| "content": word_content, |
| "duration": round(w.get("end", 0) - w.get("start", 0), 3) |
| }) |
| segment["fillerwords"] = fillerwords |
|
|
| with open(json_file, "w", encoding="utf-8") as f: |
| json.dump(data, f, ensure_ascii=False, indent=4) |
|
|
| print(f"Session {session_id} fillerword annotation done: {json_file}") |
| return data |
|
|
| if __name__ == "__main__": |
| annotate_fillerwords("000003") |
|
|
