Update app.py

app.py

@@ -13,13 +13,17 @@ from dash.exceptions import PreventUpdate
 import requests
 from pytube import YouTube
 from pydub import AudioSegment
-import openai
+import google.generativeai as genai
 
 # Initialize the Dash app
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 
-# Retrieve the OpenAI API key from Hugging Face Spaces
-openai.api_key = os.environ.get("OPENAI_API_KEY")
+# Retrieve the Google API key from Hugging Face Spaces
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
+genai.configure(api_key=GOOGLE_API_KEY)
+
+# Initialize Gemini model
+model = genai.GenerativeModel('gemini-2.0-flash-lite')
 
 def is_valid_url(url):
     try:
@@ -43,8 +47,10 @@ def download_audio(url):
 
 def transcribe_audio(file_path):
     with open(file_path, "rb") as audio_file:
-        transcript = openai.Audio.transcribe("whisper-1", audio_file)
-    return transcript["text"]
+        audio_data = audio_file.read()
+    
+    response = model.generate_content(audio_data)
+    return response.text
 
 def process_audio(contents, filename, url):
     if contents: