From 53785c27803d39b783f33f6ce097d03b48f7240f Mon Sep 17 00:00:00 2001
From: VISHAL SINGH <56965195+vishalsingha@users.noreply.github.com>
Date: Tue, 4 Aug 2020 17:04:07 +0530
Subject: [PATCH] Added few lines to demo_manager.py

Added code to remove some printing bug like ( ["do n't", "ca n't" etc.]  will now be printed as  ["don't", "can't" etc.] ).
---
 viscap/visdialch/data/demo_manager.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/viscap/visdialch/data/demo_manager.py b/viscap/visdialch/data/demo_manager.py
index 81287ca..3809349 100644
--- a/viscap/visdialch/data/demo_manager.py
+++ b/viscap/visdialch/data/demo_manager.py
@@ -249,6 +249,10 @@ def respond(self, user_question):
 
         answer = TreebankWordDetokenizer().detokenize(answer)
         
+        if "n't" in answer:
+            idx_ = answer.find("n't")
+            answer = ''.join([answer[i] for i in range(len(answer)) if i != idx_-1]) 
+        
         # Update the dialog history and return answer
         self._update(user_question, answer)
         return answer