DialogSystem/start.py at master · ShuntaNakajima/DialogSystem · GitHub

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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
# -*- coding: utf-8 -*-
from UtteranceGenerator import DialogSystem
from jwn_corpusreader import JapaneseWordNetCorpusReader
import argparse

dia = DialogSystem()

def start():
    print ('A')
    A = input("")
    print ('B')
    B = input("")
    print ('C')
    C = input("")
    print ('AA')
    AA = input("")
    print ('BB')
    BB = input("")
    print ('CC')
    CC = input("")
    print ('inputType')
    inputtype = input("")
    print ('hasTopic')
    hastopic = input("")
    print ('topic')
    dia.theme = input("")
    dia.theme = 'NEWGAME!'
    dia.TABC = ["NEWGAME!","八神コウ","髪","短い"]
    data = ("八神コウ","髪","黄色")
    #result = dia.generateUtterance(data,int(inputtype),bool(hastopic))
    result = dia.generateUtterance(data,1000,bool(0))
    print(result)

if __name__ == '__main__':

    parser = argparse.ArgumentParser()

    parser.add_argument("url",
                        help="url of the robot")
    parser.add_argument("--debug",
                        help="debug mode. executing CUI only",
                        action="store_true")
    parser.add_argument("--newgame",
                        help="using prepared topic NEWGAME!",
                        action="store_true")
    parser.add_argument("--topic",
                        help="set prepared topic")
    parser.add_argument("--outputtext",
                        help="out put with text only",
                        action="store_true")
    args = parser.parse_args()

    if args.topic:
        dia.preprocessor.GTPP[0] = [args.topic,None]
    elif args.newgame:
        dia.preprocessor.GTPP[0] = ["NEWGAME!",None]


    dia.url = args.url
    dia.start(debug = args.debug,output_type = args.outputtext)

    # 旧テスト群
    # 通信テスト
    dia.output('そろそろいい時間です、ご飯を食べに行きましょう', url='http://192.168.11.14:8080')


    print("======八神コウの髪は短い======")
    print(dia.main('八神コウの髪は短い'))

    print("======八神コウの髪は短い======")
    print(dia.main('八神コウの髪は短い'))

    print("======八神コウの髪がさー======")
    print(dia.main('八神コウの髪がさー'))

    print("======八神コウって短いよね======")
    print(dia.main('八神コウって短いよね'))

    print("======八神コウってさー======")
    print(dia.main('八神コウってさー'))

    print("======髪の毛がさー======")
    print(dia.main('髪の毛がさー'))

    print("======短いよねー======")
    print(dia.main('短いよねー'))