Update app.py

app.py

@@ -1,11 +1,8 @@
 import streamlit as st
 import os
-print(os.listdir("/home/user/app/"))
-print(os.path.exists(temp_path))  # Harus return True
 import subprocess
 import tempfile
-import deepmosaic
-deepmosaic.main("--input /tmp/tmp92go9ezb.mp4 --output output_video.mp4")
+import time
 
 def remove_mosaic(video_file):
     output_path = "output_video.mp4"
@@ -15,22 +12,33 @@ def remove_mosaic(video_file):
         temp_file.write(video_file.read())
         temp_path = temp_file.name
     
-    # Jalankan deepmosaic.py untuk memproses video
-    command = "python3 deepmosaic.py --input {} --output {}".format(temp_path, output_path)
-    subprocess.run("python3 deepmosaic.py --help", shell=True)
+    # Jalankan deepmosaic.py untuk menghapus mosaik
+    command = f"python3 deepmosaic.py --mode clean --input {temp_path} --output {output_path}"
+    process = subprocess.run(command, shell=True)
     
-    return output_path
+    # Tunggu sebentar untuk memastikan file selesai dibuat
+    time.sleep(2)
+    
+    if os.path.exists(output_path) and os.path.getsize(output_path) > 0:
+        return output_path
+    else:
+        return None
 
 st.title("🎥 DeepMosaics - Remove Mosaic from Videos")
-uploaded_file = st.file_uploader("Upload Video with Mosaic", type=["mp4", "MKV"])
+uploaded_file = st.file_uploader("Upload Video with Mosaic", type=["mp4", "avi", "mov"])
 
 if uploaded_file is not None:
     st.video(uploaded_file)
     
     if st.button("Remove Mosaic"):
-        result_path = remove_mosaic(uploaded_file)
-        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")
+        with st.spinner("Processing video, please wait..."):
+            result_path = remove_mosaic(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.")