Update app.py

app.py

@@ -10,17 +10,27 @@ ELEVENLABS_API_KEY = os.environ.get("ELEVENLABS_API_KEY", None)
 
 def download_video_from_url(url):
     try:
-        filename = f"downloaded_video_{uuid.uuid4().hex[:8]}.mp4"      
+        filename = f"downloaded_video_{uuid.uuid4().hex[:8]}.mp4"
+        
         ydl_opts = {
             'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/mp4',
             'outtmpl': filename,
             'quiet': True,
+            'no_warnings': True,
+            'ignoreerrors': True,
+            'cookiesfrombrowser': ('chrome',),  
+            'nocheckcertificate': True,
+            'age_limit': 99,  
         }
         
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             ydl.download([url])
         
-        return filename, None
+        if os.path.exists(filename) and os.path.getsize(filename) > 0:
+            return filename, None
+        else:
+            return None, "Failed to download video. Try providing a direct video URL instead of YouTube."
+            
     except Exception as e:
         return None, f"Error downloading video: {str(e)}"
 
@@ -32,7 +42,8 @@ def extract_audio(video_path, output_format="mp3"):
     
     try:
         video = VideoFileClip(video_path)
-        video.audio.write_audiofile(output_path, verbose=False, logger=None)
+        # Remove the verbose parameter that's causing issues
+        video.audio.write_audiofile(output_path, logger=None)
         video.close()
         return output_path, f"Audio extracted successfully"
     except Exception as e:
@@ -42,6 +53,7 @@ def save_transcription(transcription):
     if "error" in transcription:
         return None, transcription["error"]
     
+    # Create a filename for the transcription
     transcript_filename = f"transcription_{uuid.uuid4().hex[:8]}.txt"
     
     try:
@@ -68,11 +80,16 @@ def process_video_file(video_file, output_format, api_key, model_id):
 def process_video_url(video_url, output_format, api_key, model_id):
     if not video_url.strip():
         return None, "Please enter a video URL", None, "No URL provided"
+    
+    # Download the video
     video_path, error = download_video_from_url(video_url)
     if error:
         return None, error, None, "Video download failed, cannot transcribe"
     
+    # Extract audio from the downloaded video
     audio_path, message = extract_audio(video_path, output_format)
+    
+    # Clean up the downloaded video
     if video_path and os.path.exists(video_path):
         try:
             os.remove(video_path)
@@ -94,26 +111,24 @@ def transcribe_audio(audio_file, api_key, model_id="scribe_v1"):
     headers = {
         "xi-api-key": api_key
     }
-    files = {
-        "file": open(audio_file, "rb"),
-        "model_id": (None, model_id)
-    }
     
     try:
-        response = requests.post(url, headers=headers, files=files)
-        response.raise_for_status()
-        result = response.json()
+        with open(audio_file, "rb") as f:
+            files = {
+                "file": f,
+                "model_id": (None, model_id)
+            }
+            response = requests.post(url, headers=headers, files=files)
+            response.raise_for_status()
+            result = response.json()
         return result
     except requests.exceptions.RequestException as e:
         return {"error": f"API request failed: {str(e)}"}
     except json.JSONDecodeError:
         return {"error": "Failed to parse API response"}
-    finally:
-        files["file"].close()
 
 with gr.Blocks(title="Video to Audio to Transcription") as app:
     gr.Markdown("# Video => Audio => Transcription")
-    gr.Markdown("Upload a video or provide a URL to extract audio and transcribe it using ElevenLabs' API.")
     
     api_key = gr.Textbox(
         placeholder="Enter your ElevenLabs API key",