Improved, refactored, added options, prompts, jekyll website

pseo_main.py

@@ -9,10 +9,12 @@
 #
 #########################################################
 import sys
+import os
 
 import argparse
 import json
 import traceback
+import requests
 from loguru import logger
 logger.remove()
 logger.add(sys.stdout, 
@@ -33,11 +35,12 @@ def main():
     """
 
     parser = argparse.ArgumentParser(
-        description="Accepts user input for the number of blogs, keywords, and niche."
+        description="Accepts user input for the number of blogs, subtopics, keywords, and niche."
     )
-    parser.add_argument("--num_blogs", type=int, default=1, help="The number of blogs (default: 5).")
+    parser.add_argument("--num_blogs", type=int, default=5, help="The number of blogs (default: 5).")
     parser.add_argument("--keywords", type=str, required=True, help="The keywords.A broad idea to write multiple blogs on.")
     parser.add_argument("--niche", type=bool, default=False, help="Written blogs on long tailed search topics (default: False).")
+    parser.add_argument("--num_subtopics", type=int, default=6, help="The number of sub topics to write (default: 6).")
 
     args = parser.parse_args()
 
@@ -54,18 +57,52 @@ def main():
     logger.info(f"Keywords: {args.keywords}")
     logger.info(f"Niche blog: {args.niche}")
 
-    return args.num_blogs, args.keywords, args.niche
+    return args
+
+def check_openai_api_key(openai_api_key):
+	"""Checks if the given OpenAI API key is valid and works.
+	
+	Args:
+	    openai_api_key: The OpenAI API key to check.
+	
+	Returns:
+	    True if the OpenAI API key is valid and works, False otherwise.
+	"""
+	
+	headers = {
+	    "Authorization": f"Bearer {openai_api_key}"
+	    }
+	
+	# Make a test request to the OpenAI API.
+	response = requests.get(
+	    "https://api.openai.com/v1/engines",
+	    headers=headers
+	    )
+	
+	# If the request was successful, the API key is valid and works.
+	return response.status_code == 200
 
 
 if __name__ == "__main__":
     # Check if we have everything, we need to start writing blogs.
-    try:
-        num_blogs, keywords, niche = main()
-        logger.info(f"returned value: {num_blogs} {keywords}")
-    except TypeError as e:
-        logger.error(e)
-    except ValueError as e:
-        logger.error(e)
-    else:
-        logger.info(f"Starting to write {num_blogs} blogs on {keywords}")
-        generate_detailed_blog(num_blogs, keywords, niche)
+     """Checks for the OPENAI_API_KEY environment variable, if it is not exported or if it is not valid."""
+     openai_api_key = os.environ.get("OPENAI_API_KEY")
+     if not openai_api_key:
+         logger.error("Error: Please    export OPENAI_API_KEY=''  - before running this script.")
+         exit(1)
+     # Check if the OpenAI API key is valid.
+     if not check_openai_api_key(openai_api_key):
+         logger.error("The OPENAI_API_KEY not valid. Check your API key and make sure its correct.")
+         exit(1)
+     # The OpenAI API key is valid and works.
+     logger.info("The OPENAI_API_KEY environment variable is valid and works.")
+
+     try:
+         args = main()
+     except TypeError as e:
+         logger.error(e)
+     except ValueError as e:
+         logger.error(e)
+     else:
+         logger.info(f"Writing {args.num_blogs} blogs on '{args.keywords}' with {args.num_subtopics} subtopics.")
+         generate_detailed_blog(args.num_blogs, args.keywords, args.niche, args.num_subtopics)