Update app.py

app.py

@@ -216,8 +216,9 @@ class VideoClassifier:
             
         return self.client(inputs, parameters)
     
-    def classify_video(self,video_input,checkbox):
+    def classify_video(self,video_input,checkbox=default_checkbox):
         global classification_time , caption_time
+        print("checkbox",checkbox)
         # transcript=self.audio_extraction_space(video_input)
         try:
             transcript=self.audio_extraction(video_input)
@@ -232,6 +233,8 @@ class VideoClassifier:
         #     except:
         #         transcript=self.audio_extraction_space(video_input)
         start_time_caption = time.time()
+        captions = ""
+        
         if checkbox=="Image Captions":
             video = cv2.VideoCapture(video_input)
             length = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
@@ -259,8 +262,7 @@ class VideoClassifier:
             cv2.destroyAllWindows()
             
             # print("TIME TAKEN FOR IMAGE CAPTIONING", end_time_caption-start_time_caption)
-        else:
-            captions=""
+            
         end_time_caption = time.time()
         caption_time=end_time_caption-start_time_caption
         start_time_generation = time.time()
@@ -461,17 +463,24 @@ class VideoClassifier:
         
     )
 
+        video_examples = [
+            [os.path.join(os.path.dirname(__file__), "American_football_heads_to_India_clip.mp4")],
+            [os.path.join(os.path.dirname(__file__), "PersonalFinance_clip.mp4")],
+            [os.path.join(os.path.dirname(__file__), "Motorcycle_clip.mp4")],
+            [os.path.join(os.path.dirname(__file__), "Spirituality_1_clip.mp4")],
+            [os.path.join(os.path.dirname(__file__), "Science_clip.mp4")]
+        ]
+
+        # Define the checkbox for additional feature control
         checkbox = gr.CheckboxGroup(
-            ["Image Captions"], 
+            ["Image Captions"],
+            label="Features",
+            info="Select features for video classification",
         )
-        demo = gr.Interface(fn=self.classify_video, inputs=["playablevideo",checkbox],allow_flagging='never', examples=[
-        # Each example should be a sublist with corresponding values for each input component
-        [os.path.join(os.path.dirname(__file__), "American_football_heads_to_India_clip.mp4"), ["Image Captions"]],
-        [os.path.join(os.path.dirname(__file__), "PersonalFinance_clip.mp4"), ["Image Captions"]],
-        [os.path.join(os.path.dirname(__file__), "Motorcycle_clip.mp4"), ["Image Captions"]],
-        [os.path.join(os.path.dirname(__file__), "Spirituality_1_clip.mp4"), ["Image Captions"]],
-        [os.path.join(os.path.dirname(__file__), "Science_clip.mp4"), ["Image Captions"]],
-    ],
+
+        default_checkbox = []
+        
+        demo = gr.Interface(fn=self.classify_video, inputs=["playablevideo",checkbox],allow_flagging='never', examples=video_examples,
                         cache_examples=False, outputs=["text"],
                         css=css_code, title="Interactive Advertising Bureau (IAB) compliant Video-Ad classification")
         # demo.launch(debug=True)