diff options
Diffstat (limited to 'call_transcript_utils.py')
-rw-r--r-- | call_transcript_utils.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/call_transcript_utils.py b/call_transcript_utils.py new file mode 100644 index 0000000..ce0764c --- /dev/null +++ b/call_transcript_utils.py @@ -0,0 +1,32 @@ +import os +from typing import Optional + +CALL_TRANSCRIPTS_DIR = os.path.join(os.path.dirname(__file__), "call_transcripts") + + +def add_transcript(conversation_id: str, transcript: str) -> None: + transcript_path = os.path.join( + CALL_TRANSCRIPTS_DIR, "{}.txt".format(conversation_id) + ) + with open(transcript_path, "a") as f: + f.write(transcript) + + +def get_transcript(conversation_id: str) -> Optional[str]: + transcript_path = os.path.join( + CALL_TRANSCRIPTS_DIR, "{}.txt".format(conversation_id) + ) + if os.path.exists(transcript_path): + with open(transcript_path, "r") as f: + return f.read() + return None + + +def delete_transcript(conversation_id: str) -> bool: + transcript_path = os.path.join( + CALL_TRANSCRIPTS_DIR, "{}.txt".format(conversation_id) + ) + if os.path.exists(transcript_path): + os.remove(transcript_path) + return True + return False |