add text for ocr

Author: phatvo9

clarifai/runners/utils/data_types/__init__.py

@@ -3,11 +3,9 @@
 from .data_types import Concept  # noqa
 from .data_types import Frame  # noqa
 from .data_types import Image  # noqa
-from .data_types import Mask  # noqa
 from .data_types import MessageData  # noqa
 from .data_types import NamedFields  # noqa
 from .data_types import NamedFieldsMeta  # noqa
-from .data_types import Point  # noqa
 from .data_types import Region  # noqa
 from .data_types import Text  # noqa
 from .data_types import Video  # noqa

clarifai/runners/utils/data_types/data_types.py

@@ -284,15 +284,14 @@ def from_proto(cls, proto: PointProto) -> "Point":
 
 class Region(MessageData):
 
-  def __init__(
-      self,
-      proto_region: RegionProto = None,
-      box: List[float] = None,
-      concepts: List[Concept] = None,
-      mask: Union[Mask, 'Image', PILImage.Image, np.ndarray] = None,
-      point: Union[Point, Tuple[float, float, float]] = None,
-      track_id: str = None,
-  ):
+  def __init__(self,
+               proto_region: RegionProto = None,
+               box: List[float] = None,
+               concepts: List[Concept] = None,
+               mask: Union[Mask, 'Image', PILImage.Image, np.ndarray] = None,
+               point: Union[Point, Tuple[float, float, float]] = None,
+               track_id: str = None,
+               text: str = None):
     if proto_region is None:
       proto_region = RegionProto()
     self.proto = proto_region
@@ -306,7 +305,17 @@ def __init__(
     if point:
       self.point = point
     if track_id:
-      self.track_id
+      self.track_id = track_id
+    if text:
+      self.text = text
+
+  @property
+  def text(self):
+    return self.proto.data.text.raw
+
+  @text.setter
+  def text(self, value: str):
+    self.proto.data.text.raw = value
 
   @property
   def box(self) -> List[float]: