Upload folderMaker.py with huggingface_hub

folderMaker.py

@@ -0,0 +1,144 @@
+import os
+import hashlib
+import requests
+import urllib.parse
+from translation_data import (
+    translation_dict_A,
+    translation_dict_B,
+    translation_dict_C,
+    translation_dict_D,
+    translation_dict_F
+    # 必要に応じて translation_dict_E, translation_dict_G などを追加
+)
+
+# 設定
+AUDIO_VARIANTS = 3  # 各フレーズあたりの音声バリアント数
+OUTPUT_DIR = os.path.join('static', 'audio')  # 音声ファイルの保存先
+HASH_LENGTH = 8  # ハッシュの長さ
+VOICE_GENERATION_URL = "http://192.168.1.80:5000/voice"  # 音声生成APIのURL
+
+# バリアントごとの model_id の設定
+VARIANT_SETTINGS = {
+    1: {'model_id': 9},
+    2: {'model_id': 11},
+    3: {'model_id': 12},
+}
+
+def generate_truncated_hash(text, length=HASH_LENGTH):
+    """
+    指定されたテキストのトランケートされたSHA-256ハッシュを生成します。
+
+    Args:
+        text (str): ハッシュ化するテキスト。
+        length (int): トランケートするハッシュの長さ。
+
+    Returns:
+        str: トランケートされたハッシュ文字列。
+    """
+    hash_object = hashlib.sha256(text.encode('utf-8'))
+    hex_dig = hash_object.hexdigest()
+    return hex_dig[:length]
+
+def create_directory(path):
+    """
+    指定されたパスにディレクトリを作成します。存在しない場合のみ作成します。
+
+    Args:
+        path (str): 作成するディレクトリのパス。
+    """
+    os.makedirs(path, exist_ok=True)
+
+def generate_audio(text, model_id):
+    """
+    音声生成APIにリクエストを送り、音声データを取得します。
+    `speaker_id` は固定値 0 を使用します。
+
+    Args:
+        text (str): 音声化するテキスト。
+        model_id (int): 使用するモデルID。
+
+    Returns:
+        bytes: 取得した音声データのバイナリ。
+    """
+    params = {
+        'text': text,
+        'model_id': model_id,
+        'speaker_id': 0,  # speaker_id は常に 0 で問題ない仕様
+        'sdp_ratio': 0.2,
+        'noise': 0.6,
+        'noisew': 0.8,
+        'length': 1.4,
+        'language': 'JP',
+        'auto_split': 'true',
+        'split_interval': 0.5,
+        'assist_text_weight': 1,
+        'style': 'Neutral',
+        'style_weight': 1
+    }
+
+    try:
+        response = requests.get(VOICE_GENERATION_URL, params=params, timeout=30)
+        response.raise_for_status()  # HTTPエラーがあれば例外を発生させる
+        return response.content
+    except requests.exceptions.RequestException as e:
+        print(f"Error generating audio for text '{text}': {e}")
+        return None
+
+def process_translation_dicts():
+    """
+    すべての翻訳辞書を処理し、音声バリアントを生成してフォルダに保存します。
+    """
+    # 翻訳辞書のリスト
+    translation_dicts: list[dict[str, str]] = [
+        translation_dict_A,
+        translation_dict_B,
+        translation_dict_C,
+        translation_dict_D,
+        translation_dict_F
+        # 必要に応じて translation_dict_E, translation_dict_G などを追加
+    ]
+
+    for dict_index, translation_dict in enumerate(translation_dicts, start=1):
+        set_identifier = chr(64 + dict_index)  # 'A', 'B', 'C', etc.
+        print(f"\nProcessing translation_dict_{set_identifier}")
+
+        for chinese_sentence, japanese_sentence in translation_dict.items():
+            # 日本語の文からハッシュを生成
+            sentence_hash = generate_truncated_hash(japanese_sentence)
+            folder_path = os.path.join(OUTPUT_DIR, sentence_hash)
+
+            # フォルダを作成
+            create_directory(folder_path)
+            print(f"Created/Verified directory: {folder_path}")
+
+            for variant_num in range(1, AUDIO_VARIANTS + 1):
+                settings = VARIANT_SETTINGS.get(variant_num)
+                if not settings:
+                    print(f"No settings defined for variant {variant_num}. Skipping.")
+                    continue
+
+                model_id = settings['model_id']
+                audio_filename = f"variant_{variant_num}.mp3"
+                audio_filepath = os.path.join(folder_path, audio_filename)
+                
+                # 既にファイルが存在する場合はスキップ
+                """
+                if os.path.exists(audio_filepath):
+                    print(f"Audio variant already exists: {audio_filepath}")
+                    continue
+
+                print(f"Generating {audio_filename} with model_id={model_id}, {japanese_sentence}")
+                """
+                # 音声生成
+                audio_data = generate_audio(japanese_sentence, model_id)
+                if audio_data:
+                    # ファイルに保存
+                    with open(audio_filepath, 'wb') as f:
+                        f.write(audio_data)
+                    print(f"Saved audio variant: {audio_filepath}, {japanese_sentence}")
+                else:
+                    print(f"Failed to generate audio for variant {variant_num} of sentence: {japanese_sentence}")
+
+if __name__ == "__main__":
+    process_translation_dicts()
+    print("\nAudio variant generation completed.")