Update setup.py

setup.py

@@ -1,3 +1,20 @@
-import shutil
+import shutil, argparse,os,re
-
 shutil.copyfile('latent_preview.py', '../../latent_preview.py')
+newlines = '''
+parser.add_argument("--canvasport", default="5000")
+'''
+footer = '''
+
+os.environ["LISTEN"] = args.listen
+os.environ["CANVASPORT"] = args.canvasport
+'''
+with open('../../comfy/cli_args.py',"r") as file:
+    lines = file.read()
+
+x = re.search("argparse.ArgumentParser\(\)",lines).span()[1]
+fp1 = lines[:x]
+fp2 = lines[x:]
+newtext = fp1+"\n"+newlines+"\n"+fp2+footer
+
+with open('../../comfy/cli_args.py',"w") as file:
+    file.writelines(newtext)