benjolo commited on
Commit
89749e7
·
verified ·
1 Parent(s): 000aa13

adding caption operations

Browse files
Files changed (1) hide show
  1. backend/mongodb/endpoints/calls.py +8 -4
backend/mongodb/endpoints/calls.py CHANGED
@@ -1,4 +1,4 @@
1
- from fastapi import APIRouter, Body, Request, status
2
  from typing import List
3
  from datetime import date
4
 
@@ -28,15 +28,20 @@ async def list_calls(request: Request, limit: int):
28
  return calls.list_calls(collection, 100)
29
 
30
  @router.get("/find-call/{call_id}", response_description="Find user's calls based on User ID", response_model=UserCall)
31
- async def find_calls(request: Request, call_id: str):
32
  collection = get_collection_calls(request)
33
- return calls.find_calls(collection, call_id)
34
 
35
  @router.get("/find-user/{user_id}", response_description="Find user's calls based on User ID", response_model=List[UserCall])
36
  async def find_user_calls(request: Request, user_id: str):
37
  collection = get_collection_calls(request)
38
  return calls.find_user_calls(collection, user_id)
39
 
 
 
 
 
 
40
  '''Key terms list can have variable length -> using POST request over GET'''
41
  @router.post("/find-term/", response_description="Find calls based on key term list", response_model=List[UserCall])
42
  async def list_transcripts_by_key_terms(request: Request, key_terms: List[str]):
@@ -67,4 +72,3 @@ async def update_captions(request: Request, call_id: str, user_calls: UpdateCall
67
  async def delete_call(request: Request, call_id :str):
68
  collection = get_collection_calls(request)
69
  return calls.delete_calls(collection, call_id)
70
-
 
1
+ from fastapi import APIRouter, Body, Request, status, HTTPException
2
  from typing import List
3
  from datetime import date
4
 
 
28
  return calls.list_calls(collection, 100)
29
 
30
  @router.get("/find-call/{call_id}", response_description="Find user's calls based on User ID", response_model=UserCall)
31
+ async def find_call(request: Request, call_id: str):
32
  collection = get_collection_calls(request)
33
+ return calls.find_call(collection, call_id)
34
 
35
  @router.get("/find-user/{user_id}", response_description="Find user's calls based on User ID", response_model=List[UserCall])
36
  async def find_user_calls(request: Request, user_id: str):
37
  collection = get_collection_calls(request)
38
  return calls.find_user_calls(collection, user_id)
39
 
40
+ @router.get("/get-captions/{user_id}", response_description="Find user's calls based on User ID")
41
+ async def get_caption_text(request: Request, user_id: str):
42
+ collection = get_collection_calls(request)
43
+ return calls.get_caption_text(collection, user_id)
44
+
45
  '''Key terms list can have variable length -> using POST request over GET'''
46
  @router.post("/find-term/", response_description="Find calls based on key term list", response_model=List[UserCall])
47
  async def list_transcripts_by_key_terms(request: Request, key_terms: List[str]):
 
72
  async def delete_call(request: Request, call_id :str):
73
  collection = get_collection_calls(request)
74
  return calls.delete_calls(collection, call_id)