Commit b2e0d8ba authored by Bruno Seoane's avatar Bruno Seoane
Browse files

Remove folder endpoint

parent 8320963d
Loading
Loading
Loading
Loading
+0 −9
Original line number Diff line number Diff line
@@ -32,7 +32,6 @@ class Api:
        self.app.add_api_route("/sdapi/v1/img2img", self.img2imgapi, methods=["POST"], response_model=ImageToImageResponse)
        self.app.add_api_route("/sdapi/v1/extra-single-image", self.extras_single_image_api, methods=["POST"], response_model=ExtrasSingleImageResponse)
        self.app.add_api_route("/sdapi/v1/extra-batch-images", self.extras_batch_images_api, methods=["POST"], response_model=ExtrasBatchImagesResponse)
        self.app.add_api_route("/sdapi/v1/extra-folder-images", self.extras_folder_processing_api, methods=["POST"], response_model=ExtrasBatchImagesResponse)

    def text2imgapi(self, txt2imgreq: StableDiffusionTxt2ImgProcessingAPI):
        sampler_index = sampler_to_index(txt2imgreq.sampler_index)
@@ -126,14 +125,6 @@ class Api:

        return ExtrasBatchImagesResponse(images=list(map(encode_pil_to_base64, result[0])), html_info=result[1])
    
    def extras_folder_processing_api(self, req:ExtrasFoldersRequest):
        reqDict = setUpscalers(req)

        with self.queue_lock:
            result = run_extras(extras_mode=2, image=None, image_folder=None, **reqDict)

        return ExtrasBatchImagesResponse(images=list(map(encode_pil_to_base64, result[0])), html_info=result[1])

    def pnginfoapi(self):
        raise NotImplementedError

+1 −5
Original line number Diff line number Diff line
@@ -149,7 +149,3 @@ class ExtrasBatchImagesRequest(ExtrasBaseRequest):

class ExtrasBatchImagesResponse(ExtraBaseResponse):
    images: list[str] = Field(title="Images", description="The generated images in base64 format.")
 No newline at end of file

class ExtrasFoldersRequest(ExtrasBaseRequest):
    input_dir: str = Field(title="Input directory", description="Directory path from where to take the images")
    output_dir: str = Field(title="Output directory", description="Directory path to put the processsed images into")