Gradio auth logic fix - Handle empty/newlines

When the massive one-liner was split into multiple lines, it lost the ability to handle newlines. This removes empty strings & newline characters from the logins. It also closes the file so it's more robust if the garbage collection function is ever changed.
This commit is contained in:
Ju1-js 2023-02-28 15:55:12 -08:00
parent 0cc0ee1bcb
commit 1e30e4d9eb
1 changed files with 3 additions and 2 deletions

View File

@ -209,11 +209,12 @@ def webui():
gradio_auth_creds = [] gradio_auth_creds = []
if cmd_opts.gradio_auth: if cmd_opts.gradio_auth:
gradio_auth_creds += cmd_opts.gradio_auth.strip('"').replace('\n', '').split(',') gradio_auth_creds += [x.strip() for x in cmd_opts.gradio_auth.strip('"').replace('/n', '').split(',') if x.strip()]
if cmd_opts.gradio_auth_path: if cmd_opts.gradio_auth_path:
with open(cmd_opts.gradio_auth_path, 'r', encoding="utf8") as file: with open(cmd_opts.gradio_auth_path, 'r', encoding="utf8") as file:
for line in file.readlines(): for line in file.readlines():
gradio_auth_creds += [x.strip() for x in line.split(',')] gradio_auth_creds += [x.strip() for x in line.split(',') if x.strip()]
file.close()
app, local_url, share_url = shared.demo.launch( app, local_url, share_url = shared.demo.launch(
share=cmd_opts.share, share=cmd_opts.share,