Upload analyze.py

analyze.py

@@ -0,0 +1,224 @@
+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  # DeepSeek API キー
+
+    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}")
+'''