Spaces:
Build error
Build error
| import gradio as gr | |
| from wordcloud import WordCloud | |
| import matplotlib.pyplot as plt | |
| import openai | |
| import matplotlib as mpl | |
| import matplotlib.font_manager as fm | |
| import jieba | |
| import jieba.analyse | |
| # 載入停用詞 | |
| def load_stopwords(file_path): | |
| stopwords = set() | |
| with open(file_path, 'r', encoding='utf-8') as f: | |
| for line in f: | |
| stopwords.add(line.strip()) | |
| return stopwords | |
| # 加載停用詞 | |
| stopwords = load_stopwords("stopwords.txt") | |
| # 使用 OpenAI API 進行情緒分析 | |
| def analyze_text(api_key, text): | |
| try: | |
| openai.api_key = api_key | |
| response = openai.chat.completions.create( | |
| model="gpt-3.5-turbo", | |
| messages=[ | |
| {"role": "user", "content": f"範例輸出：正面情緒。請分析此文本情緒，輸出其主要為正面、負面或中性情緒: {text}"} | |
| ] | |
| ) | |
| return response.choices[0].message.content, None # 返回分析結果 | |
| except Exception as e: | |
| error_message = f"錯誤發生: {str(e)}" | |
| print(error_message) | |
| return None, error_message # 返回錯誤訊息 | |
| # 文本處理函數 | |
| def process_text(api_key, text, colormap): | |
| # 使用 jieba 進行中文分詞並過濾停用詞 | |
| words = [word for word in jieba.cut(text) if word not in stopwords] | |
| words_filtered = " ".join(words) | |
| # 生成文字雲 | |
| wordcloud = WordCloud( | |
| font_path='TaipeiSansTCBeta-Regular.ttf', # 繁體中文字型 | |
| width=400, | |
| height=200, | |
| colormap=colormap, # 套用選擇的顏色地圖 | |
| background_color='white', | |
| min_font_size=10, | |
| max_words=200 | |
| ).generate(words_filtered) | |
| # 保存圖片 | |
| fig, ax = plt.subplots(figsize=(8, 4)) | |
| ax.imshow(wordcloud, interpolation='bilinear') | |
| ax.axis('off') | |
| plt.savefig("wordcloud.png", bbox_inches='tight', pad_inches=0) | |
| plt.close(fig) | |
| # 呼叫情緒分析函數 | |
| analysis_result, error = analyze_text(api_key, text) | |
| return "wordcloud.png", analysis_result | |
| # 加載字型 | |
| fm.fontManager.addfont('TaipeiSansTCBeta-Regular.ttf') | |
| mpl.rc('font', family='Taipei Sans TC Beta') | |
| # jieba 設定 | |
| jieba.set_dictionary("dict.txt") | |
| jieba.analyse.set_stop_words("stopwords.txt") | |
| # 定義僅限圖片中提供的顏色地圖 | |
| color_maps = [ | |
| "Greys", "Purples", "Blues", "Greens", "Oranges", "Reds", | |
| "YlOrBr", "YlOrRd", "OrRd", "PuRd", "RdPu", "BuPu", "GnBu", | |
| "PuBu", "YlGnBu", "PuBuGn", "BuGn", "YlGn" | |
| ] | |
| # 創建 Gradio 介面 | |
| iface = gr.Interface( | |
| fn=process_text, | |
| inputs=[ | |
| gr.Textbox(label="請輸入你的 API 金鑰", placeholder="在此輸入 API 金鑰...", type="password"), | |
| gr.Textbox(label="請輸入文本", placeholder="在此輸入文本...", lines=5), | |
| gr.Dropdown(label="選擇文字雲風格", choices=color_maps, value="Blues") | |
| ], | |
| outputs=[gr.Image(type="filepath"), "text"], | |
| title="文字雲與情緒分析工具", | |
| description="輸入中文或英文文本，生成文字雲並分析其情緒（正面、負面或中性）。" | |
| ) | |
| # 啟動介面 | |
| iface.launch() | |