Update functions.py

functions.py

@@ -2,6 +2,7 @@ import whisper
 import os
 import random
 import openai
+from openai import OpenAI
 import yt_dlp
 from pytube import YouTube, extract
 import pandas as pd
@@ -21,8 +22,6 @@ from bs4 import BeautifulSoup
 import base64, time
 from annotated_text import annotated_text
 import pickle, math
-import wikipedia
-from pyvis.network import Network
 import torch
 from pydub import AudioSegment
 from langchain.docstore.document import Document
@@ -57,7 +56,7 @@ from langchain.prompts import PromptTemplate
 from langsmith import Client
 
 client = Client()
-
+openai_audio = OpenAI()
 nltk.download('punkt')
 
 
@@ -153,7 +152,7 @@ def load_whisper_api(audio):
 
     '''Transcribe YT audio to text using Open AI API'''
     file = open(audio, "rb")
-    transcript = openai.Audio.translate("whisper-1", file)
+    transcript = openai_audio.audio.transcriptions.create("whisper-2", file)
 
     return transcript