Update app.py

app.py

@@ -104,7 +104,7 @@ def get_music(prompt):
   return spec, 'output.wav'
     
   
-def merge_video(music, img_list):
+def merge_video(music, image):
     #Convert to mp3   
     mp3file_name = "audio.mp3"
     wavfile = AudioSegment.from_wav(music)
@@ -121,12 +121,17 @@ def merge_video(music, img_list):
     fourcc = cv2.VideoWriter.fourcc(*'MJPG')
     out = cv2.VideoWriter(file_name, fourcc, fps, (1024, 1024))
     
-    for image in img_list:
-        cv_img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
-        for _ in range(slide_time * fps):
-            #cv_img = cv2.resize(np.array(cv_img), (1024, 1024))
-            out.write(cv_img)
+    # for image in img_list:
+    #     cv_img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
+    #     for _ in range(slide_time * fps):
+    #         #cv_img = cv2.resize(np.array(cv_img), (1024, 1024))
+    #         out.write(cv_img)
 
+    cv_img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
+    for _ in range(slide_time * fps):
+        #cv_img = cv2.resize(np.array(cv_img),
+        out.write(cv_img)
+        
     out.release()