update img2img test1

api/core/app.py

@@ -1,22 +1,24 @@
 class Demos:
     def __init__(self):
-        from fastapi import FastAPI, HTTPException, Response
+        from fastapi import FastAPI, HTTPException, Response, File
+        from typing import Annotated
         self.api = FastAPI
         self.exception = HTTPException
         self.api_response = Response
+        self.annotated = Annotated
+        self.fileupload = File
     def validate_apikey(self,api_key)->bool:
         __validation = True
         return __validation
     @staticmethod
     def text_to_img(texto:str=None, model:str="PROMPTHERO")->bytes:
-        """Genera un BITARRAY de la imagen con el texto dado.
+        """Genera un Bytes de la imagen con el texto dado.
 
         args:
             texto           (str)   :   Texto para generar imagen
         return:
-            _img            (str)   :   Imagen en BITARRAY
+            _img            (bytes) :   Imagen en Bytes
         """
-        
         from api.core.controllers.text2image import Generador
         _imagen = str()
         if "RUNWAY" in model.upper():
@@ -66,5 +68,27 @@ class Demos:
         """
         _image = str()
         return _image
+    @staticmethod
+    def img_to_img(texto:str=None, model:str="PROMPTHERO")->bytes:
+        """Genera un Bytes de la imagen con el texto dado.
+
+        args:
+            texto           (str)   :   Texto para generar imagen
+        return:
+            _img            (bytes) :   Imagen en Bytes
+        """
+        from api.core.controllers.text2image import Generador
+        _imagen = str()
+        if "RUNWAY" in model.upper():
+            _imagen = Generador.using_runway_sd_15(prompt=texto)
+        elif "STABILITY" in model.upper():
+            _imagen = Generador.using_stability_sd_21(prompt=texto)
+        elif "REALISTIC" in model.upper():
+            _imagen = Generador.using_realistic_v14(prompt=texto)
+        elif "PROMPTHERO" in model.upper():
+            _imagen = Generador.using_prompthero_openjourney(prompt=texto)
+        else:
+            _imagen = bytes("error", 'utf-8')
+        return _imagen
     
 

api/main.py

@@ -54,4 +54,19 @@ def get_text2speach(data:dict) -> dict:
 
 
 
-    
+
+@api.post("/imagen_a_imagen/", status_code=201, responses = {201: {"content": {"image/png": {}}}} ,response_class=__main.api_response)
+def get_text2img(data:dict, file: __main.annotated[bytes, __main.fileupload()]):
+    __response=dict({"request_data":data})
+    try:
+        if data and 'texto' in data and 'modelo' in data and file is not None:
+            __response['original']= data.get('texto')
+            __image =  file
+        else:
+            raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
+    except Exception as e:
+        print(e)
+        #To-do ->agregar mas información en el error fecha, usuario, reqs
+        raise __main.exception(status_code = 403, datail=str(e))
+    finally:
+        return __main.api_response(content=__image, media_type="image/png")