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()