def main():
"""main is the entrypoint to the markov meme client.
"""
parser = get_parser()
# Will exit with subcommand help if doesn't parse
args, extra = parser.parse_known_args()
# Initialize the JuliaSet
if args.command == "generate":
# If the provided font doesn't end in ttf
font = args.font
if not args.font.endswith(".ttf"):
font = "%s.ttf" % (font)
# Determine if we have a corpus or custom corpus
corpus = args.corpus
if corpus is None:
corpus = random.choice(list_corpus())
if args.custom_corpus:
if os.path.exists(args.custom_corpus):
corpus = args.custom_corpus
text = generate_text(corpus=corpus, use_model=not args.no_model, size=10)
meme = MemeImage(image=args.custom_image, corpus=corpus)
# Add text generated, will go to top and bottom
meme.write_text(text, fontsize=args.fontsize, font=font)
meme.save_image(args.outfile)
else:
parser.print_help()
def get_corpus(prefix):
"""load a corpus file from "corpus" in the same directory as this script.
we assume a .txt extension, and return the full path to the file.
"""
selection = list_corpus(remove_ext=False)
selected = "%s.txt" % (prefix)
# Return filename if exists
corpus_folder = os.path.join(here, "data", "corpus")
filename = os.path.join(corpus_folder, selected)
if not os.path.exists(filename):
sys.exit("Missing corpus file %s" % filename)
return filename
def get_parser():
parser = argparse.ArgumentParser(description="Markov Meme Generator")
description = "actions for Markov Meme Generator"
subparsers = parser.add_subparsers(
help="markovmeme actions",
title="actions",
description=description,
dest="command",
)
# Many of these are the same as for Juliart Sets
generate = subparsers.add_parser("generate", help="generate a meme")
generate.add_argument(
"--outfile",
dest="outfile",
help="the output file to save the image (defaults to randomly generated png)",
type=str,
default=None,
)
generate.add_argument(
"--fontsize",
dest="fontsize",
help="font size of text (if desired) defaults to 16",
type=int,
default=36,
)
generate.add_argument(
"--font",
dest="font",
help="choice of font (defaults to open sans)",
type=str,
choices=["OpenSans-Regular", "Pacifico-Regular", "Anton-Regular"],
default="Anton-Regular",
)
generate.add_argument(
"--corpus",
dest="corpus",
choices=list_corpus(),
help="the corpus to use to generate the meme, matches to images.",
type=str,
default=None,
)
generate.add_argument(
"--custom-corpus",
dest="custom_corpus",
help="A custom corpus file, full path",
type=str,
default=None,
)
generate.add_argument(
"--image",
dest="custom_image",
help="A custom image file, full path",
type=str,
default=None,
)
generate.add_argument(
"--no-model",
dest="no_model",
help="Don't generate a sentence from corpus, just randomly select sentence.",
default=False,
action="store_true",
)
return parser