langchain

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