Googlenews_v1.py

#!/usr/bin/env python
# coding: utf-8

# https://github.com/Iceloof/GoogleNews

import pandas as pd
import jieba
from GoogleNews import GoogleNews
from collections import Counter
import matplotlib.pyplot as plt
from datetime import datetime
from PIL import Image
import numpy as np
from scipy.ndimage import gaussian_gradient_magnitude
from wordcloud import WordCloud, ImageColorGenerator


def main():
    # Google News初始化設定
    googlenews = GoogleNews()
    googlenews.setlang('zh-tw')
    googlenews.setperiod('1d')
    # print(dir(googlenews))

    # 使用者輸入關鍵字
    keyword = "貪污"
    # keyword = input("請輸入要搜尋的關鍵字: ")
    googlenews.search(keyword)

    # # 獲取搜尋結果
    result = googlenews.result()
    # print(type(result))

    # # 顯示新聞標題和連結
    for item in result:
        print(f"標題: {item['title']}")
        print(f"連結: {item['link']}")
        print('---')

    # 進行斷詞處理並計算詞頻
    all_titles = ' '.join([item['title'] for item in result])  # 串列生成式
    words = jieba.cut(all_titles)
    print(type(words))
    word_count = Counter(words)
    print(word_count)

    # 顯示斷詞統計表格
    df = pd.DataFrame(word_count.items(), columns=['word', 'count'])
    df_sorted = df.sort_values(by='count', ascending=False).reset_index(drop=True)
    print(df_sorted.head(10))  # 只顯示前面 N 個

    # 詢問使用者字體檔案路徑
    font_path_input = "./TaipeiSansTCBeta-Bold.ttf"
    # font_path_input = input("請輸入中文字體的檔案路徑: ")
    # 文字雲圖片遮罩檔案路徑
    img_path = "./Mask_Image/Square_mask.png"

    # 讀取遮罩圖片，轉換為 numpy 陣列
    mask_color = np.array(Image.open(img_path))

    # 每隔_個像素取一個像素
    mask_color = mask_color[::2, ::2]



    # 生成遮罩圖像副本並應用閾值
    mask_image = mask_color.copy()
    threshold = 1  # 根據圖片調整閾值
    mask_image[np.all(mask_image < threshold, axis=2)] = 255
    # 255 白色

    # 邊緣檢測
    # 使用串列生成式
    edges = np.mean([gaussian_gradient_magnitude(mask_color[:, :, i] / 255., 2) for i in range(3)], axis=0)
    mask_image[edges > .09] = 255

    # 查看 edges 的類型和屬性
    print("Type of edges:", type(edges))
    print("Data type of edges:", edges.dtype)
    print("Attributes and methods of edges:", dir(edges))
    print("Shape of edges:", edges.shape)

    # # 顏色生成
    image_colors = ImageColorGenerator(mask_image)
    image_colors.default_color = [0.9,0.9,0.9]

    # 查看 image_colors 的類型和屬性
    print("Type of image_colors:", type(image_colors))
    print("Attributes and methods of image_colors:", dir(image_colors))

    # 生成文字雲
    wordcloud = WordCloud(
    font_path=font_path_input,
    width=512,
    height=512,
    max_font_size=50,  
    max_words=4000,
    mask=mask_image,
    color_func=image_colors # 從遮罩圖片中提取顏色
    ).generate_from_frequencies(word_count)

    # 查看 wordcloud 的類型和屬性
    print("Type of wordcloud:", type(wordcloud))
    print("Attributes and methods of wordcloud:", dir(wordcloud))

    plt.imshow(wordcloud)
    plt.axis("off")
    plt.show()

    # # 儲存文字雲圖片
    today = datetime.now().strftime('%Y%m%d_%H%M') #日期格式化
    # string from time 時間轉字串
    image_filename = f"{today}_{keyword}.png"
    wordcloud.to_file("img/"+image_filename)
    print(f"文字雲圖片已儲存為: {image_filename}")


# 魔法函式，被引用時不執行，只有直接執行時才會執行，未來要引用時，可以參考這個寫法
# http://blog.castman.net/教學/2018/01/27/python-name-main.html
if __name__ == "__main__":
    main()