Update process.py

process.py

@@ -7,6 +7,8 @@ import random
 from datetime import datetime
 from pyannote.audio import Model, Inference
 from pydub import AudioSegment
+import base64
+import binascii
 
 class AudioProcessor():
     def __init__(self,cache_dir = "/tmp/hf_cache"):
@@ -142,4 +144,42 @@ class AudioProcessor():
         return matched_time
         
 
-        
+    def save_audio_from_base64(self,base64_audio,output_dir,output_filename,temp_format='webm'):
+        try:
+            # Base64デコードして音声バイナリを取得
+            try:
+                audio_binary = base64.b64decode(base64_audio)
+            except binascii.Error:
+                raise ValueError("Invalid Base64 input data")
+            
+            # 保存するディレクトリを作成
+            os.makedirs(output_dir,exist_ok=True)
+
+            # 一時ファイルに保存（実際の形式は WebM などと仮定）
+            temp_audio_path = os.path.join(output_dir,"temp_audio")
+            try:
+                with open(temp_audio_path,'wb') as f:
+                    f.write(audio_binary)
+            
+                # pydub を使って一時ファイルを WAV に変換
+                # ※ここでは WebM 形式と仮定していますが、実際の形式に合わせて format の指定を変更してください
+                try:
+                    audio = AudioSegment.from_file(temp_audio_path,format=temp_format)
+                except Exception as e:
+                    audio = AudioSegment.from_file(temp_audio_path) #形式が不明な場合は自動判別させる（ただし変換できない場合もあり）
+            
+                # 音声ファイルを保存
+                wav_audio_path = os.path.join(output_dir,output_filename) 
+                audio.export(wav_audio_path,format="wav")
+            finally:
+                #一時ファイルを削除
+                if os.path.exists(temp_audio_path):
+                    os.remove(temp_audio_path)
+            return wav_audio_path
+        except ValueError as e:
+            print(f"Value Error: {e}")
+        except FileNotFoundError as e:
+            print(f"File Not Found Error: {e}")
+        except Exception as e:
+            print(f"Unexpected Error: {e}")
+        return None