Files
ComfyBox/bin/serve.py
2023-05-11 11:32:24 -05:00

55 lines
1.6 KiB
Python

#!/usr/bin/env python
import http.server
import socketserver
PORT = 8000
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}...
"""
# python -m http.server will sometimes send incorrect MIME types.
# But most people have a system python already installed so this method is most convenient.
# So here is a list of MIME types for the server to send back per file extension.
# 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'
}
httpd = socketserver.TCPServer(("localhost", PORT), HttpRequestHandler)
try:
print(message)
httpd.serve_forever()
except KeyboardInterrupt:
pass