analyze.py

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}")

'''