ExternalAPIs.py

import openai
import os
from youtube_transcript_api import YouTubeTranscriptApi


class ExternalAPIs():

    @staticmethod
    def getCaptions(url):
        # parse url to get video key
        videoKey = url.split("/watch?v=")[1]
        # of dictionaries obtained by the get_transcript() function

        try:
            srt = YouTubeTranscriptApi.get_transcript(videoKey)
        except Exception as e:
            return "Sorry, this video's captions are not available to us."

        captions = ""
        for dic in srt:
            captions = captions + " " + dic.get("text", "")
        return captions

    def getGPT(prompt):
        openai.api_key = os.environ['GPT_API_KEY']
        messages = [
            {"role": "system", "content": "You are a helpful assistant."},
        ]

        while True:
            messages.append(
                {"role": "user", "content": prompt},
            )
            try:
                chat_completion = openai.ChatCompletion.create(
                    model="gpt-3.5-turbo",
                    messages=messages
                )
            except Exception as e:
                return "Sorry, ChatGPT is having issues."
            answer = chat_completion.choices[0].message.content
            print(f"ChatGPT: {answer}")
            messages.append({"role": "assistant", "content": answer})
            return answer