# %%
import printj
from story_gen import StoryGenerator

gen = StoryGenerator()
# # %%
# story_till_now, emotion = gen.story(story_till_now='Hello, I\'m a language model,', num_generation=3, length=10)
# printj.purple(story_till_now)
# printj.yellow(emotion)


# %%
gen.get_stats(story_till_now="For myriad of eons i’ve forgotten who I really was, harvesting the essence of all existence.",
             length=10, num_generation=3, num_tests=50)

# %%
gen.save_stats('/home/jitesh/haru/ist/results/a.xlsx')




# %%
data=gen.stats_df[gen.stats_df.sentence_no==3]
import seaborn as sns
sns.set_theme(style="whitegrid")
# ax = sns.violinplot(x="day", y="total_bill", data=tips)
ax = sns.violinplot(x="reaction_weight", y="num_reactions", data=data).set_title('Analysing ProbabilityEmote (Max reactions=3)')
# %%

gen.stats_df[gen.stats_df.sentence_no==3]
# %%
import re
len(re.findall(r'\w+', 'line ive '))
# %%