Function main

news_oversimplifier.py:13–70

()

def main():
    # loads .env variables
    load_dotenv()
    API_KEY = os.getenv("NEWS_API_KEY")

    # check validity of command-line arguments
    try:
        if len(sys.argv) == 2:
            news_query = sys.argv[1]
        else:
            raise IndexError()
    except IndexError:
        sys.exit("Please provide correct number of command-line arguments")

    try:
        # get number of articles from user
        while True:
            try:
                num_articles = int(input("Enter number of articles: "))
                break
            except ValueError:
                continue

        # fetch news articles based on user's query
        articles = fetch_news(API_KEY, query=news_query, max_articles=num_articles)

        # output printing title, summary and no. of words in the summary
        for i, article in enumerate(articles):
            capitalized_title = capitalize_title(article["title"])
            print(f"\n{i + 1}. {capitalized_title}")

            content = article.get("content") or article.get("description") or ""
            if not content.strip():
                print("No content to oversimplify.")
                continue

            summary = summarize_text(content)  # returns summary
            count = word_count(summary)  # returns word count
            print(f"\nOVERSIMPLIFIED:\n{summary}\n{count} words\n")

            # ask user whether they want to save the output in a txt file
            while True:
                saving_status = (
                    input("Would you like to save this in a text file? (y/n): ")
                    .strip()
                    .lower()
                )
                if saving_status == "y":
                    save_summary(article["title"], summary)
                    break
                elif saving_status == "n":
                    break
                else:
                    print("Try again\n")
                    continue

    except Exception as e:
        print("ERROR:", e)