File size: 1,387 Bytes
a4b70d9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
from ..errors import MissingRequirementsError
try:
from .server.website import Website
from .server.backend_api import Backend_Api
from .server.app import create_app
import_error = None
except ImportError as e:
import_error = e
def get_gui_app(demo: bool = False, timeout: int = None, stream_timeout: int = None):
if import_error is not None:
raise MissingRequirementsError(f'Install "gui" requirements | pip install -U g4f[gui]\n{import_error}')
app = create_app()
app.demo = demo
app.timeout = timeout
app.stream_timeout = stream_timeout
site = Website(app)
for route in site.routes:
app.add_url_rule(
route,
view_func=site.routes[route]['function'],
methods=site.routes[route]['methods'],
)
backend_api = Backend_Api(app)
for route in backend_api.routes:
app.add_url_rule(
route,
view_func = backend_api.routes[route]['function'],
methods = backend_api.routes[route]['methods'],
)
return app
def run_gui(host: str = '0.0.0.0', port: int = 8080, debug: bool = False) -> None:
config = {
'host' : host,
'port' : port,
'debug': debug
}
app = get_gui_app()
print(f"Running on port {config['port']}")
app.run(**config)
print(f"Closing port {config['port']}")
|