Update app.py

app.py

@@ -11,6 +11,7 @@ import arabic_reshaper # pip install arabic-reshaper
 from bidi.algorithm import get_display # pip install python-bidi
 from moviepy import VideoFileClip, TextClip, CompositeVideoClip, AudioFileClip
 import pysrt
+import instaloader
 api_key = "268976:66f4f58a2a905"
 
 
@@ -95,24 +96,67 @@ def yt_download(url):
     ys.download()
     return video_path, yt.title
 
-def insta_download(shortcode, id_1):
-    url = "https://api.one-api.ir/instagram/v1/post/"
-    request_body = {"shortcode": shortcode, "id": id_1}
-    headers = {"one-api-token": api_key, "Content-Type": "application/json"}
-    response = requests.post(url, headers=headers, json=request_body)
-    if response.status_code == 200:
-        result = response.json()
-        return result['result']
-    else:
-        print(f"Error: {response.status_code}, {response.text}")
-        return None
+def insta_download(permalink):
+    # Create an instance of Instaloader
+    L = instaloader.Instaloader()
+
+    try:
+        # Extract the shortcode from the permalink
+        if "instagram.com/reel/" in permalink:
+            shortcode = permalink.split("instagram.com/reel/")[-1].split("/")[0]
+        elif "instagram.com/p/" in permalink:
+            shortcode = permalink.split("instagram.com/p/")[-1].split("/")[0]
+        else:
+            raise ValueError("Invalid permalink format")
+
+        # Load the post using the shortcode
+        post = instaloader.Post.from_shortcode(L.context, shortcode)
+
+        # Check if the post is a video
+        if not post.is_video:
+            raise ValueError("The provided permalink is not a video.")
+
+        # Get the video URL
+        video_url = post.video_url
+
+        # Extract the filename from the URL
+        filename = video_url.split("/")[-1]
+        # Remove query parameters
+        filename = filename.split("?")[0]
+
+        # Download the video using requests
+        response = requests.get(video_url, stream=True)
+        response.raise_for_status()  # Raise an error for bad responses
+
+        # Save the content to a file
+        with open(filename, 'wb') as file:
+            for chunk in response.iter_content(chunk_size=8192):
+                file.write(chunk)
+
+        print(f"Downloaded video {filename} successfully.")
+        return filename
+    except Exception as e:
+        print(f"Failed to download video from {permalink}: {e}")
 
 def extract_audio(input_video_name):
-    extracted_audio = f"audio-{input_video_name}.wav"
-    stream = ffmpeg.input(input_video)
-    stream = ffmpeg.output(stream, extracted_audio)
-    ffmpeg.run(stream, overwrite_output=True)
-    return extracted_audio
+    # 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
+
+    # 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
 
 def transcribe(audio):
     model = WhisperModel("tiny")
@@ -198,8 +242,13 @@ def create_subtitle_clips(subtitles, videosize, fontsize, font, color, debug):
 
 def process_video(url):
 
-    input_video, input_audio = one_youtube(url)
-    input_video_name = input_video.replace(".mp4", "")
+    if type=="insta":
+        input_video=insta_download(url)
+        input_video_name = input_video.replace(".mp4", "")
+        input_audio = extract_audio(input_video)
+    elif type=="youtube":
+        input_video, input_audio = one_youtube(url)
+        input_video_name = input_video.replace(".mp4", "")
     segments = transcribe(audio=input_audio)
     language = "fa"
     subtitle_file = generate_subtitle_file(language=language, segments=segments, input_video_name=input_video_name)
@@ -223,6 +272,6 @@ def process_video(url):
 #def download_file(file_path):
  #   return gr.File.update(file_path)
 
-iface = gr.Interface(fn=process_video, inputs=["text"], outputs="file")
+iface = gr.Interface(fn=process_video, inputs=["text" ,gr.Dropdown(["insta", "youtube"])], outputs="file")
 
 iface.launch(debug=True)