Upload main.py

GUI with Client. Added history...

main.py

@@ -0,0 +1,64 @@
+# GUI with Client
+import gradio as gr
+import socket
+
+HOST = '129.159.146.88'
+PORT = 5000
+
+dialog_history = []
+MAX_WORDS = 100
+
+
+def truncate_string(text):
+    words = text.split()
+    if len(words) > MAX_WORDS:
+        words = words[-MAX_WORDS:]
+    return ' '.join(words)
+
+def sendToServer(message):
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
+        # This mechanism is not working smoothly, meanwhile.
+        # Take last 4 messages (history of conversation)
+        dialog_history.append(message)
+        message = " ".join(dialog_history[-4:])
+        message = truncate_string(message)
+
+        client_socket.connect((HOST, PORT))
+        client_socket.sendall(message.encode())
+
+        try:
+            data = client_socket.recv(1024)
+
+            reply = data.decode('utf-8')
+            dialog_history.append(reply)
+            return reply
+            # print(f'Received string from server: {received_string}')
+        except:  # sometimes there is a problem with the decoding
+            return "Бот устал и должен отдохнуть"
+            # print('decoding error, please try again')
+        finally:
+            client_socket.close()
+
+
+def clear_textbox():
+    return ""
+
+
+with gr.Blocks() as WarBot:
+    gr.Markdown(
+        """   
+        # Боевой Чат-Бот портала WarOnline
+        Пока-что это бредогенератор, тренированый на диалогах форума,<br>
+        Есть куча багов. Но мы работаем над улучшениями! :)
+        """)
+
+    with gr.Row():
+        input = gr.Textbox(lines=5, placeholder="Введите сообщение...", label="Вопрос:")
+        output = gr.Textbox(label="Ответ:", lines=5)
+
+    send_btn = gr.Button("Послать Сообщение")
+    send_btn.click(fn=sendToServer, inputs=input, outputs=output)
+    clr_btn = gr.Button("Очистить")
+    clr_btn.click(fn=clear_textbox, outputs=input)
+
+WarBot.launch()