Update app.py

app.py

@@ -3,22 +3,26 @@ from youtube_transcript_api import YouTubeTranscriptApi
 import re
 from dotenv import load_dotenv
 import os
-from openai import OpenAI
+import openai
 import tempfile
 from pydub import AudioSegment
 import io
 
-# Load the API keys from .env file
+# Load environment variables
 load_dotenv()
-groq_api_key = os.getenv('GROQ_API_KEY')
-openai_api_key = os.getenv('OPENAI_API_KEY')
 
-# Initialize clients
-groq_client = Groq(api_key=groq_api_key)
-openai_client = OpenAI(api_key=openai_api_key)
+# Set up OpenAI API key
+openai.api_key = os.getenv('OPENAI_API_KEY')
 
-if not groq_api_key or not openai_api_key:
-    raise ValueError("API keys are not set. Please check your .env file and ensure GROQ_API_KEY and OPENAI_API_KEY are set.")
+if not openai.api_key:
+    raise ValueError("API key is not set. Please check your .env file and ensure OPENAI_API_KEY is set.")
+
+def convert_to_supported_format(file):
+    audio = AudioSegment.from_file(file)
+    buffer = io.BytesIO()
+    audio.export(buffer, format="mp3")
+    buffer.seek(0)
+    return buffer
 
 def get_transcript(url):
     try:
@@ -33,13 +37,6 @@ def get_transcript(url):
     except Exception as e:
         return f"Error: {str(e)}"
 
-def convert_to_supported_format(file):
-    audio = AudioSegment.from_file(file)
-    buffer = io.BytesIO()
-    audio.export(buffer, format="mp3")
-    buffer.seek(0)
-    return buffer
-
 def transcribe_audio(file):
     file = convert_to_supported_format(file)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as temp_file:
@@ -48,11 +45,8 @@ def transcribe_audio(file):
 
     try:
         with open(temp_file_path, "rb") as audio_file:
-            transcript = openai_client.audio.transcriptions.create(
-                model="whisper-1", 
-                file=audio_file
-            )
-        return transcript.text
+            transcript = openai.Audio.transcribe("whisper-1", audio_file)
+        return transcript["text"]
     except Exception as e:
         return f"Error in transcription: {str(e)}"
     finally:
@@ -60,8 +54,8 @@ def transcribe_audio(file):
 
 def answer_question(transcript, question):
     try:
-        response = groq_client.chat.completions.create(
-            model="mixtral-8x7b-32768",
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant."},
                 {"role": "user", "content": f"Using the following transcript as context, please answer the question:\n\nTranscript:\n{transcript}\n\nQuestion:\n{question}"}