Update app.py

app.py

@@ -6,9 +6,9 @@ import cv2
 import random
    
 
-os.system("mkdir _input")
-os.system("mkdir _output")
-os.system("mkdir _outputf")
+os.system("mkdir /tmp/_input")
+os.system("mkdir /tmp/_output")
+os.system("mkdir /tmp/_outputf")
 os.system("ls")
 col1, col2 = st.columns(2)
 
@@ -25,9 +25,9 @@ if uploaded_file is not None:
     # To read file as bytes:
     bytes_data = uploaded_file.getvalue()
     st.video(bytes_data)
-    with open("inputvideo", "wb") as binary_file:
+    with open("/tmp/inputvideo", "wb") as binary_file:
        binary_file.write(bytes_data)
-    vcap = cv2.VideoCapture('inputvideo') # 0=camera
+    vcap = cv2.VideoCapture('/tmp/inputvideo') # 0=camera
     width=0
     height=0
     if vcap.isOpened(): 
@@ -48,7 +48,7 @@ def chamada():
        #exec=True 
 #           st.write("ffmpeg separando imagens")
        #if not os.path.isfile("./_input/imagem-0001.png"):
-          vcap = cv2.VideoCapture('inputvideo') # 0=camera
+          vcap = cv2.VideoCapture('/tmp/inputvideo') # 0=camera
           width=0
           height=0
           if vcap.isOpened(): 
@@ -57,49 +57,49 @@ def chamada():
               height = vcap.get(4)  # float `height`
               fps = vcap.get(5)
               -vn -acodec copy output-audio.aac
-          os.system("ffmpeg -i inputvideo -vn -acodec copy output-audio.aac ./inputvideo.acc")
-          os.system("ffmpeg -i inputvideo -compression_level 10 -pred mixed -pix_fmt rgb24 -sws_flags +accurate_rnd+full_chroma_int -s 720x1280 -r   30 ./_input/imagem-%4d.png")
+          os.system("ffmpeg -i /tmp/inputvideo -vn -acodec copy output-audio.aac /tmp/inputvideo.acc")
+          os.system("ffmpeg -i /tmp/inputvideo -compression_level 10 -pred mixed -pix_fmt rgb24 -sws_flags +accurate_rnd+full_chroma_int -s 720x1280 -r   30 /tmp/_input/imagem-%4d.png")
           percent_complete= percent_complete+ 30
           with col1:
            my_bar.progress(percent_complete )   
  #          st.write("testando imagem")
-          input_img = cv2.imread("./_input/imagem-0002.png" , cv2.IMREAD_COLOR)
+          input_img = cv2.imread("./tmp/_input/imagem-0002.png" , cv2.IMREAD_COLOR)
           input_img= cv2.cvtColor(input_img,cv2.COLOR_BGR2RGB) 
           with col1:
            st.image(input_img)
           
-          os.system("ls ./_input") 
+          os.system("ls /tmp/_input") 
           if 'myVar' not in globals():
               myVar=""
   #        st.write("melhorando faces")
           with col2:
            with st.spinner('Wait for it...'):
              # os.system("pip install git+https://github.com/TencentARC/GFPGAN.git")
-                    os.system("python3 inference_gfpgan.py -i _input -o _output -v 1.3 -s 2")
+                    os.system("python3 inference_gfpgan.py -i /tmp/_input -o /tmp/_output -v 1.3 -s 2")
           with col1:
            percent_complete= percent_complete+ 30
            my_bar.progress(percent_complete )   
          
-          os.system("ls ./_output")
+          os.system("ls /tmp/_output")
           os.system("echo ----")
-          os.system("ls ./_output/cmp")
+          os.system("ls /tmp/_output/cmp")
           os.system("echo ----")
-          os.system("ls ./_output/restored_imgs")
+          os.system("ls /tmp/./_output/restored_imgs")
           os.system("echo ----")
 #          s 1480x2560
        #   with col1:
    #        st.write("recompilando video")
           #ffmpeg -r 60 -f image2 -s 1920x1080 -i _output/restored_imgs/imagem-%4d.png   -pix_fmt yuv420p ./videoSaida/output.mp4
-          os.system("ffmpeg -y -r 30 -f image2  -i _output/restored_imgs/imagem-%4d.png   -pix_fmt yuv420p ./videoSaida/outputp1.mp4")
-          os.system("ffmpeg -i ./videoSaida/outputp1.mp4 -i inputvideo.acc -c:v copy -c:a aac output.mp4")
-          os.system("ls ./videoSaida")
+          os.system("ffmpeg -y -r 30 -f image2  -i /tmp/_output/restored_imgs/imagem-%4d.png   -pix_fmt yuv420p /tmp/videoSaida/outputp1.mp4")
+          os.system("ffmpeg -i /tmp/videoSaida/outputp1.mp4 -i /tmp/inputvideo.acc -c:v copy -c:a aac /tmp/output.mp4")
+          os.system("ls /tmp/videoSaida")
           #st.video("./videoSaida/output.mp4" )
           with col1:
         #   st.write("preparando para download do video")
            percent_complete= percent_complete+ 30
            my_bar.progress(percent_complete ) 
           with col2:
-           with open("./videoSaida/output.mp4", "rb") as file:
+           with open("/tmp/videoSaida/output.mp4", "rb") as file:
               st.video(file)
 
               btn = st.download_button(
@@ -119,7 +119,7 @@ def chamada():
           #input_img = cv2.imread("./_output/cmp/imagem-0001_0000.png" , cv2.IMREAD_COLOR)
           with col2:
          #  st.write("demonstrando imagem restaurada")
-           input_img = cv2.imread("./_output/restored_imgs/imagem-0002.png" , cv2.IMREAD_COLOR)
+           input_img = cv2.imread("/tmp/_output/restored_imgs/imagem-0002.png" , cv2.IMREAD_COLOR)
            input_img= cv2.cvtColor(input_img,cv2.COLOR_BGR2RGB) 
           
            st.image(input_img)