""" 图片生成技能 - 完整版本（支持中文） 整合多种图片生成功能 """ import os import tempfile import numpy as np from PIL import Image, ImageDraw, ImageFont import matplotlib.pyplot as plt import seaborn as sns import qrcode from wordcloud import WordCloud # 导入字体工具 try: # 尝试不同的导入方式 import sys import os sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) from font_utils import get_chinese_font, setup_matplotlib_chinese CHINESE_SUPPORT = True except ImportError as e: CHINESE_SUPPORT = False print(f"⚠️ 字体工具导入失败，中文支持可能受限: {e}") # 设置matplotlib为非交互模式 plt.switch_backend('Agg') sns.set_style("whitegrid") # 尝试设置matplotlib中文支持 if CHINESE_SUPPORT: setup_matplotlib_chinese() class ImageGenerationSkill: """图片生成技能主类""" def __init__(self, config=None): self.config = config or {} # 尝试使用中文字体 if CHINESE_SUPPORT: self.default_font = get_chinese_font(14) else: self.default_font = ImageFont.load_default() print("⚠️ 使用默认字体，中文可能显示为方框") print("Image Generation Skill v2.0 (中文支持) 初始化完成") # ========== 基础功能 ========== def create_chart(self, data, chart_type='bar', title='', style='default'): """创建图表 Args: data: 数据字典 {'labels': [], 'values': []} chart_type: 图表类型 ('bar', 'line', 'pie', 'scatter') title: 图表标题 style: 样式 ('default', 'seaborn') Returns: 图片路径 """ try: plt.figure(figsize=(10, 6)) if style == 'seaborn': sns.set_palette("husl") labels = data.get('labels', []) values = data.get('values', []) if chart_type == 'bar': if style == 'seaborn': ax = sns.barplot(x=labels, y=values) # 添加数值标签 for i, v in enumerate(values): ax.text(i, v + 0.1, str(v), ha='center') else: bars = plt.bar(labels, values) for bar in bars: height = bar.get_height() plt.text(bar.get_x() + bar.get_width()/2., height + 0.1, f'{height}', ha='center', va='bottom') elif chart_type == 'line': if style == 'seaborn': sns.lineplot(x=labels, y=values, marker='o', linewidth=2) else: plt.plot(labels, values, marker='o', linewidth=2) elif chart_type == 'pie': plt.pie(values, labels=labels, autopct='%1.1f%%', startangle=90) plt.axis('equal') elif chart_type == 'scatter' and 'x' in data and 'y' in data: if style == 'seaborn': sns.scatterplot(x=data['x'], y=data['y']) else: plt.scatter(data['x'], data['y']) if title: plt.title(title, fontsize=14, fontweight='bold') plt.tight_layout() path = tempfile.mktemp(suffix='.png') plt.savefig(path, dpi=120, bbox_inches='tight') plt.close() return path except Exception as e: plt.close() raise Exception(f'图表生成失败: {e}') def create_qrcode(self, content, size=200, output_path=None): """生成二维码""" try: qr = qrcode.QRCode( version=1, error_correction=qrcode.constants.ERROR_CORRECT_L, box_size=10, border=4, ) qr.add_data(content) qr.make(fit=True) qr_img = qr.make_image(fill_color='black', back_color='white') qr_img = qr_img.resize((size, size)) if output_path is None: output_path = tempfile.mktemp(suffix='.png') qr_img.save(output_path) return output_path except Exception as e: raise Exception(f'二维码生成失败: {e}') def text_to_image(self, text, width=400, height=200, bg_color='white', text_color='black'): """文本转图片""" try: img = Image.new('RGB', (width, height), color=bg_color) draw = ImageDraw.Draw(img) lines = text.split('\n') line_height = 30 y_start = max(20, (height - len(lines) * line_height) // 2) for i, line in enumerate(lines): y = y_start + i * line_height # 简单居中 text_width = len(line) * 10 x = max(20, (width - text_width) // 2) draw.text((x, y), line, fill=text_color, font=self.default_font) path = tempfile.mktemp(suffix='.png') img.save(path) return path except Exception as e: raise Exception(f'文本转图片失败: {e}') # ========== 高级功能 ========== def create_wordcloud(self, text, width=800, height=400, background_color='white', colormap='viridis'): """创建词云图""" try: wordcloud = WordCloud( width=width, height=height, background_color=background_color, colormap=colormap, max_words=200, contour_width=1, contour_color='steelblue' ).generate(text) path = tempfile.mktemp(suffix='.png') wordcloud.to_file(path) return path except Exception as e: raise Exception(f'词云图生成失败: {e}') def create_color_palette(self, colors, title='颜色调色板'): """创建颜色调色板""" try: fig, axes = plt.subplots(1, len(colors), figsize=(len(colors)*2, 3)) if len(colors) == 1: axes = [axes] for ax, color in zip(axes, colors): ax.add_patch(plt.Rectangle((0, 0), 1, 1, color=color)) ax.text(0.5, -0.2, color, ha='center', transform=ax.transAxes) ax.set_xlim(0, 1) ax.set_ylim(0, 1) ax.axis('off') plt.suptitle(title, fontsize=14, fontweight='bold') plt.tight_layout() path = tempfile.mktemp(suffix='.png') plt.savefig(path, dpi=100, bbox_inches='tight') plt.close() return path except Exception as e: plt.close() raise Exception(f'颜色调色板生成失败: {e}') def create_data_card(self, title, data_dict, theme='light'): """创建数据卡片""" try: # 设置颜色主题 if theme == 'dark': bg_color = '#2c3e50' text_color = '#ecf0f1' accent_color = '#3498db' else: bg_color = '#ffffff' text_color = '#2c3e50' accent_color = '#3498db' # 计算卡片尺寸 num_items = len(data_dict) card_width = 500 card_height = 150 + num_items * 40 # 创建卡片 img = Image.new('RGB', (card_width, card_height), color=bg_color) draw = ImageDraw.Draw(img) # 标题区域 draw.rectangle([0, 0, card_width, 60], fill=accent_color) draw.text((20, 20), title, fill='white') # 数据项 y = 80 for key, value in data_dict.items(): draw.text((30, y), f'• {key}:', fill=text_color) draw.text((200, y), str(value), fill=accent_color) # 如果是百分比，添加进度条 if isinstance(value, (int, float)) and 0 <= value <= 100: bar_width = 200 bar_height = 20 bar_x = 250 # 进度条背景 draw.rectangle([bar_x, y, bar_x + bar_width, y + bar_height], fill='#ecf0f1' if theme == 'light' else '#34495e') # 进度条前景 fill_width = int(bar_width * value / 100) draw.rectangle([bar_x, y, bar_x + fill_width, y + bar_height], fill=accent_color) # 百分比文本 text_color_bar = 'white' if value > 50 else text_color draw.text((bar_x + 5, y), f'{value}%', fill=text_color_bar) y += 40 # 边框 draw.rectangle([10, 70, card_width - 10, card_height - 10], outline=accent_color, width=2) path = tempfile.mktemp(suffix='.png') img.save(path) return path except Exception as e: raise Exception(f'数据卡片生成失败: {e}') def create_system_status(self, status_data): """生成系统状态图""" try: width, height = 600, 400 img = Image.new('RGB', (width, height), color='#f8f9fa') draw = ImageDraw.Draw(img) # 标题 draw.rectangle([0, 0, width, 60], fill='#2c3e50') draw.text((20, 20), '🖥️ 系统状态监控', fill='white', font=self.default_font) items = list(status_data.items()) y_start = 80 for i, (label, value) in enumerate(items): y = y_start + i * 50 draw.text((40, y), f'• {label}', fill='#34495e', font=self.default_font) draw.text((200, y), str(value), fill='#3498db', font=self.default_font) import datetime timestamp = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S') draw.text((40, height-30), f'生成时间: {timestamp}', fill='#7f8c8d', font=self.default_font) path = tempfile.mktemp(suffix='.png') img.save(path) return path except Exception as e: raise Exception(f'系统状态图生成失败: {e}') def create_comparison_chart(self, data_sets, labels, chart_type='grouped_bar'): """创建对比图表""" try: plt.figure(figsize=(12, 7)) x = np.arange(len(labels)) width = 0.8 / len(data_sets) if chart_type == 'grouped_bar': for i, data in enumerate(data_sets): offset = (i - len(data_sets)/2 + 0.5) * width bars = plt.bar(x + offset, data, width, label=f'数据集 {i+1}') # 添加数值标签 for bar in bars: height = bar.get_height() plt.text(bar.get_x() + bar.get_width()/2., height + 0.1, f'{height}', ha='center', va='bottom', fontsize=8) elif chart_type == 'stacked_bar': bottom = np.zeros(len(labels)) for i, data in enumerate(data_sets): plt.bar(x, data, width, bottom=bottom, label=f'数据集 {i+1}') bottom += data elif chart_type == 'line': for i, data in enumerate(data_sets): plt.plot(x, data, marker='o', linewidth=2, label=f'数据集 {i+1}') plt.xlabel('类别', fontsize=12) plt.ylabel('数值', fontsize=12) plt.title('数据对比分析', fontsize=14, fontweight='bold') plt.xticks(x, labels) plt.legend() plt.grid(True, alpha=0.3) plt.tight_layout() path = tempfile.mktemp(suffix='.png') plt.savefig(path, dpi=120, bbox_inches='tight') plt.close() return path except Exception as e: plt.close() raise Exception(f'对比图表生成失败: {e}') # 创建技能实例 skill = ImageGenerationSkill() # 快捷函数 def create_chart(*args, **kwargs): return skill.create_chart(*args, **kwargs) def create_qrcode(*args, **kwargs): return skill.create_qrcode(*args, **kwargs) def text_to_image(*args, **kwargs): return skill.text_to_image(*args, **kwargs) def create_wordcloud(*args, **kwargs): return skill.create_wordcloud(*args, **kwargs) def create_color_palette(*args, **kwargs): return skill.create_color_palette(*args, **kwargs) def create_data_card(*args, **kwargs): return skill.create_data_card(*args, **kwargs) def create_system_status(*args, **kwargs): return skill.create_system_status(*args, **kwargs) def create_comparison_chart(*args, **kwargs): return skill.create_comparison_chart(*args, **kwargs) __all__ = [ 'ImageGenerationSkill', 'skill', 'create_chart', 'create_qrcode', 'text_to_image', 'create_wordcloud', 'create_color_palette', 'create_data_card', 'create_system_status', 'create_comparison_chart' ]