Handling the edge cases

app.py

@@ -192,6 +192,18 @@ def predict_video(model, video_path):
     return original_frames, predicted_label, probability
 
 def display_frames_and_prediction(video_file):
+    # Ensure file is completely uploaded and of correct type
+    if video_file is None:
+        return [], "<div style='color: red;'>No video uploaded!</div>", ""
+    
+    # Check file size (10 MB limit)
+    if os.path.getsize(video_file) > 10 * 1024 * 1024:  # 10 MB
+        return [], "<div style='color: red;'>File size exceeds 10 MB limit!</div>", ""
+    
+    # Check file extension
+    if not video_file.endswith('.mp4'):
+        return [], "<div style='color: red;'>Only .mp4 files are allowed!</div>", ""
+    
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_file:
         temp_file_path = temp_file.name
 
@@ -216,7 +228,7 @@ def display_frames_and_prediction(video_file):
 
 iface = gr.Interface(
     fn=display_frames_and_prediction,
-    inputs=gr.File(label="Upload Video"),
+    inputs=gr.File(label="Upload Video", interactive=True),
     outputs=[
         gr.Gallery(label="Extracted Frames"), 
         gr.HTML(label="Prediction"),