Spaces:
Sleeping
Sleeping
| from textblob import TextBlob | |
| # 定义颜色映射 | |
| ADJECTIVE_COLORS = { | |
| "positive": "#4CAF50", # 绿色 | |
| "negative": "#F44336", # 红色 | |
| "neutral": "#FFC107" # 黄色 | |
| } | |
| # 自定义短语情感覆盖规则 | |
| PHRASE_SENTIMENT_OVERRIDES = { | |
| "significant drop": "negative", | |
| "significant drops": "negative", | |
| "sharp decline": "negative", | |
| "strong performance": "positive", | |
| "Poor performance": "negative" | |
| # 可以继续添加更多短语规则... | |
| } | |
| # 负面触发词集合 | |
| NEGATIVE_TRIGGERS = {"drop", "decline", "failure", "loss", "down", "worse", "weak", "poor"} | |
| def get_phrase_sentiment(phrase): | |
| """增强的短语情感分析逻辑""" | |
| # 1. 优先检查自定义规则 | |
| lower_phrase = phrase.lower() | |
| if lower_phrase in PHRASE_SENTIMENT_OVERRIDES: | |
| return PHRASE_SENTIMENT_OVERRIDES[lower_phrase] | |
| # 2. 检查负面触发词 | |
| words = TextBlob(phrase).words | |
| if any(w.lower() in NEGATIVE_TRIGGERS for w in words): | |
| return "negative" | |
| # 3. 默认情感分析 | |
| sentiment = TextBlob(phrase).sentiment.polarity | |
| if sentiment > 0.1: | |
| return "positive" | |
| elif sentiment < -0.1: | |
| return "negative" | |
| else: | |
| return "neutral" | |
| def highlight_adjectives(text): | |
| """高亮形容词短语并根据情感着色""" | |
| if not isinstance(text, str) or not text.strip(): | |
| return text | |
| try: | |
| blob = TextBlob(text) | |
| highlighted = [] | |
| i = 0 | |
| tags = blob.tags | |
| while i < len(tags): | |
| word, tag = tags[i] | |
| # 检查形容词短语模式 (形容词+名词) | |
| if tag.startswith('JJ') and i+1 < len(tags) and tags[i+1][1].startswith('NN'): | |
| phrase = f"{word} {tags[i+1][0]}" | |
| # 使用增强的情感分析 | |
| sentiment = get_phrase_sentiment(phrase) | |
| color = ADJECTIVE_COLORS.get(sentiment, "#000000") | |
| highlighted.append(f'<span style="color: {color}; font-weight: bold">{phrase}</span>') | |
| i += 2 # 跳过下一个词，因为已经处理了 | |
| elif tag.startswith('JJ'): # 单独形容词 | |
| sentiment = get_phrase_sentiment(word) # 也能处理单个词 | |
| color = ADJECTIVE_COLORS.get(sentiment, "#000000") | |
| highlighted.append(f'<span style="color: {color}; font-weight: bold">{word}</span>') | |
| i += 1 | |
| else: | |
| highlighted.append(word) | |
| i += 1 | |
| # 保留原始空格和标点 | |
| return " ".join(highlighted).replace(" ,", ",").replace(" .", ".").replace(" '", "'") | |
| except Exception as e: | |
| print(f"Error processing text: {e}") | |
| return text | |