Blogen-V0.1 Added features. WIP

lib/get_blog_meta_desc.py

@@ -1,17 +1,31 @@
-def generate_blog_description(blog_content):
+from .gpt_providers.openai_chat_completion import openai_chatgpt
+import google.generativeai as genai
+
+
+def generate_blog_description(blog_content, gpt_providers):
     """
         Prompt designed to give SEO optimized blog descripton
     """
     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, incorporating active verbs, 
-        avoiding all caps and excessive punctuation. Ensure relevance, engage users, and encourage clicks.
-        Use keywords naturally and provide a glimpse of the content's value to entice readers.
+        adhering to SEO best practices. Keep it between 150-160 characters. 
+        Provide a glimpse of the content's value to entice readers.
         Respond with only one of your best effort and do not include your explanations. 
         Blog Content: {blog_content}"""
-
-    try:
-        # TBD: Add logic for which_provider and which_model
-        response = openai_chatgpt(prompt)
-        return response
-    except Exception as err:
-        SystemError(f"Error in generating blog description: {err}")
+    if 'gemini' in gpt_providers:
+        try:
+            genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
+        except Exception as err:
+            logger.error("Failed in getting GEMINI_API_KEY")
+        # Use gemini-pro model for text and image.
+        model = genai.GenerativeModel('gemini-pro')
+        try:
+            response = model.generate_content(prompt)
+            return response.text
+        except Exception as err:
+            logger.error("Failed to get response from gemini.")
+    elif 'openai' in gpt_providers:
+        try:
+            response = openai_chatgpt(prompt)
+            return response
+        except Exception as err:
+            SystemError(f"Error in generating blog summary: {err}")