WIP- Try AI-Writer and Web research; working.

lib/ai_writers/gpt_blog_sections.py

@@ -0,0 +1,49 @@
+import sys
+import json
+
+from ..gpt_providers.openai_chat_completion import openai_chatgpt
+from ..gpt_providers.gemini_pro_text import gemini_text_response
+
+from loguru import logger
+logger.remove()
+logger.add(sys.stdout,
+        colorize=True,
+        format="<level>{level}</level>|<green>{file}:{line}:{function}</green>| {message}"
+    )
+
+
+# FIXME: Provide num_blogs, num_faqs as inputs.
+def get_blog_sections_from_websearch(search_keyword, search_results, gpt_providers="gemini"):
+    """Combine the given online research and gpt blog content"""
+
+    prompt = f"""
+        As a SEO expert and content writer, I will provide you with a search keyword and its google search result.
+        Your task is to write a blog title and 5 blog sub titles, from the given google search result.
+        The subtitles should be less than 40 characters and click worthy.
+        Do not explain, describe your response. Respond in json format, always name the key as 'blogSections'.
+
+        Web Research Keyword: "{search_keyword}"
+        Google search Result: "{search_results}"
+        """
+
+    if 'gemini' in gpt_providers:
+        try:
+            response = gemini_text_response(prompt)
+            if '```' in response and '\n' in response:
+                response = response.strip().split('\n')
+                # Remove the first and last lines
+                response = '\n'.join(response[1:-1])
+                response = json.loads(response)
+            return response
+        except Exception as err:
+            logger.error(f"Failed to get response from gemini: {err}")
+            logger.error(f"Gemini Error: {response.prompt_feedback}")
+            raise err
+    elif 'openai' in gpt_providers:
+        try:
+            logger.info("Calling OpenAI LLM.")
+            response = openai_chatgpt(prompt)
+            return response
+        except Exception as err:
+            logger.error(f"Failed to get response from Openai: {err}")
+            raise err