add first files

app.py

@@ -1,4 +1,5 @@
 import copy
+import subprocess
 
 from pytube import YouTube
 from scipy.signal import resample
@@ -15,6 +16,7 @@ from transformers import pipeline, BertTokenizer, BertForNextSentencePrediction
 import torch 
 import whisper
 
+subprocess.run(['apt-get', '-y', 'install', 'imagemagick'])
 
 transcriber = whisper.load_model("medium")
 sentence_transformer = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
@@ -22,12 +24,14 @@ tokenizer = BertTokenizer.from_pretrained("bert-base-cased")
 next_sentence_predict = BertForNextSentencePrediction.from_pretrained("bert-base-cased").eval()
 summarizer = pipeline("summarization", model="philschmid/bart-large-cnn-samsum")
 
+root_dir = '/home/user/app/video'
+
 
 def get_youtube(video_url):
     # YouTubeの動画をダウンロード
     print("Start download video")
     yt = YouTube(video_url)
-    abs_video_path = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download(filename='download.mp4', output_path='./movies/')
+    abs_video_path = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download(filename='download.mp4', output_path='movies/')
     print("Success download video")
     print(abs_video_path)
     
@@ -147,7 +151,7 @@ def summarize_video(video_path, ratio_sum, playback_speed):
                     end_time = movie_clip.duration
                 clip = movie_clip.subclip(start_time, end_time) 
                 clip = clip.set_pos("center").set_duration(end_time-start_time)
-                txt_clip = TextClip(clip_sentences[i]['sentences'][j], fontsize=int(movie_clip.w/40), color='white', bg_color='black', font='./fonts/Muller-Trial-Medium.ttf')
+                txt_clip = TextClip(clip_sentences[i]['sentences'][j], fontsize=int(movie_clip.w/40), color='white', bg_color='black', font=os.path.join(root_dir, '/fonts/Muller-Trial-Medium.ttf'))
                 txt_clip = txt_clip.set_duration(end_time-start_time).set_position(("center", "bottom"))
                 clip = CompositeVideoClip([clip, txt_clip])
                 tmp_clips.append(clip)