rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Initial flaskwebgui support

Browse Source
This commit is contained in:
Henk
2022-06-17 13:49:03 +02:00
parent b0a01962ab
commit f112fc3493
1 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
aiserver.py
View File

@ -323,6 +323,7 @@ class vars:
actionmode = 1 actionmode = 1
dynamicscan = False dynamicscan = False
host = False host = False
flaskwebgui = False
nopromptgen = False nopromptgen = False
rngpersist = False rngpersist = False
nogenmod = False nogenmod = False
@ -2765,6 +2766,8 @@ def do_connect():
emit('from_server', {'cmd': 'connected', 'smandelete': vars.smandelete, 'smanrename': vars.smanrename, 'modelname': getmodelname()}) emit('from_server', {'cmd': 'connected', 'smandelete': vars.smandelete, 'smanrename': vars.smanrename, 'modelname': getmodelname()})
if(vars.host): if(vars.host):
emit('from_server', {'cmd': 'runs_remotely'}) emit('from_server', {'cmd': 'runs_remotely'})
if(vars.flaskwebgui):
emit('from_server', {'cmd': 'flaskwebgui'})
if(vars.allowsp): if(vars.allowsp):
emit('from_server', {'cmd': 'allowsp', 'data': vars.allowsp}) emit('from_server', {'cmd': 'allowsp', 'data': vars.allowsp})
@ -5898,14 +5901,25 @@ if __name__ == "__main__":
vars.serverstarted = True vars.serverstarted = True
socketio.run(app, host='0.0.0.0', port=port) socketio.run(app, host='0.0.0.0', port=port)
else: else:
if args.unblock:
import webbrowser import webbrowser
webbrowser.open_new('http://localhost:{0}'.format(port)) webbrowser.open_new('http://localhost:{0}'.format(port))
print("{0}Server started!\nYou may now connect with a browser at http://127.0.0.1:{1}/{2}" print("{0}Server started!\nYou may now connect with a browser at http://127.0.0.1:{1}/{2}"
.format(colors.GREEN, port, colors.END)) .format(colors.GREEN, port, colors.END))
vars.serverstarted = True vars.serverstarted = True
if args.unblock:
socketio.run(app, port=port, host='0.0.0.0') socketio.run(app, port=port, host='0.0.0.0')
else: else:
try:
from flaskwebgui import FlaskUI
vars.serverstarted = True
vars.flaskwebgui = True
FlaskUI(app, socketio=socketio, start_server="flask-socketio", maximized=True, close_server_on_exit=False).run()
except:
import webbrowser
webbrowser.open_new('http://localhost:{0}'.format(port))
print("{0}Server started!\nYou may now connect with a browser at http://127.0.0.1:{1}/{2}"
.format(colors.GREEN, port, colors.END))
vars.serverstarted = True
socketio.run(app, port=port) socketio.run(app, port=port)
else: else: