Add CORS headers to dream server to ease integration with third-party web interfaces

Author: SebastianAigner

ldm/dream/server.py

@@ -16,6 +16,8 @@ class DreamServer(BaseHTTPRequestHandler):
     def do_GET(self):
         if self.path == "/":
             self.send_response(200)
+            self.send_header("Access-Control-Allow-Origin", "*")
+            self.send_header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept")
             self.send_header("Content-type", "text/html")
             self.end_headers()
             with open("./static/dream_web/index.html", "rb") as content:
@@ -33,6 +35,8 @@ def do_GET(self):
         elif self.path == "/cancel":
             self.canceled.set()
             self.send_response(200)
+            self.send_header("Access-Control-Allow-Origin", "*")
+            self.send_header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept")
             self.send_header("Content-type", "application/json")
             self.end_headers()
             self.wfile.write(bytes('{}', 'utf8'))
@@ -55,6 +59,8 @@ def do_GET(self):
 
     def do_POST(self):
         self.send_response(200)
+        self.send_header("Access-Control-Allow-Origin", "*")
+        self.send_header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept")
         self.send_header("Content-type", "application/json")
         self.end_headers()
 
@@ -196,6 +202,11 @@ def image_progress(sample, step):
             print(f"Canceled.")
             return
 
+    def do_OPTIONS(self):
+        self.send_response(200)
+        self.send_header("Access-Control-Allow-Origin", "*")
+        self.send_header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept")
+        self.end_headers()
 
 class ThreadingDreamServer(ThreadingHTTPServer):
     def __init__(self, server_address):