Update app.py

app.py

@@ -33,6 +33,10 @@ class Stack:
     def size(self):
         return len(self.items)
 
+    def clear(self):
+        self.items = []  # 스택을 비우기
+
+
 stack = Stack()
 
 # 발화에 명령어가 포함되어 있는지 확인
@@ -41,6 +45,12 @@ def classifier(utterance):
 
     print(utterance)
 
+    if utterance == "reset":
+        stack.clear()
+        stack.push(node_info["items"][0])
+        stack.push(node_info["items"][1])
+        return str("Reset!")
+
     top = stack.peek()
 
     global_check = re.compile(r'^#global')
@@ -108,7 +118,8 @@ demo = gr.Interface(fn=classifier, inputs="textbox", outputs="textbox",
                         ["missing"],
                         ["select tooth 5 buccal mesial pocket depth"],
                         ["save"],
-                        ["stop listening"]
+                        ["stop listening"],
+                        ["reset"]
                               ])
     
 if __name__ == "__main__":