File size: 1,241 Bytes
67d6834 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
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")
|