Upload WarBot.py

Updated the WarBot

WarBot.py

@@ -113,13 +113,18 @@ def get_response(quote:str,model,tokenizer,model_punct):
     response = remove_duplicates(re.sub(r"\d{4,}", "", response))  # Remove the consequent numbers with 4 or more digits
     response = re.sub(r'\.\.+', '', response) # Remove the "....." thing
 
-    if len(response)>200:
-        resps = wrap(response,200)
-        for i in range(len(resps)):
-            resps[i] = model_punct.enhance_text(resps[i], lan='ru')
-            response = ''.join(resps)
-    else:
-        response = model_punct.enhance_text(response, lan='ru')
+    maxLen = 170
+
+    try:
+        if len(response)>maxLen: # We shall play with it
+            resps = wrap(response,maxLen)
+            for i in range(len(resps)):
+                resps[i] = model_punct.enhance_text(resps[i], lan='ru')
+                response = ''.join(resps)
+        else:
+            response = model_punct.enhance_text(response, lan='ru')
+    except:
+	    pass # sometimes the string is getting too long
 
     response = re.sub(r'[UNK]', '', response)  # Remove the [UNK] thing
     return response
@@ -130,4 +135,4 @@ def get_response(quote:str,model,tokenizer,model_punct):
     #print('please wait...')
     #response = wrap(get_response(quote,model,tokenizer,model_punct),60)
     #for phrase in response:
-    #    print(phrase)
+    #    print(phrase)