fix

app.py

@@ -1,13 +1,9 @@
 import gradio as gr
 from transformers import pipeline
-import numpy as np
 import requests
 import subprocess
 import os
 import urllib.parse
-from moviepy.editor import *
-from ffmpy import FFmpeg
-import time
 
 term = 9
 
@@ -102,32 +98,71 @@ def extract_audio(video_path, audio_path):
         print("Audio extracted successfully.")
     else:
         print("Error extracting audio.")
- 
+
+def download_and_extract_audio(video_link, start_time, end_time, audio_output):
+    video_path = "temp_video.mp4"
+
+    # Downloading video
+    response = requests.get(video_link)
+    if response.status_code == 200:
+        with open(video_path, 'wb') as file:
+            file.write(response.content)
+    else:
+        print(f"Error downloading video: {response.status_code}")
+        return
+
+    # Extracting audio
+    command = [
+        'ffmpeg', '-i', video_path, '-ss', start_time, '-to', end_time, 
+        '-q:a', '0', '-map', 'a', audio_output, '-y'
+    ]
+    subprocess.run(command)
+
+    # Clean up downloaded video
+    if os.path.exists(video_path):
+        os.remove(video_path)
+
+    if os.path.exists(audio_output):
+        print("Audio extracted successfully.")
+    else:
+        print("Error extracting audio.")
+        
 # 600000,10000
 
 def transcribe(num1,num2):
 
+    num1,num2 = int(num1),int(num2)
+
     videos = get_sejm_videos(term)
     print(videos[0]['videoLink'])
 
-    if download_video(offset_time(videos[0]['videoLink'],num1,num2), "./video.mov"):
-        time.sleep(2)
-        extract_audio("./video.mov", "./audio.mp3")
-        # Load your MP4 file
+    start_time = "00:00:00"  # Start time of the segment
+    end_time = "01:00:00"   # End time of the segment
+    audio_output = "sejm_audio.mp3"
 
-        #ff = FFmpeg(
-        #    inputs={'./video.mp4': None},
-        #    outputs={'./audio.mp3': None}
-        #)
-        #ff.run()
-        #video = VideoFileClip("./video.mp4")
+    print(offset_time(videos[0]['videoLink'],num1,num2))
 
-        ## Extract the audio from the video
-        #audio = video.audio
+    download_and_extract_audio(offset_time(videos[0]['videoLink'],num1,num2), start_time, end_time, audio_output)
 
-        ## Write the audio to an MP3 file
-        #audio.write_audiofile("./audio.mp3")
-        return transcriber("./audio.mp3")["text"]
+    return transcriber(audio_output)["text"]
+    
+    # if download_video(offset_time(videos[0]['videoLink'],num1,num2), "./video.mov"):
+    #     extract_audio("./video.mov", "./audio.mp3")
+    #     # Load your MP4 file
+# 
+    #     #ff = FFmpeg(
+    #     #    inputs={'./video.mp4': None},
+    #     #    outputs={'./audio.mp3': None}
+    #     #)
+    #     #ff.run()
+    #     #video = VideoFileClip("./video.mp4")
+# 
+    #     ## Extract the audio from the video
+    #     #audio = video.audio
+# 
+    #     ## Write the audio to an MP3 file
+    #     #audio.write_audiofile("./audio.mp3")
+    #     return transcriber(audio_output)["text"]
 
 
 demo = gr.Interface(