kertser commited on
Commit
9e46021
1 Parent(s): cad68e8

Upload 2 files

Browse files
Files changed (2) hide show
  1. WarClient.py +3 -1
  2. WarOnline_Chat.py +13 -4
WarClient.py CHANGED
@@ -6,9 +6,11 @@ PORT = 5000
6
  def getReply(message):
7
  # Returns a reply from the server
8
 
9
- # Remove the finel "." - somehow it stucks the model
10
  if message.endswith("."):
11
  message = message.rstrip(".")
 
 
12
 
13
  with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
14
  try:
 
6
  def getReply(message):
7
  # Returns a reply from the server
8
 
9
+ # Remove the endings that stuck the model
10
  if message.endswith("."):
11
  message = message.rstrip(".")
12
+ if message.endswith(" ?"):
13
+ message = message.replace(" ?","?")
14
 
15
  with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
16
  try:
WarOnline_Chat.py CHANGED
@@ -223,16 +223,25 @@ def WarOnlineBot():
223
  if msg['messageID'].split('-')[-1] in messages_for_bot_IDs:
224
 
225
  originalQuote = msg['reply']
226
- quote = remove_non_english_russian_chars(msg['reply'])
 
 
 
 
227
  message = ""
228
 
229
- while not message:
 
 
 
230
  message = WarClient.getReply(message=quote)
 
 
 
 
231
 
232
  # Post-processing fixes:
233
  message = fixString(message)
234
-
235
- print('Quote: ', originalQuote)
236
  print('Reply: ', message)
237
 
238
  login(username=username, password=password, thread_url=thread_url)
 
223
  if msg['messageID'].split('-')[-1] in messages_for_bot_IDs:
224
 
225
  originalQuote = msg['reply']
226
+ if originalQuote == "": # Just images, no text
227
+ continue
228
+ else:
229
+ quote = remove_non_english_russian_chars(msg['reply'])
230
+
231
  message = ""
232
 
233
+ print('Quote: ', originalQuote)
234
+
235
+ FailureCounter = 0 # In case there is a bug in the model
236
+ while (not message) and (FailureCounter<3):
237
  message = WarClient.getReply(message=quote)
238
+ FailureCounter+=1
239
+
240
+ if FailureCounter == 3:
241
+ continue # Skip that answer
242
 
243
  # Post-processing fixes:
244
  message = fixString(message)
 
 
245
  print('Reply: ', message)
246
 
247
  login(username=username, password=password, thread_url=thread_url)