WIP- Try AI-Writer and Web research; working. Working on usuability aspects.

lib/blog_metadata/get_blog_category.py

@@ -1,8 +1,9 @@
 import sys
+import os
 
-from .gpt_providers.openai_gpt_provider import openai_chatgpt
-from .gpt_providers.gemini_pro_text import gemini_text_response
-
+from pathlib import Path
+from dotenv import load_dotenv
+load_dotenv(Path('../.env'))
 from loguru import logger
 logger.remove()
 logger.add(sys.stdout,
@@ -10,11 +11,15 @@ logger.add(sys.stdout,
         format="<level>{level}</level>|<green>{file}:{line}:{function}</green>| {message}"
     )
 
+from ..gpt_providers.openai_gpt_provider import openai_chatgpt
+from ..gpt_providers.gemini_pro_text import gemini_text_response
 
-def get_blog_categories(blog_article, gpt_providers):
+
+def get_blog_categories(blog_article):
     """
     Function to generate blog categories for given blog content.
     """
+    gpt_providers = os.environ["GPT_PROVIDER"]
     prompt = f"""As an expert SEO and content writer, I will provide you with blog content.
             Suggest only 2 blog categories which are most relevant to provided blog content,
             by identifying the main topic. Also consider the target audience and the
@@ -22,7 +27,7 @@ def get_blog_categories(blog_article, gpt_providers):
             The blog content is: '{blog_article}'"
     """
     logger.info("Generating blog categories for the given blog.")
-    if 'gemini' in gpt_providers:
+    if 'google' in gpt_providers:
         try:
             response = gemini_text_response(prompt)
             return response