Update app.py

app.py

@@ -94,25 +94,24 @@ def infer(ref_video_in, ref_image_in):
 
         # Execute the inference command
         command = ['python', 'inference.py', '--inference_config', file_path]
-        process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
-        # Print logs in real-time
-        while True:
-            output = process.stdout.readline()
-            if output == '' and process.poll() is not None:
-                break
-            if output:
-                print(output.strip())
-        
-        # Print any remaining output
-        for output in process.stdout:
-            print(output.strip())
-        for error in process.stderr:
-            print(error.strip())
-        
-        # Wait for the process to complete and get the return code
-        return_code = process.wait()
-        print("Inference script finished with return code:", return_code)
+        process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, bufsize=1)
     
+        # Create threads to handle stdout and stderr
+        stdout_thread = threading.Thread(target=stream_output, args=(process.stdout,))
+        stderr_thread = threading.Thread(target=stream_output, args=(process.stderr,))
+    
+        # Start the threads
+        stdout_thread.start()
+        stderr_thread.start()
+    
+        # Wait for the process to complete and the threads to finish
+        process.wait()
+        stdout_thread.join()
+        stderr_thread.join()
+    
+        print("Inference script finished with return code:", process.returncode)
+        # Print the directory contents
+        print_directory_contents('./outputs')
     
     return "done"