Update app.py

app.py

@@ -100,10 +100,13 @@ def process_media(file_path, is_url=False):
         if is_url:
             logger.info(f"Processing URL: {file_path}")
             try:
-                response = requests.get(file_path)
+                response = requests.get(file_path, stream=True)
                 response.raise_for_status()
-                temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
-                temp_file.write(response.content)
+                content_type = response.headers.get('content-type', '')
+                extension = mimetypes.guess_extension(content_type) or ''
+                temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=extension)
+                for chunk in response.iter_content(chunk_size=8192):
+                    temp_file.write(chunk)
                 temp_file.close()
                 logger.info(f"URL content downloaded: {temp_file.name}")
             except Exception as e:
@@ -116,24 +119,22 @@ def process_media(file_path, is_url=False):
             temp_file.close()
             logger.info(f"Uploaded file saved: {temp_file.name}")
 
-        file_extension = os.path.splitext(temp_file.name)[1].lower()
-        logger.info(f"Detected file extension: {file_extension}")
-        
-        if file_extension in VIDEO_FORMATS:
-            logger.info("Processing video file")
-            video = VideoFileClip(temp_file.name)
-            audio = video.audio
+        # Try to extract audio using moviepy
+        try:
+            audio = AudioFileClip(temp_file.name)
             wav_path = temp_file.name + ".wav"
             audio.write_audiofile(wav_path)
-            video.close()
-        elif file_extension in AUDIO_FORMATS:
-            logger.info("Processing audio file")
-            audio = AudioSegment.from_file(temp_file.name, format=file_extension[1:])
-            wav_path = temp_file.name + ".wav"
-            audio.export(wav_path, format="wav")
-        else:
-            logger.error(f"Unsupported file format: {file_extension}")
-            return f"Unsupported file format: {file_extension}. Please upload a supported audio or video file.", False
+            audio.close()
+        except Exception as e:
+            logger.warning(f"Could not process with moviepy: {str(e)}")
+            # If moviepy fails, try with pydub
+            try:
+                audio = AudioSegment.from_file(temp_file.name)
+                wav_path = temp_file.name + ".wav"
+                audio.export(wav_path, format="wav")
+            except Exception as e:
+                logger.error(f"Could not process audio: {str(e)}")
+                return f"Could not process audio: {str(e)}", False
 
         logger.info(f"Audio extracted to WAV: {wav_path}")