server.py

from flask import Flask, request
from get_transcript import id_to_transcript
from get_chat import id_to_chat, id_to_chat_split
from magic import magic
from get_video import id_to_name
from match_chat import match_chat
from chat_downloader.errors import NoChatReplay, ChatDownloaderError
from youtube_transcript_api._errors import TranscriptsDisabled, CouldNotRetrieveTranscript
from convert_timestamp import timestamp_to_seconds
from errors import *

APP = Flask(__name__, static_folder="./build", static_url_path="/")

@APP.errorhandler(APIError)
def handle_exception(err):
    response = {"error": err.description}
    response["message"] = "" if len(err.args) == 0 else err.args[0]
    APP.logger.error(f"{err.description}: {response['message']}")
    return response, err.code

@APP.route("/")
def index():
    return APP.send_static_file("index.html")

@APP.route("/transcript", methods=["get"])
def get_transcript_from_id():
    video_id = request.args.get("id")

    transcript = id_to_transcript(video_id)

    transcriptString = "\n".join(transcript)

    return {
        "transcript": transcriptString
    } 

@APP.route("/title", methods=["get"])
def get_title():
    video_id = request.args.get("id")

    try:
        name = id_to_name(video_id)
    except CouldNotGetName:
        raise APIVideoError("Could not retrieve this video. Please check that the link is correct.")
    except:
        raise APIError("Could not retrieve this video.")

    return {
        "name": name
    }

@APP.route("/magic", methods=["get"])
def get_magic():
    video_id = request.args.get("id")
    try:
        transcript = id_to_transcript(video_id)
        live_chat = id_to_chat_split(video_id)
        qna = magic(transcript, live_chat)
        qna.extend(match_chat(live_chat))
        qna = sorted(qna, key=lambda k: timestamp_to_seconds(k['time']))
        if (len(qna) == 0):
            raise NoQuestionsFound("No questions found")
    except TranscriptsDisabled:
        raise APITranscriptError("Transcripts have been disabled on this video.")
    except CouldNotRetrieveTranscript as e:
        raise APITranscriptError(f"Transcript could not be retrieved from this video: {e}")
    except NoChatReplay:
        raise APIChatError("This video does not have a chat replay.")
    except ChatDownloaderError:
        raise APIChatError("Chat could not be retrieved from this video.")
    except NoQuestionsFound:
        raise APIError("No questions found for this video.")
    except:
        raise APIError("Could not retrieve video details.")

    return {
        "qna": qna
    }