Alwrity - WIP - main_config

lib/blog_metadata/get_blog_category.py

@@ -1,9 +1,6 @@
 import sys
 import os
 
-from pathlib import Path
-from dotenv import load_dotenv
-load_dotenv(Path('../.env'))
 from loguru import logger
 logger.remove()
 logger.add(sys.stdout,
@@ -11,15 +8,13 @@ logger.add(sys.stdout,
         format="<level>{level}</level>|<green>{file}:{line}:{function}</green>| {message}"
     )
 
-from ..gpt_providers.openai_text_gen import openai_chatgpt
-from ..gpt_providers.gemini_pro_text import gemini_text_response
+from ..gpt_providers.text_generation.main_text_generation import llm_text_gen
 
 
 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
@@ -27,15 +22,8 @@ def get_blog_categories(blog_article):
             The blog content is: '{blog_article}'"
     """
     logger.info("Generating blog categories for the given blog.")
-    if 'google' in gpt_providers.lower():
-        try:
-            response = gemini_text_response(prompt)
-            return response
-        except Exception as err:
-            logger.error(f"Failed to get response from gemini: {err}")
-    elif 'openai' in gpt_providers.lower():
-        try:
-            response = openai_chatgpt(prompt)
-            return response
-        except Exception as err:
-            SystemError(f"Error in generating blog get_blog_categories: {err}")
+    try:
+        response = llm_text_gen(prompt)
+        return response
+    except Exception as err:
+        logger.error(f"get_blog_categories:Failed to get response from LLM: {err}")