Update deepmosaic.py

deepmosaic.py

@@ -1,99 +1,53 @@
 import os
 import sys
 import traceback
+import streamlit as st
+import tempfile
+import subprocess
+
 try:
-    from cores import Options,add,clean,style
+    from cores import Options, add, clean, style
     from util import util
     from models import loadmodel
 except Exception as e:
-    print(e)
-    input('Please press any key to exit.\n')
-    sys.exit(0)
+    st.error(f"Error loading dependencies: {e}")
+    sys.exit(1)
 
-opt = Options().getparse(test_flag = True)
+opt = Options().getparse(test_flag=True)
 if not os.path.isdir(opt.temp_dir):
     util.file_init(opt)
 
-def main():
+def process_video(input_video):
+    """ Memproses video untuk menghapus mosaik. """
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_file:
+        temp_file.write(input_video.read())
+        temp_path = temp_file.name
+    
+    output_path = temp_path.replace(".mp4", "_output.mp4")
+    command = ["python3", "deepmosaic.py", "--mode", "clean", "--input", temp_path, "--output", output_path]
     
-    if os.path.isdir(opt.media_path):
-        files = util.Traversal(opt.media_path)
-    else:
-        files = [opt.media_path]        
-    if opt.mode == 'add':
-        netS = loadmodel.bisenet(opt,'roi')
-        for file in files:
-            opt.media_path = file
-            if util.is_img(file):
-                add.addmosaic_img(opt,netS)
-            elif util.is_video(file):
-                add.addmosaic_video(opt,netS)
-                util.clean_tempfiles(opt, tmp_init = False)
-            else:
-                print('This type of file is not supported')
-            util.clean_tempfiles(opt, tmp_init = False)
-
-    elif opt.mode == 'clean':
-        netM = loadmodel.bisenet(opt,'mosaic')
-        if opt.traditional:
-            netG = None
-        elif opt.netG == 'video':
-            netG = loadmodel.video(opt)
-        else:
-            netG = loadmodel.pix2pix(opt)
-        
-        for file in files:
-            opt.media_path = file
-            if util.is_img(file):
-                clean.cleanmosaic_img(opt,netG,netM)
-            elif util.is_video(file):
-                if opt.netG == 'video' and not opt.traditional:            
-                    clean.cleanmosaic_video_fusion(opt,netG,netM)
-                else:
-                    clean.cleanmosaic_video_byframe(opt,netG,netM)
-                util.clean_tempfiles(opt, tmp_init = False)
-            else:
-                print('This type of file is not supported')
-
-    elif opt.mode == 'style':
-        netG = loadmodel.style(opt)
-        for file in files:
-            opt.media_path = file
-            if util.is_img(file):
-                style.styletransfer_img(opt,netG)
-            elif util.is_video(file):
-                style.styletransfer_video(opt,netG)
-                util.clean_tempfiles(opt, tmp_init = False)
-            else:
-                print('This type of file is not supported')
-
-    util.clean_tempfiles(opt, tmp_init = False)
-
-if __name__ == '__main__':
-    if opt.debug:
-        main()
-        sys.exit(0)
     try:
-        main()
-        print('Finished!')
-    except Exception as ex:
-        print('--------------------ERROR--------------------')
-        print('--------------Environment--------------')
-        print('DeepMosaics: 0.5.1')
-        print('Python:',sys.version)
-        import torch
-        print('Pytorch:',torch.__version__)
-        import cv2
-        print('OpenCV:',cv2.__version__)
-        import platform
-        print('Platform:',platform.platform())
+        subprocess.run(command, check=True)
+        return output_path
+    except subprocess.CalledProcessError as e:
+        st.error(f"Error processing video: {e}")
+        return None
+
+st.title("🎥 DeepMosaics - Remove Video Mosaic")
+uploaded_file = st.file_uploader("Upload a video file", type=["mp4", "avi", "mov"])
 
-        print('--------------BUG--------------')
-        ex_type, ex_val, ex_stack = sys.exc_info()
-        print('Error Type:',ex_type)
-        print(ex_val)
-        for stack in traceback.extract_tb(ex_stack):
-            print(stack)
-        input('Please press any key to exit.\n')
-        #util.clean_tempfiles(tmp_init = False)
-        sys.exit(0)
+if uploaded_file is not None:
+    st.video(uploaded_file)
+    
+    if st.button("Remove Mosaic"):
+        with st.spinner("Processing video, please wait..."):
+            result_path = process_video(uploaded_file)
+            
+        if result_path:
+            st.success("Processing complete!")
+            st.video(result_path)
+            
+            with open(result_path, "rb") as file:
+                st.download_button("Download Processed Video", file, "output_video.mp4")
+        else:
+            st.error("Failed to process video. Please try again.")