add support for fastapi

- support for fastapi
- first endpoint

Author: kouloumos

README.md

@@ -144,6 +144,23 @@ mp3_link="https://anchor.fm/s/7d083a4/podcast/play/64348045/https%3A%2F%2Fd3ctxl
 tstbtc transcribe $mp3_link --loc "stephan-livera-podcast" --title 'SLP455 Anant Tapadia - Single Sig or Multi Sig?' --date '2023-02-01' --tags 'multisig' --speakers 'Anant Tapadia' --speakers 'Stephan Livera' --category 'podcast'
 ```
 
+## Running as a Server
+
+In addition to running as a CLI application, you can also run tstbtc as a server.
+This allows you to interact with the transcription functionalities via HTTP requests.
+
+This is WIP
+
+### Starting the Server
+
+To start the server, navigate to the application directory and run:
+
+```sh
+fastapi run server.py
+```
+
+The server will be accessible at http://localhost:8000.
+
 ## Testing
 
 To run the unit tests

app/media_processor.py

@@ -2,6 +2,7 @@
 import soundfile as sf
 import os
 import ffmpeg
+import yt_dlp
 
 from app import (
     logging,
@@ -86,3 +87,23 @@ def convert_to_mp3(self, input_path, output_path=None):
         except ffmpeg.Error as e:
             logger.error(f"Error converting {input_path} to mp3: {e}")
             raise Exception(f"Error converting {input_path} to mp3: {e}")
+
+    def get_youtube_video_url(self, youtube_url):
+        """
+        Extracts and returns the direct URL of a YouTube video, allowing it
+        to be played directly without going through the YouTube platform.
+        """
+        ydl_opts = {
+            'format': 'best',
+            'quiet': True,
+            'no_warnings': True,
+        }
+
+        try:
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                info_dict = ydl.extract_info(youtube_url, download=False)
+                video_url = info_dict.get("url", None)
+                return video_url
+        except Exception as e:
+            logger.error(f"Error extracting video URL: {e}")
+            return None

requirements.txt

@@ -15,4 +15,5 @@ flake8==6.0.0
 isort==5.12.0
 feedparser==6.0.10
 PyYAML==6.0.1
-librosa==0.10.2.post1
+librosa==0.10.2.post1
+fastapi==0.111.0

server.py

@@ -0,0 +1,44 @@
+import logging
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+
+from app.logging import configure_logger, get_logger
+from app.media_processor import MediaProcessor
+
+logger = get_logger()
+app = FastAPI()
+
+origins = [
+    "http://localhost:3000",
+    "https://staging-review.btctranscripts.com/",
+    "https://review.btctranscripts.com/"
+]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all methods
+    allow_headers=["*"],  # Allows all headers
+)
+
+class VideoURLRequest(BaseModel):
+    youtube_url: str
+
+@app.post("/media/youtube-video-url")
+async def youtube_video_url(request: VideoURLRequest):
+    """Extract the direct video URL from a YouTube link"""
+    try:
+        configure_logger(log_level=logging.INFO)
+        processor = MediaProcessor()
+        video_url = processor.get_youtube_video_url(request.youtube_url)
+        if video_url:
+            return {"status": "success", "video_url": video_url}
+        else:
+            error_msg = f"Failed to extract the video URL for {request.youtube_url}."
+            logger.error(error_msg)
+            raise HTTPException(status_code=500, detail=error_msg)
+    except Exception as e:
+        logger.error(f"Error extracting video URL: {e}")
+        raise HTTPException(status_code=500, detail=str(e))