Merge pull request #96 from TesseractCat/main

Keep a log of requests for dream_web
This commit is contained in:
Lincoln Stein 2022-08-25 21:42:17 -04:00 committed by GitHub
commit d051d86df6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -85,6 +85,12 @@ class DreamServer(BaseHTTPRequestHandler):
print(f"Prompt generated with output: {outputs}")
post_data['initimg'] = '' # Don't send init image back
# Append post_data to log
with open("./outputs/img-samples/dream_web_log.txt", "a") as log:
for output in outputs:
log.write(f"{output[0]}: {json.dumps(post_data)}\n")
outputs = [x + [post_data] for x in outputs] # Append config to each output
result = {'outputs': outputs}
self.wfile.write(bytes(json.dumps(result), "utf-8"))