idk

Dockerfile

@@ -14,8 +14,6 @@ RUN apt update && \
                    python3-pip \
                    python3.8-venv && \
     rm -rf /var/lib/apt/lists
-
-RUN python3 -m venv /opt/venv
 WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
 RUN python3 -m pip install --no-cache-dir --upgrade pip && \

api/core/app.py

@@ -1,8 +1,9 @@
 class Demos:
     def __init__(self):
-        from fastapi import FastAPI, HTTPException
+        from fastapi import FastAPI, HTTPException, Response
         self.api = FastAPI
         self.exception = HTTPException
+        self.api_response = Response
     def validate_apikey(self,api_key)->bool:
         __validation = True
         return __validation

api/core/controllers/text2image.py

@@ -1,45 +1,61 @@
 from diffusers import DiffusionPipeline as Pipe
 import torch
 
+class Converter:
+    import io
+    from PIL import Image
+
+    img = Image.open(fh, mode='r')
+    roi_img = img.crop(box)
+
+    img_byte_arr = io.BytesIO()
+    roi_img.save(img_byte_arr, format='PNG')
+    img_byte_arr = img_byte_arr.getvalue()
+
 class Generador:
-    def using_runway_sd_15(prompt:str)->list:
-        try:
+    def img_to_bytes(image) -> bytes:
+        import io
+        _imgByteArr = io.BytesIO()
+        image.save(_imgByteArr, format=image.format)
+        return _imgByteArr.getvalue()
+    def using_runway_sd_15(prompt:str)->bytes|None:
+        try:            
             _generador = Pipe.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
             _generador.to("cuda")
             _imagen = _generador(prompt).images[0]
-            _response = list(_imagen.getdata())
+            _response = Generador.img_to_bytes(img=_imagen)
         except Exception as e:
-            _response = list(str(e))
+            _response = None
         finally:
             return _response
-    def using_stability_sd_21(prompt:str)->list:
+    def using_stability_sd_21(prompt:str)->bytes|None:
         try:
             _generador = Pipe.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
             _generador.to("cuda")
             _imagen = _generador(prompt).images[0]
-            _response = list(_imagen.getdata())
+            _response = Generador.img_to_bytes(img=_imagen)
         except Exception as e:
-            _response = list(str(e))
+            _response = None
         finally:
             return _response
-    def using_realistic_v14(prompt:str)->list:
+    def using_realistic_v14(prompt:str)->bytes|None:
         try:
             _generador = Pipe.from_pretrained("SG161222/Realistic_Vision_V1.4", torch_dtype=torch.float16)
             _generador.to("cuda")
             _imagen = _generador(prompt).images[0]
-            _response = list(_imagen.getdata())
+            _response = Generador.img_to_bytes(img=_imagen)
         except Exception as e:
-            _response = list(str(e))
+            _response = None
         finally:
             return _response
-    def using_prompthero_openjourney(prompt:str)->list:
+    def using_prompthero_openjourney(prompt:str)->bytes|None:
         try:
             _generador = Pipe.from_pretrained("prompthero/openjourney", torch_dtype=torch.float16)
             _generador.to("cuda")
             _imagen = _generador(prompt).images[0]
-            _response = list(_imagen.getdata())
+            _response = Generador.img_to_bytes(img=_imagen)
         except Exception as e:
-            _response = list(str(e))
+            _response = None
         finally:
             return _response
     

api/main.py

@@ -82,7 +82,7 @@ def get_resumen(data:dict) -> dict:
     finally:
         return __response
     
-@api.post("/texto_a_imagen/", status_code=201)
+@api.post("/texto_a_imagen/", status_code=201, responses = {201: {"content": {"image/png": {}}}} ,response_class=__main.api_response)
 def get_text2img(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
@@ -97,7 +97,7 @@ def get_text2img(data:dict) -> dict:
         #To-do ->agregar mas información en el error fecha, usuario, reqs
         raise __main.exception(status_code = 403, datail=e)
     finally:
-        return __response
+        return __main.api_response(content=__response, media_type="image/png")
 
 @api.post("/texto_a_video/", status_code=201)
 def get_text2video(data:dict) -> dict:
@@ -130,3 +130,8 @@ def get_text2speach(data:dict) -> dict:
         raise __main.exception(status_code = 403, datail=e)
     finally:
         return __response
+
+
+
+
+