fix error

app.py

@@ -156,7 +156,7 @@ def gen_emotion_feature(frame_dir, emotion_dir):
         for i in range(0, len(emolist) ):
             f.write(str(i) + " "+emolist[i]+"\n")
 
-def gen_scene_feature(video, scene_dir):
+def gen_scene_feature(video, scene_dir, frame_dir):
     video_stream = open_video(str(video))
     
     scene_manager = SceneManager()
@@ -173,9 +173,16 @@ def gen_scene_feature(video, scene_dir):
             sec += 1
     
     fpathname = scene_dir / "scene.lab"
-    with open(fpathname,'w',encoding = 'utf-8') as f:
-        for i in range(0, len(scenedict)):
-            f.write(str(i) + " "+scenedict[i]+"\n")
+
+    if len(scene_list) == 0:
+        fsize = len( os.listdir(frame_dir) )
+        with open(fpathname,'w',encoding = 'utf-8') as f:
+            for i in range(0, fsize):
+                f.write(str(i) + " "+"0"+"\n")
+    else:
+        with open(fpathname,'w',encoding = 'utf-8') as f:
+            for i in range(0, len(scenedict)):
+                f.write(str(i) + " "+scenedict[i]+"\n")
 
 def gen_scene_offset_feature(scene_dir, scene_offset_dir):
     src = scene_dir / "scene.lab"
@@ -421,7 +428,7 @@ class Video2music:
         split_video_into_frames(video, frame_dir)
         gen_semantic_feature(frame_dir, semantic_dir)
         gen_emotion_feature(frame_dir, emotion_dir)
-        gen_scene_feature(video, scene_dir)
+        gen_scene_feature(video, scene_dir, frame_dir)
         gen_scene_offset_feature(scene_dir, scene_offset_dir)
         gen_motion_feature(video, motion_dir)
 
@@ -802,28 +809,27 @@ with gr.Blocks(css=css) as demo:
     with gr.Row():
         with gr.Column():
             radio = gr.Radio(["Upload Video", "YouTube URL"], value="Upload Video", label = "Choose the input method")
-            with gr.Row(visible=True) as mainA:
-                with gr.Column(visible=True) as colA:
-                    with gr.Row(visible=True) as rowA:
-
-                        with gr.Column():
-                            input_video = gr.Video(label="Input Video", height=250)
-                        with gr.Column():
-                            with gr.Row():
-                                input_primer = gr.Textbox(label="Input Primer", value="C Am F G")
-                                input_key = gr.Dropdown(choices=["C major", "A minor"], value="C major", label="Input Key")
-                        with gr.Column():
-                            btn = gr.Button("Generate")
-
-                    with gr.Row(visible=False) as rowB:
-                        with gr.Column():
-                            input_video_yt = gr.Textbox(label="YouTube URL")
-                        with gr.Column():
-                            with gr.Row():
-                                input_primer_yt = gr.Textbox(label="Input Primer", value="C Am F G")
-                                input_key_yt = gr.Dropdown(choices=["C major", "A minor"], value="C major", label="Input Key")
-                        with gr.Column():
-                            btn_yt = gr.Button("Generate")
+            # with gr.Row(visible=True) as mainA:
+            # with gr.Column(visible=True) as colA:
+            with gr.Column(visible=True) as rowA:
+                with gr.Row():
+                    input_video = gr.Video(label="Input Video", height=250)
+                with gr.Row():
+                    with gr.Row():
+                        input_primer = gr.Textbox(label="Input Primer", value="C Am F G")
+                        input_key = gr.Dropdown(choices=["C major", "A minor"], value="C major", label="Input Key")
+                with gr.Row():
+                    btn = gr.Button("Generate")
+
+            with gr.Column(visible=False) as rowB:
+                with gr.Row():
+                    input_video_yt = gr.Textbox(label="YouTube URL")
+                with gr.Row():
+                    with gr.Row():
+                        input_primer_yt = gr.Textbox(label="Input Primer", value="C Am F G")
+                        input_key_yt = gr.Dropdown(choices=["C major", "A minor"], value="C major", label="Input Key")
+                with gr.Row():
+                    btn_yt = gr.Button("Generate")
 
         with gr.Column():
             with gr.Row():