AI news report writer, web researched

lib/ai_writers/ai_news_article_writer.py

@@ -0,0 +1,101 @@
+######################################################
+#
+# Alwrity, as an AI news writer, will have to be factually correct. 
+# We will do multiple rounds of web research and cite our sources. 
+# 'include_urls' will focus news articles only from well known sources. 
+# Choosing a country will help us get better results.
+#
+######################################################
+
+import sys
+import os
+import json
+from textwrap import dedent
+from pathlib import Path
+from datetime import datetime
+
+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.text_generation.main_text_generation import llm_text_gen
+from ..ai_web_researcher.google_serp_search import perform_serper_news_search
+
+
+def ai_news_generation(news_keywords, news_country, news_language):
+    """ Generate news aritcle based on given keywords. """
+    # Use to store the blog in a string, to save in a *.md file.
+    blog_markdown_str = ""
+
+    logger.info(f"Researching and Writing News Article on keywords: {news_keywords}")
+    # Call on the got-researcher, tavily apis for this. Do google search for organic competition.
+    try:
+        google_news_result = perform_serper_news_search(news_keywords, news_country, news_language)
+        blog_markdown_str = write_news_google_search(news_keywords, news_country, news_language, google_news_result)
+        #print(blog_markdown_str)
+    except Exception as err:
+        logger.error(f"Failed in Google News web research: {err}")
+    logger.info("\n######### Draft1: Finished News article from Google web search: ###########\n\n")
+
+
+def write_news_google_search(news_keywords, news_country, news_language, search_results):
+    """Combine the given online research and gpt blog content"""
+    news_language = get_language_name(news_language)
+    news_country = get_country_name(news_country)
+    
+    prompt = f"""
+        As an experienced {news_language} news journalist and editor, 
+        I will provide you with my 'News keywords' and its 'google search results'.
+        Your goal is to write a News report, backed by given google search results.
+        Important, as a news report, its imperative that your content is factually correct and cited.
+        
+        Follow below guidelines:
+        1). Understand and utilize the provided google search result json.
+        2). Always provide in-line citations and provide referance links. 
+        3). Understand the given news item and adapt your tone accordingly.
+        4). Always include the dates when then news was reported.
+        6). Do not explain, describe your response.
+        7). Your blog should be highly formatted in markdown style and highly readable.
+        8). Important: Please read the entire prompt before writing anything. Follow the prompt exactly as I instructed.
+
+        \n\nNews Keywords: "{news_keywords}"\n\n
+        Google search Result: "{search_results}"
+        """
+    logger.info("Generating blog and FAQs from Google web search results.")
+    try:
+        response = llm_text_gen(prompt)
+        return response
+    except Exception as err:
+        logger.error(f"Exit: Failed to get response from LLM: {err}")
+        exit(1)
+
+
+def get_language_name(language_code):
+    languages = {
+        "es": "Spanish",
+        "vn": "Vietnamese",
+        "en": "English",
+        "ar": "Arabic",
+        "hi": "Hindi",
+        "de": "German",
+        "zh-cn": "Chinese (Simplified)"
+        # Add more language codes and corresponding names as needed
+    }
+    return languages.get(language_code, "Unknown")
+
+def get_country_name(country_code):
+    countries = {
+            "es": "Spain",
+            "vn": "Vietnam",
+            "pk": "Pakistan",
+            "in": "India",
+            "de": "Germany",
+            "cn": "China"
+        # Add more country codes and corresponding names as needed
+        }
+    return countries.get(country_code, "Unknown")