Alwrity - WIP - main_config

lib/blog_metadata/get_blog_title.py

@@ -1,13 +1,6 @@
 import os
 import sys
 
-from pathlib import Path
-from dotenv import load_dotenv
-load_dotenv(Path('../../.env'))
-
-from ..gpt_providers.openai_text_gen import openai_chatgpt
-from ..gpt_providers.gemini_pro_text import gemini_text_response
-
 from loguru import logger
 logger.remove()
 logger.add(sys.stdout,
@@ -15,13 +8,14 @@ logger.add(sys.stdout,
         format="<level>{level}</level>|<green>{file}:{line}:{function}</green>| {message}"
     )
 
+from ..gpt_providers.text_generation.main_text_generation import llm_text_gen
+
 
 def generate_blog_title(blog_article, keywords=None, example_titles=None, num_titles=1):
     """
     Given a blog title generate an outline for it
     """
     prompt = ''
-    gpt_providers = os.environ["GPT_PROVIDER"]
     logger.info("Generating blog title.")
     if not keywords and not example_titles:
         prompt = f"""As a SEO expert, I will provide you with a blog content. 
@@ -51,16 +45,9 @@ def generate_blog_title(blog_article, keywords=None, example_titles=None, num_ti
             Negative Keywords: Unvieling, unleash, power of. Dont use such words in your title.
             Blog Article: '{keywords}'
         """
-    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:
-            logger.info("Calling OpenAI LLM.")
-            response = openai_chatgpt(prompt)
-            return response
-        except Exception as err:
-            SystemError(f"Failed to get response from Openai: {err}") 
+    try:
+        response = llm_text_gen(prompt)
+        return response
+    except Exception as err:
+        logger.error(f"Failed to get response from LLM: {err}")
+        raise err