Update app.py

app.py

@@ -6,6 +6,8 @@ import cv2
 import os
 import urllib.request
 import moviepy
+import threading
+
 
 hide_streamlit_style = """
 <style>
@@ -16,6 +18,15 @@ hide_streamlit_style = """
 
 """
 
+
+def sound_extract(file):
+    video = moviepy.editor.VideoFileClip(file)
+    video.audio.write_audiofile(file.split('.')[0] + '.mp3')
+
+def remove_file(file):
+    os.remove('deep_'+file)
+
+
 st.set_page_config(layout="wide")
 
 
@@ -44,12 +55,17 @@ def upscale(file,task,_progressBar = None):
                 st.success('Done!', icon="✅")
             return True
         elif file.type.split('/')[0] == 'video':
+
+            t1 = threading.Thread(target=sound_extract, args=(file.name,))
+
+            t1.start()
+            
             cap = cv2.VideoCapture(file.name)
             fourcc = cv2.VideoWriter_fourcc(*'mp4v')
             fps = int(cap.get(cv2.CAP_PROP_FPS))
             width = int(cap.get(3))
             height = int(cap.get(4))
-            writer = cv2.VideoWriter("processed_"+file.name,fourcc,fps,(width*int(task),height*int(task)))
+            writer = cv2.VideoWriter("deep_"+file.name,fourcc,fps,(width*int(task),height*int(task)))
             length = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
             step_size = 1.0/length
             progress = 0
@@ -65,8 +81,7 @@ def upscale(file,task,_progressBar = None):
                 print(progress)
                 progress += step_size
                 my_bar.progress(progress-0.000000001)
-            with st.sidebar:
-                st.success('Done! Thankyou for your patience', icon="✅")
+            
             return True
         return True
 
@@ -109,6 +124,11 @@ def upscale(file,task,_progressBar = None):
 
         # If file is video
         elif file.type.split('/')[0] == 'video':
+
+            t1 = threading.Thread(target=sound_extract, args=(file.name,))
+
+            t1.start()
+            
             cap = cv2.VideoCapture(file.name)
             fourcc = cv2.VideoWriter_fourcc(*'mp4v')
             fps = int(cap.get(cv2.CAP_PROP_FPS))
@@ -117,7 +137,7 @@ def upscale(file,task,_progressBar = None):
             if height > 2160 or width > 3840:
                 with st.sidebar:
                     st.success("Sorry, I can't processed Video with resolution above 4k. Please select custom size option.!", icon="ℹ️")
-            writer = cv2.VideoWriter("processed_" + file.name, fourcc, fps, (int(task[0]),int(task[1])))
+            writer = cv2.VideoWriter("deep_" + file.name, fourcc, fps, (int(task[0]),int(task[1])))
             length = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
             step_size = 1.0 / length
             progress = 0
@@ -131,8 +151,7 @@ def upscale(file,task,_progressBar = None):
                 writer.write(frame)
                 progress += step_size
                 my_bar.progress(progress-0.000000001)
-            with st.sidebar:
-                st.success('Done! Thankyou for your patience', icon="✅")
+            
             return True
         return "It's second"
 
@@ -241,6 +260,17 @@ with col2:
         print(task)
         st.session_state.disable_download = not upscale(file,task,progressBar)
 
+        if file.type.split('/')[0] == 'video':
+            with st.sidebar:
+                st.info("Preparing for Download. Please wait.", icon="ℹ️")
+                audio = moviepy.editor.AudioFileClip(file.name.split('.')[0] + '.mp3')
+                video = moviepy.editor.VideoFileClip("deep_"+file.name)
+                videoClip = video.set_audio(audio)
+                videoClip.write_videofile('processed_'+file.name)
+                t2 = threading.Thread(target=remove_file(file.name))
+                t2.start()
+                st.success('Done! Thankyou for your patience', icon="✅")        
+
         #print(resulted_file.shape)
 
     st.markdown("\n")