Update app.py

app.py

@@ -17,7 +17,8 @@ def move_uploaded_files(files, temp_dir):
     for file in files:
         target_file_path = os.path.join(temp_dir, os.path.basename(file.name))
         saved_files.append(target_file_path)
-        shutil.move(file.name, target_file_path)
+        if os.path.exists(file.name):
+            shutil.move(file.name, target_file_path)
     return saved_files
 
 
@@ -30,12 +31,9 @@ def process_images(files):
     os.makedirs(hashdir_path_img, exist_ok=True)
     saved_files = move_uploaded_files(files, hashdir_path_img)
     saved_files_str = ' '.join(saved_files)
-    # return 'Before: {}\nAfter: {}'.format(files_str, saved_files_str), saved_files[0]
 
     command = [
-        # "CUDA_VISIBLE_DEVICES=0",
-        "python",
-        "-W", "ignore",
+        "python", "-W", "ignore",
         "./coarse_init_infer.py",
         "--img_base_path", hashdir_path,
         "--n_views", str(n_view),
@@ -43,8 +41,9 @@ def process_images(files):
     ]
     
     result = subprocess.run(command, capture_output=True, text=True)
+    result_tree = subprocess.run(["tree", hashdir_path], capture_output=True, text=True)
     
-    return result.stdout, result.stderr
+    return result_tree.stdout, result.stdout, result.stderr
 
 def run():
     with gr.Blocks() as demo:
@@ -54,7 +53,10 @@ def run():
         btn = gr.Button(value="Run")
         # outputs = [gr.Image(type="numpy", label="Processed Images", interactive=False), gr.Video(label="Output Video")]    
         # outputs = [gr.Textbox(label="Log"), gr.Image(type="filepath", label="Processed Images", interactive=False)]    
-        outputs = [gr.Textbox(label="Log (stdout)"), gr.Textbox(label="Log (stderr)")]    
+        outputs = [
+            gr.Textbox(label="Output dirs"), 
+            gr.Textbox(label="Log (stdout)"), 
+            gr.Textbox(label="Log (stderr)")]    
         btn.click(fn=process_images, inputs=inputs, outputs=outputs)
 
     demo.launch(server_name="0.0.0.0", server_port=7860)