bluenevus commited on
Commit
953582f
·
verified ·
1 Parent(s): 0af167d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -1
app.py CHANGED
@@ -15,7 +15,21 @@ import requests
15
  from pytube import YouTube
16
  from pydub import AudioSegment
17
  import google.generativeai as genai
18
- from moviepy.editor import VideoFileClip
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
 
20
  # Set up logging
21
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -73,6 +87,8 @@ def download_media(url):
73
  def extract_audio(file_path):
74
  logger.info(f"Extracting audio from video: {file_path}")
75
  try:
 
 
76
  video = VideoFileClip(file_path)
77
  audio = video.audio
78
  audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
 
15
  from pytube import YouTube
16
  from pydub import AudioSegment
17
  import google.generativeai as genai
18
+
19
+ # Try different import statements for moviepy
20
+ try:
21
+ from moviepy.editor import VideoFileClip
22
+ except ImportError:
23
+ try:
24
+ import moviepy.editor as mpy
25
+ VideoFileClip = mpy.VideoFileClip
26
+ except ImportError:
27
+ try:
28
+ import moviepy
29
+ VideoFileClip = moviepy.VideoFileClip
30
+ except ImportError:
31
+ logging.error("Failed to import VideoFileClip from moviepy. Please check the installation.")
32
+ VideoFileClip = None
33
 
34
  # Set up logging
35
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
87
  def extract_audio(file_path):
88
  logger.info(f"Extracting audio from video: {file_path}")
89
  try:
90
+ if VideoFileClip is None:
91
+ raise ImportError("VideoFileClip is not available. Cannot extract audio.")
92
  video = VideoFileClip(file_path)
93
  audio = video.audio
94
  audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")