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}")

lib/blog_metadata/get_blog_meta_desc.py

@@ -1,8 +1,5 @@
 import sys
 import os
-from pathlib import Path
-from dotenv import load_dotenv
-load_dotenv(Path('../.env'))
 
 from loguru import logger
 logger.remove()
@@ -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 generate_blog_description(blog_content):
     """
         Prompt designed to give SEO optimized blog descripton
     """
-    gpt_providers = os.environ["GPT_PROVIDER"]
     logger.info("Generating Blog Meta Description for the given blog.")
     prompt = f"""As an expert SEO and blog writer, Compose a compelling meta description for the given blog content, 
         adhering to SEO best practices. Keep it between 150-160 characters. 
@@ -27,15 +22,9 @@ def generate_blog_description(blog_content):
         Respond with only one of your best effort and do not include your explanations. 
         Blog Content: '{blog_content}'"""
 
-    if 'google' in gpt_providers.lower():
-        try:
-            response = gemini_text_response(prompt)
-            return response
-        except Exception as err:
-            logger.error("Failed to get response from gemini.")
-    elif 'openai' in gpt_providers.lower():
-        try:
-            response = openai_chatgpt(prompt)
-            return response
-        except Exception as err:
-            SystemError(f"Error in generating blog summary: {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

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

lib/blog_metadata/get_tags.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,8 +8,7 @@ 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_tags(blog_article):
@@ -25,15 +21,9 @@ def get_blog_tags(blog_article):
          for the given blog content. Only reply with comma separated values. 
          Blog content:  {blog_article}."""
     logger.info("Generating Blog tags for the given blog post.")
-    if 'google' in gpt_providers.lower():
-        try:
-            response = gemini_text_response(prompt)
-            return response
-        except Exception as err:
-            logger.error("Failed to get response from gemini.")
-    elif 'openai' in gpt_providers.lower():
-        try:
-            response = openai_chatgpt(prompt)
-            return response
-        except Exception as err:
-            SystemError(f"Error in generating blog summary: {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