Rytr AI短文案生成产品设计

import random class ToneInjector:    TONE_MAP = {        'formal': '请以正式且专业的语气撰写以下内容:',        'casual': '请以轻松随意的语气撰写以下内容:',        'persuasive': '请以具有说服力和感染力的语气撰写以下内容:',        'humorous': '请以幽默风趣的语气撰写以下内容:',    }     def __init__(self, tone_name):        self.tone_name = tone_name        self.prefix = self.TONE_MAP.get(tone_name, self.TONE_MAP['formal'])     def inject(self, base_prompt):        return f'{self.prefix}\n{base_prompt}'  class UseCaseTemplate:    def __init__(self, template_name, max_length, required_fields):        self.template_name = template_name        self.max_length = max_length        self.required_fields = required_fields     def validate_input(self, user_input):        missing = [f for f in self.required_fields if f not in user_input]        if missing:            raise ValueError(f'缺少必填字段: {missing}')        return True     def build_prompt(self, user_input):        self.validate_input(user_input)        parts = [f'[模板:{self.template_name}]', f'[长度限制:{self.max_length}字]']        for key, value in user_input.items():            parts.append(f'[{key}:{value}]')        return ' '.join(parts)  class MultilingualGenerator:    SUPPORTED_LANGUAGES = ['en', 'zh', 'ja', 'ko', 'fr', 'de', 'es', 'pt', 'ru', 'ar']     def __init__(self, target_lang):        if target_lang not in self.SUPPORTED_LANGUAGES:            raise ValueError(f'不支持的语言: {target_lang}')        self.target_lang = target_lang     def generate(self, prompt):        lang_labels = {'en': 'English', 'zh': '中文', 'ja': '日本語', 'ko': '한국어', 'fr': 'Francais', 'de': 'Deutsch'}        label = lang_labels.get(self.target_lang, self.target_lang)        return f'[{label}] 生成文案: {prompt[:50]}...'  class QualityScorer:    def score(self, text, keywords=None):        fluency = round(random.uniform(7.0, 9.5), 1)        attractiveness = round(random.uniform(6.5, 9.0), 1)        relevance = round(random.uniform(7.0, 9.5), 1)        overall = round((fluency + attractiveness + relevance) / 3, 1)        return {'fluency': fluency, 'attractiveness': attractiveness, 'relevance': relevance, 'overall': overall}  class RytrPipeline:    def __init__(self, template_name, tone, language, max_length=100, required_fields=None):        self.template = UseCaseTemplate(template_name, max_length, required_fields or ['product_name', 'keywords'])        self.tone_injector = ToneInjector(tone)        self.lang_generator = MultilingualGenerator(language)        self.scorer = QualityScorer()     def run(self, user_input, num_variants=3):        base_prompt = self.template.build_prompt(user_input)        toned_prompt = self.tone_injector.inject(base_prompt)        variants = []        for i in range(num_variants):            text = self.lang_generator.generate(f'{toned_prompt} [变体{i+1}]')            scores = self.scorer.score(text)            variants.append({'text': text, 'scores': scores})        variants.sort(key=lambda v: v['scores']['overall'], reverse=True)        return {'template': self.template.template_name, 'tone': self.tone_injector.tone_name, 'language': self.lang_generator.target_lang, 'variants': variants, 'best': variants[0]}  if __name__ == '__main__':    pipeline = RytrPipeline(template_name='product_description', tone='persuasive', language='zh', max_length=200, required_fields=['product_name', 'keywords'])    user_input = {'product_name': '无线蓝牙耳机', 'keywords': '降噪,长续航,舒适佩戴', 'target_audience': '年轻上班族'}    result = pipeline.run(user_input, num_variants=3)    print(f'最佳变体综合评分: {result["best"]["scores"]["overall"]}')    for idx, v in enumerate(result['variants']):        print(f'变体{idx+1}: 流畅度={v["scores"]["fluency"]}, 吸引力={v["scores"]["attractiveness"]}, 相关性={v["scores"]["relevance"]}')