1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
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)
|
