lauchmelder/RedditWordCloud
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Added more info prints

Browse source
This commit is contained in:
Robert 2020-07-14 11:46:07 +02:00
parent a490422587
commit 3ad1c000ab
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
.gitignore vendored
View file

@ -7,4 +7,6 @@ images
*.json *.json
*.log *.log
*.txt *.txt
rawtext.py *.pdf
rawtext.py
pdfcloud.py

4
cloud.py
View file

@ -148,6 +148,7 @@ wc_obj = WordCloud(font_path="ARIALUNI.TTF",
contour_color=cc contour_color=cc
) )
print("Processing words...", end=" ", flush=True)
words = wc_obj.process_text(' '.join(comments)) words = wc_obj.process_text(' '.join(comments))
with open(f"wordlists/{args.w}.csv") as list: with open(f"wordlists/{args.w}.csv") as list:
@ -173,8 +174,11 @@ for (word, freq) in words.items():
out_freq *= args.boost out_freq *= args.boost
out_dict[word] = out_freq out_dict[word] = out_freq
print("Done!")
print("Create cloud...", end=" ", flush=True)
wordcloud = wc_obj.generate_from_frequencies(out_dict) wordcloud = wc_obj.generate_from_frequencies(out_dict)
print("Done!")
if args.color is True: if args.color is True:
wordcloud.recolor(color_func=colors) wordcloud.recolor(color_func=colors)