68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
#!/usr/bin/python3
|
|
|
|
#########################################################
|
|
#
|
|
# This is the main module for calling pseo related functions.
|
|
# This is the end user interface and is user driven.
|
|
# TBD: argsparser and taking config file. For usuability,
|
|
# no editing of code should be required.
|
|
#
|
|
#########################################################
|
|
import sys
|
|
|
|
import argparse
|
|
import json
|
|
import traceback
|
|
from loguru import logger
|
|
logger.add(sys.stdout, colorize=True, format="<green>{time}</green> <level>{message}</level>")
|
|
|
|
from lib.generate_image_from_prompt import generate_image, gen_new_from_given_img
|
|
from lib.get_text_response import get_prompt_reply, generate_detailed_blog
|
|
|
|
|
|
def main():
|
|
"""Parses user arguments and prints them to the console.
|
|
|
|
Raises:
|
|
TypeError: If the user input is not valid.
|
|
ValueError: If the number of blogs is less than 1.
|
|
"""
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description="Accepts user input for the number of blogs, keywords, and niche."
|
|
)
|
|
parser.add_argument("--num_blogs", type=int, default=1, help="The number of blogs (default: 1).")
|
|
parser.add_argument("--keywords", type=str, required=True, help="The keywords.")
|
|
parser.add_argument("--niche", type=bool, default=False, help="Whether the blog is a niche blog (default: False).")
|
|
|
|
args = parser.parse_args()
|
|
|
|
# Check if the user input is valid
|
|
if not isinstance(args.num_blogs, int) or not isinstance(args.keywords, str) or not isinstance(args.niche, bool):
|
|
raise TypeError("Invalid user input.")
|
|
|
|
# Check if the number of blogs is less than 1
|
|
if args.num_blogs < 1:
|
|
raise ValueError("The number of blogs must be at least 1.")
|
|
|
|
# Print the user input to the console
|
|
print(f"Number of blogs: {args.num_blogs}")
|
|
print(f"Keywords: {args.keywords}")
|
|
print(f"Niche blog: {args.niche}")
|
|
|
|
return args.num_blogs, args.keywords, args.niche
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Check if we have everything, we need to start writing blogs.
|
|
try:
|
|
num_blogs, keywords, niche = main()
|
|
print(f"returned value: {num_blogs} {keywords}")
|
|
except TypeError as e:
|
|
print(e)
|
|
except ValueError as e:
|
|
print(e)
|
|
else:
|
|
print(f"Starting to write {num_blogs} on {keywords}")
|
|
generate_detailed_blog(num_blogs, keywords, niche)
|