Update app.py

app.py

@@ -4,38 +4,6 @@ import requests
 import os
 from gradio_client import Client, handle_file
 
-def extract_audio(input_video_name):
-    # Define the input video file and output audio file
-    mp3_file = "audio.mp3"
-    # Load the video clip
-    video_clip = VideoFileClip(input_video_name)
-
-    # Extract the audio from the video clip
-    audio_clip = video_clip.audio
-    duration = audio_clip.duration
-    print(f"Audio duration: {duration}")
-    # Write the audio to a separate file
-    audio_clip.write_audiofile(mp3_file)
-
-    # Close the video and audio clips
-    audio_clip.close()
-    video_clip.close()
-
-    print("Audio extraction successful!")
-    return mp3_file, duration
-
-
-def download_video(url):
-    response = requests.get(url, stream=True)
-    response.raise_for_status()
-    video_file = "video.mp4"
-    with open(video_file, 'wb') as file:
-        for chunk in response.iter_content(chunk_size=8192):
-            if chunk:
-                file.write(chunk)
-    print("Video downloaded successfully!")
-    return video_file
-
 
 def main(url, clip_type, parameters, progress=gr.Progress()):
     """
@@ -80,20 +48,13 @@ def main(url, clip_type, parameters, progress=gr.Progress()):
         color, font = parameters.split(",")
         progress(0, desc="پردازش شروع شد")
         yield "پردازش شروع شد", None
-
-        progress(0.05, desc="در حال دریافت ویدئو")
-        yield "در حال دریافت ویدئو", None
-        video = download_video(url)
-
-        progress(10, desc="استخراج صوت")
-        yield "استخراج صوت", None
-        mp3_file, duration = extract_audio(video)
-
+        time.sleep(2)
+        
         progress(0.35, desc="تبدیل متن به صوت")
         yield "تبدیل متن به صوت", None
         client = Client("rayesh/transcribe")
         srt_file = client.predict(
-        		mp3_file=handle_file(mp3_file),
+        		url=url,
         		api_name="/transcribe"
         )
         client.close()
@@ -122,6 +83,8 @@ def main(url, clip_type, parameters, progress=gr.Progress()):
         client.close()
         output_video_file = output_video_file['video']
         yield "درحال پردازش ویدئو", output_video_file
+
+        
 with gr.Blocks() as demo:
     gr.Markdown("Start typing below and then click **Run** to see the progress and final output.")
     with gr.Column():