gpt4free/g4f/gui/__init__.py
hlohaus 1edd0fff17 Refactor model handling and improve timeout functionality
- Removed empty string mapping from model_map in AnyModelProviderMixin.
- Updated clean_name function to exclude 'chat' from version patterns.
- Added stream_timeout parameter to AsyncGeneratorProvider for more flexible timeout handling.
- Enhanced chunk yielding in AsyncAuthedProvider to support stream_timeout, allowing for better control over asynchronous responses.
2025-09-04 18:11:05 +02:00

47 lines
1.4 KiB
Python

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']}")