neuro_llc/ComfyBox
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Merge pull request #126 from space-nuko/server-args

Browse Source 
Add arguments to serve.py
This commit is contained in:
space-nuko
2023-07-07 09:49:27 -05:00
committed by GitHub
parent 228ea20dcb 33ed379a98
commit 3b49bac47b
1 changed files with 32 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
bin/serve.py
View File

@@ -2,15 +2,19 @@
import http.server
import socketserver
import argparse
PORT = 8000
parser = argparse.ArgumentParser()
parser.add_argument("-l", "--listen", type=str, default="localhost", help="Listen address for ComfyBox server")
parser.add_argument("-p", "--port", type=int, default=8000, help="Port for ComfyBox server")
args = parser.parse_args()
message = f"""Starting ComfyBox.
Be sure you've started ComfyUI already using this command:
python main.py --enable-cors-header
Serving at http://localhost:{PORT}...
Serving at http://{args.listen}:{args.port}...
"""
# python -m http.server will sometimes send incorrect MIME types.
@@ -19,33 +23,34 @@ Serving at http://localhost:{PORT}...
# Hopefully this will cover everything.
class HttpRequestHandler(http.server.SimpleHTTPRequestHandler):
extensions_map = {
'': 'application/octet-stream',
'.manifest': 'text/cache-manifest',
'.html': 'text/html',
'.png': 'image/png',
'.jpg': 'image/jpg',
'.jpeg': 'image/jpeg',
'.gif': 'image/gif',
'.svg': 'image/svg+xml',
'.css': 'text/css',
'.js': 'application/x-javascript',
'.mjs': 'application/x-javascript',
'.cjs': 'application/x-javascript',
'.wasm': 'application/wasm',
'.json': 'application/json',
'.xml': 'application/xml',
'.xml': 'application/xml',
'.pdf': 'application/pdf',
'.webp': 'image/webp',
'.avif': 'image/avif',
'.heic': 'image/heic',
'.heif': 'image/heif',
'.mp3': 'audio/mpeg',
'.mp4': 'video/mp4',
'.m4v': 'video/mp4'
"": "application/octet-stream",
".manifest": "text/cache-manifest",
".html": "text/html",
".png": "image/png",
".jpg": "image/jpg",
".jpeg": "image/jpeg",
".gif": "image/gif",
".svg": "image/svg+xml",
".css": "text/css",
".js": "application/x-javascript",
".mjs": "application/x-javascript",
".cjs": "application/x-javascript",
".wasm": "application/wasm",
".json": "application/json",
".xml": "application/xml",
".xml": "application/xml",
".pdf": "application/pdf",
".webp": "image/webp",
".avif": "image/avif",
".heic": "image/heic",
".heif": "image/heif",
".mp3": "audio/mpeg",
".mp4": "video/mp4",
".m4v": "video/mp4",
}
httpd = socketserver.TCPServer(("localhost", PORT), HttpRequestHandler)
httpd = socketserver.TCPServer((args.listen, args.port), HttpRequestHandler)
try:
print(message)