Update app.py

app.py

@@ -20,13 +20,27 @@ def move_uploaded_files(files, temp_dir):
 
 
 def process_images(files):
+    n_view = len(files)
     files_str = ','.join([f.name for f in files])
     hashdir = hashlib.md5(files_str.encode('utf8')).hexdigest()
     hashdir_path = os.path.join(TMP_DIR, hashdir)
     os.makedirs(hashdir_path, exist_ok=True)
     saved_files = move_uploaded_files(files, hashdir_path)
     saved_files_str = ' '.join(saved_files)
-    return 'Before: {}\nAfter: {}'.format(files_str, saved_files_str), saved_files[0]
+    # return 'Before: {}\nAfter: {}'.format(files_str, saved_files_str), saved_files[0]
+
+    command = [
+        "python",
+        "-W", "ignore",
+        "./coarse_init_infer.py",
+        "--img_base_path", hashdir,
+        "--n_views", n_view,
+        "--focal_avg",
+    ]
+    
+    result = subprocess.run(command, capture_output=True, text=True)
+    
+    return result.stdout, result.stderr
 
 def run():
     with gr.Blocks() as demo:
@@ -35,7 +49,8 @@ def run():
         inputs = gr.File(file_count="multiple", file_types=["image"], label="Upload Images")
         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"), gr.Image(type="filepath", label="Processed Images", interactive=False)]    
+        outputs = [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)