Update app.py

app.py

@@ -479,6 +479,50 @@ class VideoClassifier:
             text-align: center;
         }
         """
+
+        # First interface for model selection
+        interface_1 = gr.Interface(
+            self.save_model_choice,
+            inputs=gr.Dropdown(
+                choices=['gemini', 'mistral'],
+                label="Select Model",
+                info="Default model: Gemini"
+            ),
+            outputs="text",
+            css=css_code,
+            title="Model Selection",
+        )
+
+        # Second interface for video classification
+        video_examples = [
+            [os.path.join(os.path.dirname(__file__), "American_football_heads_to_India_clip.mp4")],
+            [os.path.dirname(__file__) + "/PersonalFinance_clip.mp4"],
+            [os.path.dirname(__file__) + "/Motorcycle_clip.mp4"],
+            [os.path.dirname(__file__) + "/Spirituality_1_clip.mp4"],
+            [os.path.dirname(__file__) + "/Science_clip.mp4"],
+        ]
+
+        checkbox = gr.CheckboxGroup(
+            ["Image Captions and Audio for Classification"],
+            label="Features",
+            info="default : Audio for classification",
+        )
+
+        demo = gr.Interface(
+            fn=self.classify_video,
+            inputs=["playablevideo", checkbox],
+            outputs=["text"],
+            examples=video_examples,
+            css=css_code,
+            allow_flagging='never',
+            title="Interactive Advertising Bureau (IAB) compliant Video-Ad classification",
+        )
+
+        # Tabbed interface with the custom CSS and highlighting
+        gr.TabbedInterface(
+            [interface_1, demo], ["Model Selection", "Video Classification"]
+        ).launch(debug=True, share=True)
+        
         # css_code = """
         # .gradio-container {background-color: #FFFFFF;color:#000000;background-size: 200px; background-image:url(https://gitlab.ignitarium.in/saran/logo/-/raw/aab7c77b4816b8a4bbdc5588eb57ce8b6c15c72d/ign_logo_white.png);background-repeat:no-repeat; position:relative; top:1px; left:5px; padding: 50px;text-align: right;background-position: right top;}
         # """
@@ -493,37 +537,37 @@ class VideoClassifier:
         # }
         # """
 
-        interface_1 = gr.Interface(
-        self.save_model_choice,
-        inputs=gr.Dropdown(choices=['gemini', 'mistral'], label="Select Model", info="Default model: Gemini"),
-        # outputs=interface_1_output,
-        outputs="text"
+    #     interface_1 = gr.Interface(
+    #     self.save_model_choice,
+    #     inputs=gr.Dropdown(choices=['gemini', 'mistral'], label="Select Model", info="Default model: Gemini"),
+    #     # outputs=interface_1_output,
+    #     outputs="text"
         
-    )
-
-        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 and Audio for Classification"],
-            label="Features",
-            info="default : Audio for classification",
-        )
-
-        default_checkbox = []
+    # )
+
+    #     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 and Audio for Classification"],
+    #         label="Features",
+    #         info="default : Audio for classification",
+    #     )
+
+    #     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)
+    #     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)
 
-        gr.TabbedInterface([interface_1, demo], ["Model Selection", "Video Classification"]).launch(debug=True)
+    #     gr.TabbedInterface([interface_1, demo], ["Model Selection", "Video Classification"]).launch(debug=True)
 
     def run_inference(self, video_path,model):
         result = self.classify_video(video_path)