""" 高级图片生成技能 - 整合多个库 """ 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 import ascii_magic class AdvancedImageSkill: """高级图片生成技能""" def __init__(self): plt.switch_backend('Agg') sns.set_style("whitegrid") self.default_font = ImageFont.load_default() print("Advanced Image Skill initialized") def create_advanced_chart(self, data, chart_type='bar', title='', style='seaborn'): """创建高级图表 Args: data: 数据字典 chart_type: 图表类型 ('bar', 'line', 'scatter', 'heatmap', 'pie') title: 图表标题 style: 样式 ('seaborn', 'matplotlib') 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 == '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']) elif chart_type == 'heatmap' and 'matrix' in data: if style == 'seaborn': sns.heatmap(data['matrix'], annot=True, fmt='.1f', cmap='YlOrRd') else: plt.imshow(data['matrix'], cmap='hot', interpolation='nearest') plt.colorbar() elif chart_type == 'pie': plt.pie(values, labels=labels, autopct='%1.1f%%', startangle=90) plt.axis('equal') # 确保饼图是圆形 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_wordcloud(self, text, width=800, height=400, background_color='white', colormap='viridis'): """创建词云图 Args: text: 文本内容 width: 图片宽度 height: 图片高度 background_color: 背景颜色 colormap: 颜色映射 Returns: 图片路径 """ 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_ascii_art(self, image_path=None, text=None, columns=100): """创建ASCII艺术 Args: image_path: 图片路径（将图片转为ASCII） text: 文本（创建文本ASCII艺术） columns: 列数 Returns: ASCII文本或图片路径 """ try: if image_path and os.path.exists(image_path): # 从图片创建ASCII艺术 output = ascii_magic.from_image( image_path, columns=columns, mode=ascii_magic.Modes.HTML ) # 保存为HTML path = tempfile.mktemp(suffix='.html') ascii_magic.to_html_file(path, output) return path elif text: # 从文本创建ASCII艺术 # 这里简化处理，实际可以使用更复杂的ASCII艺术生成 ascii_text = f""" ╔{'═' * (columns-2)}╗ ║{' ' * (columns-2)}║ ║{text.center(columns-2)}║ ║{' ' * (columns-2)}║ ╚{'═' * (columns-2)}╝ """ path = tempfile.mktemp(suffix='.txt') with open(path, 'w', encoding='utf-8') as f: f.write(ascii_text) return path else: raise Exception('需要提供图片路径或文本') except Exception as e: raise Exception(f'ASCII艺术生成失败: {e}') def create_color_palette(self, colors, title='颜色调色板'): """创建颜色调色板展示 Args: colors: 颜色列表，如 ['#FF5733', '#33FF57', '#3357FF'] title: 标题 Returns: 图片路径 """ 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'): """创建数据卡片 Args: title: 卡片标题 data_dict: 数据字典 theme: 主题 ('light', 'dark') Returns: 图片路径 """ 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' if theme == 'dark' else 'white') # 数据项 y = 80 for key, value in data_dict.items(): # 键 draw.text((30, y), f'• {key}:', fill=text_color) # 值（如果是数字，可以添加进度条） value_x = 200 draw.text((value_x, 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) # 百分比文本 draw.text((bar_x + 5, y), f'{value}%', fill='white' if value > 50 else text_color) 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_comparison_chart(self, data_sets, labels, chart_type='grouped_bar'): """创建对比图表 Args: data_sets: 数据集列表，每个数据集是一个数值列表 labels: 标签列表 chart_type: 图表类型 ('grouped_bar', 'stacked_bar', 'line') Returns: 图片路径 """ 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}') # 创建技能实例 advanced_skill = AdvancedImageSkill() # 导出函数 def create_advanced_chart(*args, **kwargs): return advanced_skill.create_advanced_chart(*args, **kwargs) def create_wordcloud(*args, **kwargs): return advanced_skill.create_wordcloud(*args, **kwargs) def create_ascii_art(*args, **kwargs): return advanced_skill.create_ascii_art(*args, **kwargs) def create_color_palette(*args, **kwargs): return advanced_skill.create_color_palette(*args, **kwargs) def create_data_card(*args, **kwargs): return advanced_skill.create_data_card(*args, **kwargs) def create_comparison_chart(*args, **kwargs): return advanced_skill.create_comparison_chart(*args, **kwargs) __all__ = [ 'AdvancedImageSkill', 'advanced_skill', 'create_advanced_chart', 'create_wordcloud', 'create_ascii_art', 'create_color_palette', 'create_data_card', 'create_comparison_chart' ]