| import streamlit as st | |
| import cv2 | |
| import tempfile | |
| import numpy as np | |
| from NudeDetector.NudeDetector import NudeDetector | |
| def nude_video_app(): | |
| st.title("Video Nudity Detection App") | |
| uploaded_file = st.file_uploader("Upload a video", type=["mp4", "avi"]) | |
| if uploaded_file is not None: | |
| with tempfile.NamedTemporaryFile(delete=False) as temp_file: | |
| temp_file.write(uploaded_file.read()) | |
| uploaded_video_path = temp_file.name | |
| cap = cv2.VideoCapture(uploaded_video_path) | |
| skip_interval = 20 | |
| frame_index = 0 | |
| placeholder1 = st.empty() | |
| col1, col2 = st.columns([1, 1]) | |
| with col1: | |
| placeholder2 = st.empty() | |
| with col2: | |
| placeholder3 = st.empty() | |
| while True: | |
| success, img = cap.read() | |
| if not success: | |
| break | |
| if frame_index % skip_interval == 0: | |
| detector = NudeDetector(img) | |
| if detector.is_nude(): | |
| placeholder1.error("Image is Nude") | |
| else: | |
| placeholder1.success("Image is Safe") | |
| placeholder2.image(cv2.cvtColor(detector.visualize(), cv2.COLOR_BGR2RGB), caption='Visualized Image', use_column_width=True) | |
| placeholder3.image(cv2.cvtColor(detector.blur_nudity(), cv2.COLOR_BGR2RGB), caption='Nudity Blurred Image', use_column_width=True) | |
| frame_index += 1 | |
| if __name__ == "__main__": | |
| nude_video_app() | |