Update app.py

app.py

@@ -440,42 +440,41 @@ class VideoClassifier:
         
         return "Model selected: " + model_name
 
-    def launch_interface(self):
+    def launch_interface():
         css_code = """
-            .gradio-container {
-                background-color: #d6cbd6;
-            }
-        
-            /* Button styling for all buttons */
-            button {
-                background-color: #d6cbd6; /* Default color for all other buttons */
-                color: black;
-                border: 1px solid black;
-                padding: 10px;
-                margin-right: 10px;
-                font-size: 16px; /* Increase font size */
-                font-weight: bold; /* Make text bold */
-            }
-        
-            /* Style for the second button */
-            button:nth-child(2) {
-                background-color: #927fc7; /* Red color for the second button */
-            }
+        .gradio-container {
+            background-color: #d6cbd6;
+        }
+    
+        /* Button styling for all buttons */
+        button {
+            background-color: #d6cbd6; /* Default color for all other buttons */
+            color: black;
+            border: 1px solid black;
+            padding: 10px;
+            margin-right: 10px;
+            font-size: 16px; /* Increase font size */
+            font-weight: bold; /* Make text bold */
+        }
+    
+        /* Style for the second button */
+        button:nth-child(2) {
+            background-color: #927fc7; /* Custom color for the second button */
+        }
         """
-
+    
         # First interface for model selection
         interface_1 = gr.Interface(
-            self.save_model_choice,
+            fn=save_model_choice,
             inputs=gr.Dropdown(
                 choices=['gemini', 'mistral'],
                 label="Select Model",
-                info="Default model: Gemini"
+                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")],
@@ -484,29 +483,30 @@ class VideoClassifier:
             [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",
+            info="default: Audio for classification",
         )
-
-        demo = gr.Interface(
-            fn=self.classify_video,
-            inputs=["playablevideo", checkbox],
-            outputs=["text"],
+    
+        interface_2 = gr.Interface(
+            fn=classify_video,
+            inputs=[gr.PlayableVideo(), checkbox],
+            outputs="text",
             examples=video_examples,
+            title="Video Classification",
             css=css_code,
-            allow_flagging='never',
-            title="Interactive Advertising Bureau (IAB) compliant Video-Ad classification",
         )
-
-        # Tabbed interface with the custom CSS and highlighting
+    
+        # Create a tabbed interface
         gr.TabbedInterface(
-            [gr.Row([gr.Column([interface_1], size=12), gr.Column([demo], size=12)])],
+            [interface_1, interface_2],
             ["Model Selection", "Video Classification"],
             css=css_code,
         ).launch(debug=True, share=True)
+    
+    launch_interface()
         
         # 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;}