Upload 3 files

app.py

@@ -0,0 +1,15 @@
+import streamlit as st
+from nude_video_app import nude_video_app
+from nude_image_app import nude_image_app
+
+st.set_page_config(page_title="Nudity Detector by Adil Khan")
+def main():
+    app_selection = st.sidebar.radio("Select App", ("Nudity Detection Video App", "Nudity Detection Image App"))
+    if app_selection == "Nudity Detection Video App":
+        nude_video_app()
+    elif app_selection == "Nudity Detection Image App":
+        nude_image_app()
+
+
+if __name__ == "__main__":
+    main()

nude_image_app.py

@@ -0,0 +1,31 @@
+import streamlit as st
+import cv2
+import numpy as np
+from NudeDetector.NudeDetector import NudeDetector
+
+
+def nude_image_app():
+    st.title("Image Nudity Detection App")
+
+    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+
+    if uploaded_file is not None:
+        image = cv2.imdecode(np.fromstring(uploaded_file.read(), np.uint8), cv2.IMREAD_UNCHANGED)
+        detector = NudeDetector(image)
+
+        st.subheader("Nudity Detection Result:")
+        if detector.is_nude():
+            st.error("Image is Nude")
+        else:
+            st.success("Image is Safe")
+        col1, col2 = st.columns([1, 1])
+        with col1:
+            st.subheader("Visualized Image:")
+            st.image(cv2.cvtColor(detector.visualize(), cv2.COLOR_BGR2RGB), caption='Visualized Image', use_column_width=True)
+        with col2:
+            st.subheader("Nudity Blurred Image:")
+            st.image(cv2.cvtColor(detector.blur_nudity(), cv2.COLOR_BGR2RGB), caption='Nudity Blurred Image', use_column_width=True)
+
+
+if __name__ == "__main__":
+    nude_image_app()

nude_video_app.py

@@ -0,0 +1,42 @@
+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()