Upload 5 files

Several updates

WarBot.py

@@ -85,7 +85,7 @@ def get_response(quote:str,model,tokenizer,model_punct,temperature=0.2):
         user_inpit_ids = tokenizer.encode(f"|0|{get_length_param(quote, tokenizer)}|" \
                                                       + quote + tokenizer.eos_token, return_tensors="pt")
     except:
-        return "" # Exception in tokenization
+        return "Exception in tokenization" # Exception in tokenization
 
     chat_history_ids = user_inpit_ids # To be changed
 
@@ -110,7 +110,7 @@ def get_response(quote:str,model,tokenizer,model_punct,temperature=0.2):
                     #device='cpu'
                 )
     except:
-        return "" # Exception in generation
+        return "Exception" # Exception in generation
 
     response = tokenizer.decode(output_id[0], skip_special_tokens=True)
     response = removeSigns(response)

WarClient_old.py

@@ -0,0 +1,18 @@
+import socket
+
+HOST = '129.159.146.88'
+PORT = 5000
+
+#message = "Если посмотреть на небо, можно увидеть как улетает башня от Т-72"
+#message = "Тестовый бот WarOnline - доложить о работоспособности систем!"
+
+with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
+    client_socket.connect((HOST, PORT))
+    client_socket.sendall(message.encode())
+    print('Wait...')
+    data = client_socket.recv(1024)
+    try:
+        received_string = data.decode('utf-8')
+        print(f'Received string from server: {received_string}')
+    except: #sometimes there is a problem with the decoding
+        print('decoding error, please try again')

WarOnline_Chat.py

@@ -8,6 +8,7 @@ import urllib.request as urllib
 import warnings
 import schedule
 import time
+from tqdm import tqdm
 warnings.filterwarnings("ignore")
 
 # Define the login URL and the thread URL
@@ -110,8 +111,7 @@ def allQuotesFor(thread_url=thread_url, username=username, startingPage=1):
         response = requests.get(thread_url + 'page-' + str(page))
         if response.status_code == 200:
 
-            # Payload of the function
-            response = session.get(thread_url)
+            # Core of the function
             html_content = response.content
 
             # Parse the HTML content using BeautifulSoup
@@ -133,8 +133,13 @@ def allQuotesFor(thread_url=thread_url, username=username, startingPage=1):
                         if matchID:
                             messageID = matchID.group(1)
 
-                        reply = data.text.split("Click to expand...")[-1].replace('\n', ' ').strip()
+                        if 'Click' in data.text:
+                            reply = data.text.split("Click to expand...")[-1].replace('\n', ' ').strip()
+                        elif "Нажмите" in data.text:
+                            reply = data.text.split("Нажмите для раскрытия...")[-1].replace('\n', ' ').strip()
+
                         allquotes.append({'reply': reply, 'messengerName': messengerName, 'messageID': messageID})
+
                     except:
                         continue # There was no text in quote, move next
 
@@ -151,28 +156,42 @@ def allQuotesFor(thread_url=thread_url, username=username, startingPage=1):
 def WarOnlineBot():
     # Get All Quotes
     quotes = allQuotesFor(thread_url=thread_url, username=username, startingPage=1)
-
-    # Search for quotes withou the relpy and answer them
+    repliedMessageIDs = []
     for quote in quotes:
-        reply = allQuotesFor(thread_url=thread_url, username=quote['messengerName'], startingPage=1)
-        if not reply:
+        replies = allQuotesFor(thread_url=thread_url, username=quote['messengerName'], startingPage=1)
+        for reply in replies: # Remove all non-relevant and previously answered replies
+
+            if reply['messengerName']!=username or reply['messageID'] in repliedMessageIDs:
+                replies.remove(reply)
+            else:
+                repliedMessageIDs.append(reply['messageID'])
+                repliedMessageIDs = list(set(repliedMessageIDs))
+            # There is still some strange error. Will take a look later
+
+        message = ""
+        if not replies: # Meaning that there are no answers
+            while not message:
+                message = WarClient.getReply(message=quote['reply'])
             print('Quote: ', quote['reply'])
-            print(WarClient.getReply(message=quote['reply']))
+            print('Reply: ',message)
+            post(message=message, thread_url=thread_url, post_url=post_url, quoted_by=quote['messengerName'],
+                 quote_text=quote['reply'])
+            print('posted the message to the forum')
+            time.sleep(5) # Standby time for server load release
 
 
 if __name__ == '__main__':
-    timeout = 1 # min
-    login(username=username, password=password, thread_url=thread_url)
-
+    timeout = 2 # min
     #post(message=message, thread_url=thread_url, post_url=post_url,quoted_by='Василий Пупкин',quote_text='Testing the XenForo response mechanism')
 
-    # Set up schedule function
-    schedule.every(timeout).minutes.do(WarOnlineBot)
-
     # Start the scheduler
     while True:
-        schedule.run_pending()
-        time.sleep(1)
-
-
-
+        login(username=username, password=password, thread_url=thread_url)
+        print("logged in")
+        WarOnlineBot()
+        p_bar = tqdm(range(60 * timeout))
+
+        for i in p_bar:
+            p_bar.update(1)
+            p_bar.refresh()
+            time.sleep(1)