Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
-
+import re
 import json
 
 # JSON 파일 읽기
-with open('node.json', 'r') as file:
+with open('perio_demo.json', 'r') as file:
     node_info = json.load(file)
 
 ##############################################################################
@@ -43,21 +43,27 @@ def classifier(utterance):
 
     top = stack.peek()
 
-    if utterance == "save":
-        while(True):
-            if stack.size() == 0:
-                return
-            peek = stack.peek()
-            if peek['start_node'] == 3:
-                return
-            else:
-                stack.pop()
+    global_check = re.compile(r'^#global')
 
     # top node 의 dest node 를 모두 가져온다.
     for dest_node in top['dest_node']:
 
         # node_info 에서 start_node == dest_node 인 노드를 찾는다.
         for node in node_info["items"]:
+
+            if node["utterance"] == utterance:
+                if node["fallback"] is not None:
+                    if global_check.match(node["fallback"]):
+                        node["fallback"] = int(re.sub(r'\D', '', node["fallback"]))
+                        while(True):
+                            if stack.size() == 0:
+                                return
+                            peek = stack.peek()
+                            if peek['start_node'] == node["fallback"]:
+                                return str(str(node['depth']) + ':' + node['intent'])
+                            else:
+                                stack.pop()
+
             if node['start_node'] == dest_node:
                 # candidate_node.append(node)
                 # 최신 노드보다 발화 노드가 다음 state 인지 확인
@@ -92,20 +98,18 @@ stack.push(node_info["items"][0])
 ## 페리오 차트로 분기
 stack.push(node_info["items"][1])
 
-
-
 demo = gr.Interface(fn=classifier, inputs="textbox", outputs="textbox",
                     examples=[
                         ["start listening"],
                         ["select tooth 1"],
                         ["pocket depth 1 2 3"],
                         ["select tooth 2"],
-                        ["treatment plan"],
+                        ["treatment note"],
                         ["missing"],
                         ["select tooth 5 buccal mesial pocket depth"],
-                        ["save"]
-                        ]
-                    )
+                        ["save"],
+                        ["stop listening"]
+                              ])
     
 if __name__ == "__main__":
-    demo.launch(share=True)   
+    demo.launch()