Update app.py

app.py

@@ -41,20 +41,28 @@ def download_youtube_audio(url):
         }],
         'outtmpl': output_path
     }
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        ydl.download([url])
-    
-    if os.path.exists(output_path):
-        print(f"Audio download completed. File saved at: {output_path}")
-        print(f"File size: {os.path.getsize(output_path)} bytes")
-    else:
-        print(f"Error: File {output_path} not found after download.")
-    
-    return output_path
+    try:
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            ydl.download([url])
+        
+        if os.path.exists(output_path):
+            print(f"Audio download completed. File saved at: {output_path}")
+            print(f"File size: {os.path.getsize(output_path)} bytes")
+            return output_path
+        else:
+            print(f"Error: File {output_path} not found after download.")
+            return None
+    except Exception as e:
+        print(f"Error downloading YouTube audio: {e}")
+        return None
 
 @spaces.GPU(duration=60)
 def transcribe_audio(file_path):
     print(f"Starting transcription of file: {file_path}")
+    if not os.path.exists(file_path):
+        print(f"Error: File {file_path} does not exist.")
+        return None
+
     if file_path.endswith(('.mp4', '.avi', '.mov', '.flv')):
         print("Video file detected. Extracting audio...")
         try:
@@ -65,8 +73,8 @@ def transcribe_audio(file_path):
             file_path = audio_path
         except Exception as e:
             print(f"Error extracting audio from video: {e}")
-            raise
-    
+            return None
+
     output_file = generate_unique_filename('.json')
     command = [
         "insanely-fast-whisper",
@@ -80,25 +88,33 @@ def transcribe_audio(file_path):
     print(f"Executing command: {' '.join(command)}")
     try:
         result = subprocess.run(command, check=True, capture_output=True, text=True)
+        print(f"Standard output: {result.stdout}")
+        print(f"Error output: {result.stderr}")
     except subprocess.CalledProcessError as e:
         print(f"Error running insanely-fast-whisper: {e}")
-        raise
-    
+        print(f"Standard output: {e.stdout}")
+        print(f"Error output: {e.stderr}")
+        cleanup_file(file_path)
+        cleanup_file(output_file)
+        return None
+
     try:
         with open(output_file, "r") as f:
             transcription = json.load(f)
     except json.JSONDecodeError as e:
         print(f"Error decoding JSON: {e}")
-        raise
-    
+        cleanup_file(file_path)
+        cleanup_file(output_file)
+        return None
+
     if "text" in transcription:
         result = transcription["text"]
     else:
         result = " ".join([chunk["text"] for chunk in transcription.get("chunks", [])])
-    
+
     cleanup_file(file_path)
     cleanup_file(output_file)
-    
+
     return result
 
 @spaces.GPU(duration=60)
@@ -121,16 +137,23 @@ def process_youtube(url):
         return "Please enter a YouTube URL.", None
     try:
         audio_file = download_youtube_audio(url)
+        if audio_file is None:
+            return "Error downloading YouTube audio.", None
         transcription = transcribe_audio(audio_file)
+        if transcription is None:
+            return "Error transcribing audio.", None
         return transcription, None
     except Exception as e:
         return f"Processing error: {str(e)}", None
     finally:
-        cleanup_file(audio_file)
+        if audio_file:
+            cleanup_file(audio_file)
 
 def process_uploaded_video(video_path):
     try:
         transcription = transcribe_audio(video_path)
+        if transcription is None:
+            return "Error transcribing video.", None
         return transcription, None
     except Exception as e:
         return f"Processing error: {str(e)}", None