Update HebEMO.py

HebEMO.py

@@ -24,7 +24,9 @@ class HebEMO:
         input_path(str): the path to the text file (txt file, each row for different instance)
         returns pandas DataFrame of the analyzed texts and save it to the same dir of the input file
         '''
-        from pyplutchik import plutchik
+        try: from pyplutchik import plutchik
+        except: from spider_plot import spider_plot
+        
         import matplotlib.pyplot as plt
         import pandas as pd
         import time
@@ -75,7 +77,8 @@ class HebEMO:
                 hebEMO[emo] = abs(hebEMO_df[emo]-(1-hebEMO_df['confidence_'+emo]))
 
             for i in range(0,1):    
-                ax = plutchik(hebEMO.to_dict(orient='records')[i])
+                try: ax = plutchik(hebEMO.to_dict(orient='records')[i])
+                except: ax = spider_plot(hebEMO)
                 print(hebEMO_df[0][i])
                 plt.show()
             return (plt.figure())