Spaces:
Running
Running
| import json | |
| import requests | |
| import os | |
| class TextAnalyzer: | |
| """ | |
| テキストのハラスメント検出と会話評価を行うクラス。 | |
| """ | |
| def __init__(self, file_path, keywords): | |
| """ | |
| TextAnalyzer クラスのコンストラクタ。 | |
| Args: | |
| file_path (str): 分析するテキストファイルのパス。 | |
| keywords (list): ハラスメント検出に使用するキーワードのリスト。 | |
| """ | |
| self.file_path = file_path | |
| self.keywords = keywords | |
| self.text_content = None # テキストファイルの内容を格納 | |
| self.harassment_detected = False # ハラスメントが検出されたかどうか | |
| self.harassment_keywords = [] # 検出されたハラスメントキーワードのリスト | |
| self.deepseek_analysis = {} # DeepSeek API による分析結果を格納する辞書 | |
| self.api_key = None | |
| def load_text(self): | |
| """ | |
| テキストファイルを読み込み、その内容を self.text_content に格納する。 | |
| Returns: | |
| bool: ファイルの読み込みに成功した場合は True、失敗した場合は False。 | |
| """ | |
| try: | |
| with open(self.file_path, 'r', encoding='utf-8') as file: | |
| self.text_content = file.read() | |
| return True | |
| except Exception as e: | |
| print(f"ファイル読み込みエラー: {e}") | |
| return False | |
| def detect_harassment(self): | |
| """ | |
| テキスト内容からハラスメントを検出する。 | |
| Returns: | |
| bool: ハラスメントが検出された場合は True、それ以外は False。 | |
| """ | |
| if not self.text_content: | |
| return False | |
| self.harassment_keywords = [] | |
| for keyword in self.keywords: | |
| if keyword in self.text_content: | |
| self.harassment_detected = True | |
| self.harassment_keywords.append(keyword) | |
| return self.harassment_detected | |
| def analyze_with_deepseek(self, api_key=None, api_url="https://api.deepseek.com/v1/chat/completions"): | |
| """ | |
| DeepSeek API を使用して会話を分析する。会話レベルやハラスメントの詳細な検出を行う。 | |
| Args: | |
| api_key (str, optional): DeepSeek API キー。指定されない場合は環境変数から取得。 | |
| api_url (str, optional): DeepSeek API の URL。デフォルトは標準のチャット補完エンドポイント。 | |
| Returns: | |
| bool: 分析に成功した場合は True、失敗した場合は False。 | |
| """ | |
| if not self.text_content: | |
| return False | |
| # 提供された API キーを使用するか、環境変数から取得する | |
| if api_key: | |
| self.api_key = api_key | |
| else: | |
| self.api_key = os.environ.get("DEEPSEEK_API_KEY") | |
| if not self.api_key: | |
| print("DeepSeek API キーが提供されておらず、環境変数にも見つかりませんでした。") | |
| return False | |
| headers = { | |
| "Content-Type": "application/json", | |
| "Authorization": f"Bearer {self.api_key}" | |
| } | |
| prompt = f""" | |
| 以下の会話を分析し、結果を JSON 形式で返してください。1 から 10 のスケールで評価し、10 が最高です。 | |
| 厳密に評価してください。ハラスメントが存在する場合は、その種類を具体的に記述してください。 | |
| 評価基準: | |
| 1. conversationLevel: 会話のレベル (初心者、中級者、上級者)。 | |
| 2. harassmentPresent: 会話にハラスメント表現が含まれているかどうか (true/false)。 | |
| 3. harassmentType: ハラスメントが存在する場合、その種類を具体的に記述。 | |
| 4. topicAppropriateness: 会話のトピックが適切かどうか。 | |
| 5. improvementSuggestions: 会話を改善するための具体的な提案。 | |
| 6. repetition: 同じことがどの程度繰り返されているか。(1-10) | |
| 7. pleasantConversation: 会話がどの程度心地よいか。(1-10) | |
| 8. blameOrHarassment: 会話がどの程度相手を責めたり、ハラスメントをしているか。(1-10) | |
| 会話内容: | |
| {self.text_content} | |
| JSON 形式のみを返してください。 | |
| """ | |
| data = { | |
| "model": "deepseek-chat", | |
| "messages": [{"role": "user", "content": prompt}], | |
| "response_format": {"type": "json_object"} | |
| } | |
| try: | |
| response = requests.post(api_url, headers=headers, json=data) | |
| response.raise_for_status() | |
| result = response.json() | |
| deepseek_response = json.loads(result["choices"][0]["message"]["content"]) | |
| # 指定されたキーを使用して、インスタンス変数に値を割り当てる | |
| self.deepseek_analysis = { | |
| "conversationLevel": deepseek_response.get("conversationLevel"), | |
| "harassmentPresent": deepseek_response.get("harassmentPresent"), | |
| "harassmentType": deepseek_response.get("harassmentType"), | |
| "topicAppropriateness": deepseek_response.get("topicAppropriateness"), | |
| "improvementSuggestions": deepseek_response.get("improvementSuggestions"), | |
| "repetition": deepseek_response.get("repetition"), | |
| "pleasantConversation": deepseek_response.get("pleasantConversation"), | |
| "blameOrHarassment": deepseek_response.get("blameOrHarassment"), | |
| } | |
| return True | |
| except requests.exceptions.RequestException as e: | |
| print(f"DeepSeek API リクエストエラー: {e}") | |
| return False | |
| except json.JSONDecodeError as e: | |
| print(f"DeepSeek API レスポンスの JSON デコードエラー: {e}") | |
| print(f"レスポンス内容: {response.text}") | |
| return False | |
| except KeyError as e: | |
| print(f"DeepSeek API レスポンスのキーエラー: {e}") | |
| print(f"レスポンス内容: {response.text}") | |
| return False | |
| except Exception as e: | |
| print(f"DeepSeek API エラー: {e}") | |
| return False | |
| def get_analysis_results(self): | |
| """ | |
| 分析結果を返す。 | |
| Returns: | |
| dict: 分析結果を含む辞書。 | |
| """ | |
| results = { | |
| "text_content": self.text_content, | |
| "basic_harassment_detection": { | |
| "detected": self.harassment_detected, | |
| "matching_keywords": self.harassment_keywords | |
| }, | |
| "deepseek_analysis": self.deepseek_analysis | |
| } | |
| return results | |
| def analyze(self, api_key=None): | |
| """ | |
| すべての分析を実行し、結果を返す。 | |
| Args: | |
| api_key (str, optional): DeepSeek API キー。 | |
| Returns: | |
| dict: 分析結果またはエラーメッセージを含む辞書。 | |
| """ | |
| if not self.load_text(): | |
| return {"error": "テキストファイルの読み込みに失敗しました。"} | |
| self.detect_harassment() | |
| if not self.analyze_with_deepseek(api_key): | |
| return {"error": "DeepSeek API 分析に失敗しました。"} | |
| return self.get_analysis_results() | |
| ''' | |
| # 使用例 | |
| if __name__ == "__main__": | |
| # ハラスメント検出用のキーワード例 | |
| harassment_keywords = [ | |
| "バカ", "馬鹿", "アホ", "死ね", "クソ", "うざい", | |
| "きもい", "キモい", "ブス", "デブ", "ハゲ", | |
| "セクハラ", "パワハラ", "モラハラ" | |
| ] | |
| # 分析インスタンスの作成 | |
| analyzer = TextAnalyzer("./2.txt", harassment_keywords) | |
| # DeepSeek API キー (環境変数から取得するか、直接渡す) | |
| # api_key = os.environ.get("DEEPSEEK_API_KEY") | |
| # 分析の実行 | |
| results = analyzer.analyze(api_key=api_key) | |
| # 結果の出力 | |
| print(json.dumps(results, ensure_ascii=False, indent=2)) | |
| # 特定の値へのアクセス例 | |
| if "deepseek_analysis" in results and results["deepseek_analysis"]: | |
| deepseek_data = results["deepseek_analysis"] | |
| conversation_level = deepseek_data.get("conversationLevel") | |
| harassment_present = deepseek_data.get("harassmentPresent") | |
| harassment_type = deepseek_data.get("harassmentType") | |
| repetition = deepseek_data.get("repetition") | |
| pleasantConversation = deepseek_data.get("pleasantConversation") | |
| blameOrHarassment = deepseek_data.get("blameOrHarassment") | |
| print("\n--- DeepSeek 分析結果 ---") | |
| print(f"会話レベル: {conversation_level}") | |
| print(f"ハラスメントの有無: {harassment_present}") | |
| print(f"ハラスメントの種類: {harassment_type}") | |
| print(f"繰り返しの程度: {repetition}") | |
| print(f"会話の心地よさ: {pleasantConversation}") | |
| print(f"非難またはハラスメントの程度: {blameOrHarassment}") | |
| ''' |