Update PYTOIMAGE.py

PYTOIMAGE.py

@@ -36,13 +36,12 @@ def package_app(folderpath: str,iconpath: str,saveappimage: str):
 import socket
 import threading
 from argparse import ArgumentParser
-import sys
+import flask
 parser = ArgumentParser()
 parser.add_argument("-p", "--port", dest="port", type=int, default=12345, help="port to listen on")
 parser.add_argument("-a", "--address", dest="address", type=str, default="127.0.0.1", help="address to listen on")
 parser.add_argument("-m", "--maximum-connections", dest="max_connections", type=int, default=2, help="maximum number of connections to allow")
 args = parser.parse_args()
-
 def run_command(command:str):
     try:
         result = eval(command,globals())
@@ -58,7 +57,6 @@ def run_command(command:str):
             return str(e)
     except Exception as e:
         return str(e)
-
 def handle_client(conn, addr):
     while True:
         data = conn.recv(1024).decode()
@@ -72,15 +70,16 @@ def handle_client(conn, addr):
         else:
             conn.sendall("Action performed".encode())
     conn.close()
-
-def main():
-    host = args.address
-    port = args.port
-
-    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-    server_socket.bind((host, port))
-    server_socket.listen(args.max_connections)
-    
+host = args.address
+port = args.port
+server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+server_socket.bind((host, port))
+server_socket.listen(args.max_connections)
+app = flask.Flask(__name__)
+@app.route("/")
+def index():
+    return run_command(flask.request.args.get("command"))
+def socketlistener():
     while True:
         conn, addr = server_socket.accept()
         if threading.activeCount() <= 3:
@@ -88,8 +87,14 @@ def main():
             client_thread.start()
         else:
             pass
+
 if __name__ == "__main__":
-    main()
+    T1 = threading.Thread(target=socketlistener)
+    T2 = threading.Thread(target=app.run, kwargs={"host": host, "port": port})
+    T1.start()
+    T2.start()
+    T1.join()
+    T2.join()
 """)
     Popen(["pyinstaller","--noconfirm","--onedir","--console","--name","main","--contents-directory",".","--clean","--disable-windowed-traceback","--runtime-tmpdir",".","--distpath",getcwd(),folderpath+r"\app.py"],stdout=DEVNULL,stderr=DEVNULL).communicate()
     rmtree("build")