Create add and remove url endpoints (#263)

* create add and remove url endpoints

* add get_url + update endpoints

src/wordcab_transcribe/models.py

@@ -23,7 +23,7 @@
 from typing import List, Literal, NamedTuple, Optional, Union
 
 from faster_whisper.transcribe import Segment
-from pydantic import BaseModel, field_validator
+from pydantic import BaseModel, HttpUrl, field_validator
 from tensorshare import TensorShare
 
 
@@ -484,6 +484,13 @@ class Config:
         }
 
 
+class UrlSchema(BaseModel):
+    """Request model for the add_url endpoint."""
+
+    task: Literal["transcription", "diarization"]
+    url: HttpUrl
+
+
 class DiarizationSegment(NamedTuple):
     """Diarization segment model for the API."""
 

src/wordcab_transcribe/router/v1/endpoints.py

@@ -30,6 +30,9 @@
 )
 from wordcab_transcribe.router.v1.diarize_endpoint import router as diarize_router
 from wordcab_transcribe.router.v1.live_endpoint import router as live_router
+from wordcab_transcribe.router.v1.manage_remote_url import (
+    router as manage_remote_url_router,
+)
 from wordcab_transcribe.router.v1.transcribe_endpoint import router as transcribe_router
 from wordcab_transcribe.router.v1.youtube_endpoint import router as youtube_router
 
@@ -43,10 +46,16 @@
 live_routers = (live_router, "/live", "live")
 transcribe_routers = (transcribe_router, "/transcribe", "transcription")
 diarize_routers = (diarize_router, "/diarize", "diarization")
+manage_remote_url_routers = (
+    manage_remote_url_router,
+    "/url",
+    "remote-url",
+)
 
 routers = []
 if settings.asr_type == "async":
     routers.extend(async_routers)
+    routers.append(manage_remote_url_routers)
 elif settings.asr_type == "live":
     routers.append(live_routers)
 elif settings.asr_type == "only_transcription":

src/wordcab_transcribe/router/v1/manage_remote_url.py

@@ -0,0 +1,97 @@
+# Copyright 2023 The Wordcab Team. All rights reserved.
+#
+# Licensed under the Wordcab Transcribe License 0.1 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://github.com/Wordcab/wordcab-transcribe/blob/main/LICENSE
+#
+# Except as expressly provided otherwise herein, and to the fullest
+# extent permitted by law, Licensor provides the Software (and each
+# Contributor provides its Contributions) AS IS, and Licensor
+# disclaims all warranties or guarantees of any kind, express or
+# implied, whether arising under any law or from any usage in trade,
+# or otherwise including but not limited to the implied warranties
+# of merchantability, non-infringement, quiet enjoyment, fitness
+# for a particular purpose, or otherwise.
+#
+# See the License for the specific language governing permissions
+# and limitations under the License.
+"""Add Remote URL endpoint for remote transcription or diarization."""
+
+from typing import List, Union
+
+from fastapi import APIRouter, HTTPException
+from fastapi import status as http_status
+from loguru import logger
+from pydantic import HttpUrl
+from typing_extensions import Literal
+
+from wordcab_transcribe.dependencies import asr
+from wordcab_transcribe.models import UrlSchema
+from wordcab_transcribe.services.asr_service import ExceptionSource, ProcessException
+
+router = APIRouter()
+
+
+@router.get(
+    "",
+    response_model=Union[List[HttpUrl], str],
+    status_code=http_status.HTTP_200_OK,
+)
+async def get_url(task: Literal["transcription", "diarization"]) -> List[HttpUrl]:
+    """Get Remote URL endpoint for remote transcription or diarization."""
+    result: List[UrlSchema] = await asr.get_url(task)
+
+    if isinstance(result, ProcessException):
+        logger.error(result.message)
+        if result.source == ExceptionSource.get_url:
+            raise HTTPException(
+                status_code=http_status.HTTP_405_METHOD_NOT_ALLOWED,
+                detail=str(result.message),
+            )
+        else:
+            raise HTTPException(
+                status_code=http_status.HTTP_500_INTERNAL_SERVER_ERROR,
+                detail=str(result.message),
+            )
+
+    return result
+
+
+@router.post(
+    "/add",
+    response_model=Union[UrlSchema, str],
+    status_code=http_status.HTTP_200_OK,
+)
+async def add_url(data: UrlSchema) -> UrlSchema:
+    """Add Remote URL endpoint for remote transcription or diarization."""
+    result: UrlSchema = await asr.add_url(data)
+
+    if isinstance(result, ProcessException):
+        logger.error(result.message)
+        raise HTTPException(
+            status_code=http_status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=str(result.message),
+        )
+
+    return result
+
+
+@router.post(
+    "/remove",
+    response_model=Union[UrlSchema, str],
+    status_code=http_status.HTTP_200_OK,
+)
+async def remove_url(data: UrlSchema) -> UrlSchema:
+    """Remove Remote URL endpoint for remote transcription or diarization."""
+    result: UrlSchema = await asr.remove_url(data)
+
+    if isinstance(result, ProcessException):
+        logger.error(result.message)
+        raise HTTPException(
+            status_code=http_status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=str(result.message),
+        )
+
+    return result