Update backend/mongodb/operations/calls.py

backend/mongodb/operations/calls.py

@@ -20,13 +20,13 @@ def list_calls(collection, limit: int):
 
 
 '''Finding calls based on call id'''
-def find_calls(collection, call_id: str):
+def find_call(collection, call_id: str):
     user_calls = collection.find_one({"call_id": call_id})
     if user_calls is not None:
         return user_calls
     else:
         raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Call with ID: '{call_id}' not found.")
-    
+
 
 '''Finding calls based on user id'''
 def find_user_calls(collection, user_id: str):
@@ -94,10 +94,10 @@ def update_calls(collection, call_id: str, calls: UpdateCall = Body(...)):
 
 def update_captions(collection, call_id: str, calls: UserCaptions = Body(...)):
     # calls = {k: v for k, v in calls.model_dump().items() if v is not None}
-    calls = {k: v for k, v in calls.items() if v is not None}
+    captions = {k: v for k, v in calls.items() if v is not None}
     if len(calls) >= 1:
         update_result = collection.update_one({"call_id": call_id},
-                                              {"$push": {"captions": calls}})
+                                              {"$push": {"captions": captions}})
 
         if update_result.modified_count == 0:
             raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Captions not updated!")
@@ -115,3 +115,17 @@ def delete_calls(collection, call_id: str):
         return f"Call deleted sucessfully!"
 
     raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"Call not found!")
+
+
+def get_caption_text(collection, call_id):
+    call_record = find_call((collection), call_id)
+
+    try: # Check if call has any captions first
+        caption_records = call_record['captions']
+    except KeyError:
+        return None
+
+    # iterate through caption embedded document and store original text
+    combined_text = [caption['original_text'] for caption in caption_records]
+
+    return " ".join(combined_text)