reflexive 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

reflexive/visual/display.py

@@ -0,0 +1,110 @@
+
+#from spacy import displacy
+
+#from reflexive.common.parameters import Parameters
+
+import logging
+try:
+    import coloredlogs
+    coloredlogs.install(level='INFO')
+except:
+    print("Colored logs not available")
+
+class Display:
+    
+    logger = logging.getLogger(__name__)
+    
+    def __init__(self): #,parameters:Parameters):
+        self.name="Display"
+        self.priority_tags = ["AR","EP","VR_EV_CN","ER_AF","RR","KP"]
+        self.colours = {"VR_EV_CN": "#ff6644","ER_AF": "#dd44cc","AR": "#00cc00","EP": "#aacc33","RR": "#00aaff","KP":"#aaaacc"}
+        self.options = {"ents": ["VR_EV_CN","ER_AF","AR","EP","RR","KP"], "colors": self.colours}
+        
+
+
+    def add_reflexive_offsets(self,df):
+        temp_df = df.copy()
+        temp_df['reflexive_offsets'] = temp_df.reflexiveResults.apply(self.collect_reflexive_offsets)
+        return temp_df
+        
+    def add_keyphrase_offsets(self,df):
+        temp_df = df.copy()
+        temp_df['keyphrase_offsets'] = temp_df.KeyPhraseResults.apply(self.collect_keyphrase_offsets)
+        return temp_df
+    
+    def add_offsets(self,df):
+        df = self.add_reflexive_offsets(df)
+        return self.add_keyphrase_offsets(df)
+    
+    def create_displacy(self,df):
+        all_ents = list(df.apply(self.render_record,axis=1))
+        #html_out = displacy.render(all_ents,manual=True,style="ent", options=options,page=True,jupyter=False)
+        # with open(f"{path}{prefix}annotated_reflections{postfix}.html","w") as fp:
+        #     fp.write(html_out)
+        #displacy.render(all_ents,manual=True,style="ent", options=options)
+        return all_ents
+            
+    def render_record(self,record,title="----"):
+        #timestamp = record['timestamp'].split('T')[0]
+        #pseudonym = record['pseudonym']
+        #point_round = record['point_round']
+        #title = f"{pseudonym} ({point_round}) - {timestamp}"
+        tags = self.priority_tags
+        text = record['text']
+        reflexive_offsets = record['reflexive_offsets']
+        keyphrase_offsets = record['keyphrase_offsets']
+        ents = []
+        taken = []
+        offsets = []
+        for tag in tags:
+            if tag in reflexive_offsets:
+                offsets = reflexive_offsets[tag]
+            elif tag in keyphrase_offsets:
+                offsets = keyphrase_offsets[tag]
+            
+            for off in offsets:
+                new_ent = {}
+                if off[0] in taken:
+                    # the start offset is taken
+                    x = None
+                elif off[1] in taken:
+                    # the end offset is taken
+                    x = None
+                else:
+                    # both start and end is available
+                    taken.append(off[0])
+                    taken.append(off[1])
+                    #print(taken)
+                    new_ent["start"] = off[0]
+                    new_ent["end"] = off[1]
+                    new_ent["label"] = tag
+                    ents.append(new_ent)
+
+        text_ents = {
+            "text": text, #.replace('\r\n','\n'),
+            "ents": ents,
+            "title": title
+        }
+
+        return text_ents
+    
+    def collect_keyphrase_offsets(self,krs):
+        new_krs = {}
+        for kr in krs:
+            if kr['Score']>0.98:
+                new_krs.setdefault("KP",[]).append((kr['BeginOffset'],kr['EndOffset']))
+        return new_krs
+
+    def collect_reflexive_offsets(self,rrs):
+        new_rrs = {}
+        for rr in rrs:
+            if rr['Score']>0.5:
+                ent_type = rr['Type']
+                if ent_type in ['VR','EV','CN']:
+                    label = "VR_EV_CN"
+                elif ent_type in ['ER','AF']:
+                    label = "ER_AF"
+                else:
+                    label = ent_type
+                new_rrs.setdefault(label,[]).append((rr['BeginOffset'],rr['EndOffset']))
+        return new_rrs

{reflexive-0.1.0.dist-info → reflexive-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: reflexive
-Version: 0.1.0
+Version: 0.1.1
 Summary: Supports AWS Reflexive Expressions Analysis
 Home-page: https://github.com/nlytx/reflexive
 Author: Andrew Gibson

{reflexive-0.1.0.dist-info → reflexive-0.1.1.dist-info}/RECORD

@@ -10,8 +10,10 @@ reflexive/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 reflexive/common/local.py,sha256=euhMmdRnL5c5d_Co2f2ehNP7Lv43T571tDCWQzpKaF4,1752
 reflexive/common/parameters.py,sha256=p_0YdP8d0q1l76dOK3Xaj24GKPUVtzMY1buvzgT-IPU,3044
 reflexive/common/util.py,sha256=_aft-pgX3oDPgvK1HaazdH9kIt52cVD-tjm6M24hq6M,3175
-reflexive-0.1.0.dist-info/LICENSE.txt,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-reflexive-0.1.0.dist-info/METADATA,sha256=hn7GFc8BxusB3rSAR6-cO4WplMxVk_CiRImLRY-Xayo,597
-reflexive-0.1.0.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
-reflexive-0.1.0.dist-info/top_level.txt,sha256=pOMr-QGleRBRCFBozgvM-UUUmOjD_-naJfu1522E2V8,10
-reflexive-0.1.0.dist-info/RECORD,,
+reflexive/visual/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+reflexive/visual/display.py,sha256=XiRC3Asx9Ed3xCNCt3_DlbF45JcifNbXH_B44jxqqp0,3886
+reflexive-0.1.1.dist-info/LICENSE.txt,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+reflexive-0.1.1.dist-info/METADATA,sha256=_Q2YGxhybMZuK1qDVeAhketHbXha-m5pYXDBy1JTv7c,597
+reflexive-0.1.1.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
+reflexive-0.1.1.dist-info/top_level.txt,sha256=pOMr-QGleRBRCFBozgvM-UUUmOjD_-naJfu1522E2V8,10
+reflexive-0.1.1.dist-info/RECORD,,