| import os | |
| from fastapi import FastAPI, Response, Request | |
| from fastapi.responses import RedirectResponse | |
| #from g4f.image import images_dir | |
| import g4f.api | |
| import g4f.Provider | |
| import demo | |
| from demo.BackendApi import BackendApi | |
| g4f.Provider.__map__["Feature"] = BackendApi | |
| #g4f.models.demo_models | |
| #from g4f.image import copy_images | |
| def create_app(): | |
| g4f.debug.logging = True | |
| g4f.api.AppConfig.gui = True | |
| g4f.api.AppConfig.demo = True | |
| app = FastAPI() | |
| # Add CORS middleware | |
| app.add_middleware( | |
| g4f.api.CORSMiddleware, | |
| allow_origin_regex=".*", | |
| allow_credentials=True, | |
| allow_methods=["*"], | |
| allow_headers=["*"], | |
| ) | |
| api = g4f.api.Api(app) | |
| api.register_routes() | |
| api.register_authorization() | |
| api.register_validation_exception_handler() | |
| # @app.get("/download/{filename}", response_class=RedirectResponse) | |
| # async def download(filename, request: Request): | |
| # filename = os.path.basename(filename) | |
| # if "." not in filename: | |
| # target = os.path.join(images_dir, filename) | |
| # filename = f"{filename}.jpg" | |
| # target = os.path.join(images_dir, filename) | |
| # if not os.path.exists(target): | |
| # url = str(request.query_params).split("url=", 1)[1] | |
| # if url: | |
| # source_url = url.replace("%2F", "/").replace("%3A", ":").replace("%3F", "?") | |
| # await copy_images( | |
| # [source_url], | |
| # target=target, | |
| # ssl=False, | |
| # headers=demo.headers if source_url.startswith(BackendApi.url) else None) | |
| # if not os.path.exists(target): | |
| # return Response(status_code=404) | |
| # return RedirectResponse(f"/images/{filename}") | |
| gui_app = g4f.api.WSGIMiddleware(g4f.api.get_gui_app(g4f.api.AppConfig.demo)) | |
| app.mount("/", gui_app) | |
| return app | |
| app = create_app() |