Update app.py

app.py

@@ -9,24 +9,15 @@ from scenedetect.detectors import ContentDetector
 
 from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
 
-# —————————————————————————————————————————————————
-
-title = "Scene Edit Detection"
-description = "<p style='text-align: center'>Copy of @fffiloni's gradio demo of PySceneDetect. <br />Automatically find every shots in a video sequence</p><p style='text-align: center'> 1. gives you timecode in/out for each shot. 2. saves each shot as a splitted mp4 video chunk for you to download. 3. diplays a thumbnail for each shot as a gallery output.<br /> <img id='visitor-badge' alt='visitor badge' src='https://visitor-badge.glitch.me/badge?page_id=gradio-blocks.scene-edit-detection' style='display: inline-block'/></b></p>"
-article = "<p style='text-align: center'><a href='http://scenedetect.com/en/latest/' target='_blank'>PySceneDetect website</a> | <a href='https://github.com/Breakthrough/PySceneDetect' target='_blank'>Github Repo</a></p>"
-
-# —————————————————————————————————————————————————
-
-# SET INPUTS
-video_input = gr.Video(source="upload", format="mp4", label="Video Sequence", mirror_webcam=False)
-threshold = gr.Slider(label="Threshold pixel comparison: if exceeded, triggers a scene cut. Default: 27.0", minimum=15.0, maximum=40.0, value=27.0)
-
-# —————————————————————————————————————————————————
 
 def convert_to_tuple(list):
     return tuple(list);
 
 
+def clear_app():
+    return None, 27, None, None, None
+
+
 def find_scenes(video_path, threshold):
     # file name without extension
     filename = os.path.splitext(os.path.basename(video_path))[0]
@@ -46,6 +37,8 @@ def find_scenes(video_path, threshold):
     #print(scene_list)
     
     timecodes = []
+    if not scene_list:
+        raise ValueError("There are no scenes detected in this video.")
     timecodes.append({"title": filename + ".mp4", "fps": scene_list[0][0].get_framerate()})
     
     shots = []
@@ -125,6 +118,7 @@ def find_scenes(video_path, threshold):
     # return results 
     return timecodes, shots, stills
 
+
 # —————————————————————————————————————————————————
 
 # SET DATA AND COMPONENTS OUTPUTS
@@ -147,8 +141,30 @@ gradio_components_outputs = []
 
 # ANOTHER SOLUTION WOULD BE USING A (FUTURE ?) "VIDEO GALLERY" GRADIO COMPONENT FROM LIST :)
 
-outputs = [gr.JSON(label="Shots detected"), gr.File(label="Downloadable Shots"), gr.Gallery(label="Still Images from each shot").style(grid=3)]
 
-# —————————————————————————————————————————————————
-print('Hello Sylvain')
-gr.Interface(fn=find_scenes, inputs=[video_input, threshold], outputs=outputs, title=title, description=description, article=article).launch()
+
+with gr.Blocks(gr.themes.Default()) as demo:
+    with gr.Column():
+        gr.Markdown("""
+            # Scene Edit Detection
+            Copy of @fffiloni's gradio demo of PySceneDetect.
+            Automatically find all the shots in a video sequence.
+            """)
+        with gr.Row():
+            with gr.Column():
+                video_input = gr.Video(sources="upload", format="mp4", label="Video Sequence", mirror_webcam = False)
+                threshold = gr.Slider(label="Threshold pixel comparison: if exceeded, triggers a scene cut. Default: 27.0", minimum=15.0, maximum=40.0, value=27.0)
+                with gr.Row():
+                    clear_button = gr.Button(value=("Clear"))
+                    run_button = gr.Button(value = "Submit", variant = "primary")
+                    
+
+            with gr.Column():
+                json_output = gr.JSON(label="Shots detected")
+                file_output = gr.File(label="Downloadable Shots")
+                gallery_output = gr.Gallery(label="Still Images from each shot")
+
+        run_button.click(fn=find_scenes, inputs=[video_input, threshold], outputs=[json_output, file_output, gallery_output])
+        clear_button.click(fn=clear_app, outputs=[video_input, threshold, json_output, file_output, gallery_output])
+if __name__ == "__main__":
+    demo.launch(debug=True)