Update app.py

app.py

@@ -40,11 +40,16 @@ def download_youtube_audio(url):
             'key': 'FFmpegExtractAudio',
             'preferredcodec': 'wav',
         }],
-        'outtmpl': output_path
+        'outtmpl': output_path,
+        'keepvideo': True,
     }
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         ydl.download([url])
     
+    # Check if the file was renamed to .wav.wav
+    if os.path.exists(output_path + ".wav"):
+        os.rename(output_path + ".wav", output_path)
+    
     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")
@@ -109,7 +114,9 @@ def transcribe_audio(file_path):
     print("Transcription completed.")
     
     # Cleanup
-    cleanup_files(temp_audio, output_file)
+    cleanup_files(output_file)
+    if temp_audio:
+        cleanup_files(temp_audio)
     
     return result
 
@@ -137,6 +144,7 @@ def process_youtube(url):
         return "Please enter a YouTube URL.", None
     print(f"Processing YouTube URL: {url}")
     
+    audio_file = None
     try:
         audio_file = download_youtube_audio(url)
         if not os.path.exists(audio_file):
@@ -151,7 +159,8 @@ def process_youtube(url):
         print(f"Error processing YouTube: {e}")
         return f"Processing error: {str(e)}", None
     finally:
-        cleanup_files(audio_file)
+        if audio_file and os.path.exists(audio_file):
+            cleanup_files(audio_file)
         print(f"Directory content after processing: {os.listdir('.')}")
 
 def process_uploaded_video(video_path):
@@ -164,8 +173,6 @@ def process_uploaded_video(video_path):
     except Exception as e:
         print(f"Error processing video: {e}")
         return f"Processing error: {str(e)}", None
-    finally:
-        cleanup_files(video_path)
 
 print("Setting up Gradio interface...")
 with gr.Blocks(theme=gr.themes.Soft()) as demo: