init

1 files changed, 54 insertions, 0 deletions

diff --git a/demo.py b/demo.py
new file mode 100755
index 0000000..fe29b99
--- /dev/null
+++ b/demo.py
@@ -0,0 +1,54 @@
+#!/usr/bin/env python2
+
+import sys
+import subprocess
+from nltk.stem.porter import PorterStemmer
+from src.functionalizer import Functionalizer
+import re
+import os
+
+def pretty_print_prolog(ans):
+  ans = re.sub(r'\w+\(([^)]+)\)', r'\1', ans)
+  parts = ans[1:-2].split(',')
+  print '\n'.join(parts)
+  #for part in parts:
+  #  if 'stateid' in part:
+  #    part = part[8:-1]
+  #  print part
+
+MOSES='/home/jacob/src/3p/mosesdecoder/dist/bin/moses_chart'
+#WORK_DIR='/home/jacob/src/smt-semparse/work/DEMO'
+WORK_DIR=os.path.realpath('latest')
+
+fcr = Functionalizer(None)
+
+moses_args = [MOSES, 
+              '-drop-unknown',
+              '-f', '%s/mert-work/moses.ini' % WORK_DIR]
+
+moses = subprocess.Popen(moses_args, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
+                         stderr=subprocess.PIPE)
+
+prolog_args = ['/usr/bin/swipl',
+               '-l', '/home/jacob/src/3p/wasp-1.0/data/geo-funql/eval/eval.pl']
+
+prolog = subprocess.Popen(prolog_args, stdin=subprocess.PIPE,
+    stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+while True:
+
+  print '\n\n? ',
+  line = sys.stdin.readline()
+  print
+  print >>moses.stdin, line
+  mrl = moses.stdout.readline().strip()
+  moses.stdout.readline()
+
+  fun = fcr.functionalize(mrl)
+  print '!', fun
+
+  plg = 'execute_funql_query(%s, A), print(A), nl.\n' % fun
+  print >>prolog.stdin, plg
+  answer = prolog.stdout.readline()
+  print
+  pretty_print_prolog(answer)