Alwrity - WIP - main_config

lib/blog_postprocessing/blog_proof_reader.py

@@ -1,19 +1,13 @@
 import os
 import sys
-
-from pathlib import Path
-from dotenv import load_dotenv
-load_dotenv(Path('../../.env'))
 import configparser
 
-from ..gpt_providers.gemini_pro_text import gemini_text_response
-from ..gpt_providers.openai_text_gen import openai_chatgpt
+from ..gpt_providers.text_generation.main_text_generation import llm_text_gen
 
 
 def blog_proof_editor(blog_content):
     """ Helper for blog proof reading. """
 
-    gpt_provider = os.environ["GPT_PROVIDER"]
     try:
         config_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', 'main_config'))
         config = configparser.ConfigParser()
@@ -38,15 +32,8 @@ def blog_proof_editor(blog_content):
 
         \n\nMy Blog: '{blog_content}'. """
 
-    if 'openai' in gpt_provider.lower():
-        try:
-            response = openai_chatgpt(prompt)
-            return response
-        except Exception as err:
-            SystemError(f"Openai Error Blog Proof Reading: {err}")
-    elif 'google' in gpt_provider.lower():
-        try:
-            response = gemini_text_response(prompt)
-            return response
-        except Exception as err:
-            SystemError(f"Gemini Error Blog Proof Reading: {err}")
+    try:
+        response = llm_text_gen(prompt)
+        return response
+    except Exception as err:
+        logger.error(f"Error Blog Proof Reading: {err}")

lib/blog_postprocessing/humanize_blog.py

@@ -1,17 +1,18 @@
 import os
 import sys
 
-from pathlib import Path
-from dotenv import load_dotenv
-load_dotenv(Path('../../.env'))
+from loguru import logger
+logger.remove()
+logger.add(sys.stdout,
+        colorize=True,
+        format="<level>{level}</level>|<green>{file}:{line}:{function}</green>| {message}"
+    )
 
-from ..gpt_providers.gemini_pro_text import gemini_text_response
-from ..gpt_providers.openai_text_gen import openai_chatgpt
+from ..gpt_providers.text_generation.main_text_generation import llm_text_gen
 
 
 def blog_humanize(blog_content):
     """ Helper for blog proof reading. """
-    gpt_provider = os.environ["GPT_PROVIDER"]
 
     prompt = f"""The following is what I will refer to as an 'Exception-list'. 
     Do Not include any of the words or phrases on this list in your future responses to this chat thread. 
@@ -25,15 +26,9 @@ def blog_humanize(blog_content):
         
         \n\nBlog Content: '{blog_content}'
         """
-    if 'openai' in gpt_provider.lower():
-        try:
-            response = openai_chatgpt(prompt)
-            return response
-        except Exception as err:
-            SystemError(f"Openai Error Blog Proof Reading: {err}")
-    elif 'google' in gpt_provider.lower():
-        try:
-            response = gemini_text_response(prompt)
-            return response
-        except Exception as err:
-            SystemError(f"Gemini Error Blog Proof Reading: {err}")
+    try:
+        response = llm_text_gen(prompt)
+        return response
+    except Exception as err:
+        logger.error(f"Openai Error Blog Proof Reading: {err}")
+        raise err