update methods v0

api/core/app.py

@@ -1,4 +1,3 @@
-
 class Demos:
     def __init__(self):
         from fastapi import FastAPI, HTTPException
@@ -17,15 +16,17 @@ class Demos:
         return:
             _texto          (str)   :   Texto extraído desde la fuente dada.
         """
+        from api.core.controllers.extractor import TextFromURL
         if from_url:
-            _texto = str("URL")
+            with TextFromURL(url=from_url) as web_content:
+                _texto = web_content.get_raw().get_text() if web_content.is_ok() else web_content.get_error()
         elif from_pdf:
             _texto = str("PDF")
         else:
             _texto = str("Ninguna opción seleccionada")
         return _texto
     @staticmethod
-    def generar_bloques(texto:str=None,s:int=1500)->list:
+    def generar_bloques(texto:str=None,size:int=1500)->list:
         """Genera bloques de texto de longitud S, a partir de un texto plano.
 
         args:
@@ -34,7 +35,8 @@ class Demos:
         return:
             _bloques        (list)  :   Lista de bloques de textos de longitud s.
         """
-        _bloques = list()
+        from api.core.controllers.handlers import Generator
+        _bloques = Generator.get_bloques(from_text=texto,size=size)
         return _bloques
     @staticmethod
     def traducir(texto:str=None, idioma:str="EN/ES")->str:
@@ -46,13 +48,14 @@ class Demos:
         return:
             _traduccion     (str)   :   Traducción del texto
         """
+        from api.core.controllers.text2text import Traductor
         _traduccion = str()
         if "EN/ES" in idioma.upper():
-            _traduccion = "input inglés, output español"
+            _traduccion = Traductor.EN_ES(texto=texto)
         elif "ES/EN" in idioma.upper():
-            _traduccion = "input español, output inglés"
+            _traduccion = Traductor.ES_EN(texto=texto)
         elif "AR/ES" in idioma.upper():
-            _traduccion = "input arabe, output español"
+            _traduccion = Traductor.AR_ES(texto=texto)
         else:
             _traduccion = "Idioma no válido"
         return _traduccion
@@ -65,10 +68,11 @@ class Demos:
         return:
             _resumen        (str)   :   Resumen generado
         """
-        _bloques = list()
-        return _bloques
+        from api.core.controllers.text2text import Abstractor
+        _resumen = Abstractor.resumen(texto=texto)
+        return _resumen
     @staticmethod
-    def text_to_img(texto:str=None)->str:
+    def text_to_img(texto:str=None, model:str="PROMPTHERO")->list:
         """Genera un BITARRAY de la imagen con el texto dado.
 
         args:
@@ -76,8 +80,20 @@ class Demos:
         return:
             _img            (str)   :   Imagen en BITARRAY
         """
-        _img = str()
-        return _img
+        
+        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 = list(["error"])
+        return _imagen
     @staticmethod
     def text_to_video(texto:str=None)->str:
         """Genera un BITARRAY del video con el texto dado.

api/core/controllers/extractor.py

@@ -0,0 +1,24 @@
+class TextFromURL:
+  __raw = None
+  def __init__(self, url:str):
+    self.__source=url
+  def __enter__(self):
+    try:
+      import requests as REQS
+      import bs4 as BS4
+      self.__raw = BS4.BeautifulSoup(REQS.get(self.get_source()).content, "html.parser")
+    except Exception as er:
+      print(er)
+      self.__error = er
+    finally:
+      return self    
+  def __exit__(self, *args):
+    [print(e) for e in args if e is not None]
+  def get_source(self):
+    return self.__source
+  def get_raw(self):
+    return self.__raw
+  def get_error(self):
+    return self.__error
+  def is_ok(self):
+    return True if self.get_raw() else False

api/core/controllers/handlers.py

@@ -0,0 +1,13 @@
+class Generator:
+    def get_bloques(from_text:str, size:int=1500):
+        _bloques = list()
+        _t = str(from_text)
+        if len(_t)>size:
+            _ = int(len(_t)/2)
+            _ts = [_t[n:n+_] for n in range(0,len(_t),_)]
+            for m in [0,1]:
+                _np = [nt for nt in Generator.get_bloques(from_text=_ts[m],size=size)]
+                _bloques+=_np
+        else:
+            _bloques.append(_t)
+        return _bloques

api/core/controllers/text2image.py

@@ -0,0 +1,25 @@
+from diffusers import DiffusionPipeline as Pipe
+import torch
+
+class Generador:
+    def using_runway_sd_15(prompt:str)->list:
+        _generador = Pipe.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
+        _generador.to("cuda")
+        _imagen = _generador(prompt).images[0]
+        return list(_imagen.getdata())
+    def using_stability_sd_21(prompt:str)->list:
+        _generador = Pipe.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
+        _generador.to("cuda")
+        _imagen = _generador(prompt).images[0]
+        return list(_imagen.getdata())
+    def using_realistic_v14(prompt:str)->list:
+        _generador = Pipe.from_pretrained("SG161222/Realistic_Vision_V1.4", torch_dtype=torch.float16)
+        _generador.to("cuda")
+        _imagen = _generador(prompt).images[0]
+        return list(_imagen.getdata())
+    def using_prompthero_openjourney(prompt:str)->list:
+        _generador = Pipe.from_pretrained("prompthero/openjourney", torch_dtype=torch.float16)
+        _generador.to("cuda")
+        _imagen = _generador(prompt).images[0]
+        return list(_imagen.getdata())
+    

api/core/controllers/text2text.py

@@ -0,0 +1,23 @@
+from transformers import pipeline as Pipe
+
+class Traductor:
+    def EN_ES(texto:str)->str:
+        _traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-en-es")
+        _traduccion = _traductor(texto)[0]
+        return _traduccion.get('generated_text')
+    
+    def ES_EN(texto:str)->str:
+        _traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-es-en")
+        _traduccion = _traductor(texto)[0]
+        return _traduccion.get('generated_text')
+    
+    def AR_ES(texto:str)->str:
+        _traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-ar-es")
+        _traduccion = _traductor(texto)[0]
+        return _traduccion.get('generated_text')
+
+class Abstractor:
+    def resumen(texto:str)->str:
+        _abstractor = Pipe("text2text-generation", model="facebook/bart-large-cnn")
+        _resumen = _abstractor(texto)[0]
+        return _resumen.get('generated_text')

api/main.py

@@ -1,20 +1,9 @@
 from api.core.app import Demos
 __main = Demos()
 api = __main.api()
-@api.get("/", status_code=201)
-def home() -> dict:
-    __response=dict({"request_data":None})
-    try:
-        __response['message']= "Bienvenido"
-    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=e)
-    finally:
-        return __response
 
 @api.post("/texto_desde_web/", status_code=201)
-def __get_text_from_url(data:dict) -> dict:
+def get_text_from_url(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
         if data:
@@ -29,7 +18,7 @@ def __get_text_from_url(data:dict) -> dict:
         return __response
 
 @api.post("/texto_desde_pdf/", status_code=201)
-def __get_text_from_pdf(data:dict) -> dict:
+def get_text_from_pdf(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
         if data:
@@ -44,13 +33,13 @@ def __get_text_from_pdf(data:dict) -> dict:
         return __response
 
 @api.post("/generar_bloques/", status_code=201)
-def __get_blocks(data:dict) -> dict:
+def get_blocks(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
         if data:
             __response['original']=data.get('texto')
             __response['bloques']=__main.generar_bloques(texto=data.get('texto'),
-                                                         s=data.get('size'))
+                                                         size=data.get('size'))
         else:
             raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
     except Exception as e:
@@ -61,7 +50,7 @@ def __get_blocks(data:dict) -> dict:
         return __response
 
 @api.post("/traducir/", status_code=201)
-def __get_traduccion(data:dict) -> dict:
+def get_traduccion(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
         if data:
@@ -78,7 +67,7 @@ def __get_traduccion(data:dict) -> dict:
         return __response
 
 @api.post("/resumir/", status_code=201)
-def __get_resumen(data:dict) -> dict:
+def get_resumen(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
         if data:
@@ -94,7 +83,7 @@ def __get_resumen(data:dict) -> dict:
         return __response
     
 @api.post("/texto_a_imagen/", status_code=201)
-def __get_text2img(data:dict) -> dict:
+def get_text2img(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
         if data:
@@ -110,7 +99,7 @@ def __get_text2img(data:dict) -> dict:
         return __response
 
 @api.post("/texto_a_video/", status_code=201)
-def __get_text2video(data:dict) -> dict:
+def get_text2video(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
         if data:
@@ -126,7 +115,7 @@ def __get_text2video(data:dict) -> dict:
         return __response
 
 @api.post("/texto_a_audio/", status_code=201)
-def __get_text2speach(data:dict) -> dict:
+def get_text2speach(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
         if data:

requirements.txt

@@ -1,4 +1,7 @@
 fastapi
 pydantic
 uvicorn
-typing
+typing
+diffusers
+diffusers[torch]
+diffusers[flax]