refine segment.py

1 files changed, 7 insertions, 2 deletions

diff --git a/segment.py b/segment.py
index fbd7f65..b65bfd7 100755
--- a/segment.py
+++ b/segment.py
@@ -26,10 +26,15 @@ def segmentOneText(infile, outfile, reportfile):
         return
 
     #begin processing
-    cmdline = './ngseg <"' + infile + '" >"' + outfile + '"'
-    subprocess = Popen(cmdline, shell=True, stderr=PIPE, \
+    cmdline = './ngseg >"' + outfile + '"'
+    subprocess = Popen(cmdline, shell=True, stdin=PIPE, stderr=PIPE, \
                            close_fds=True)
 
+    with open(infile, 'rb') as f:
+        subprocess.stdin.writelines(f.readlines())
+    subprocess.stdin.close()
+    f.close()
+
     lines = subprocess.stderr.readlines()
     if lines:
         print('found error report')