Upload app.py

app.py

@@ -1,151 +1,29 @@
-from flask import Flask, request, jsonify, render_template, send_from_directory,redirect, make_response, Response, session, url_for 
+from flask import Flask, request, jsonify, render_template, send_from_directory
 import base64
 from pydub import AudioSegment  # 変換用にpydubをインポート
 import os
 import shutil
-import numpy as np
-import string
-import random
-from datetime import datetime, timedelta
-from pyannote.audio import Model, Inference
-from pydub import AudioSegment
-from flask_sqlalchemy import SQLAlchemy
-from dotenv import load_dotenv
-from google.oauth2 import id_token
-from google_auth_oauthlib.flow import Flow
-from google.auth.transport import requests as google_requests
-
-
-
-# Hugging Face のトークン取得（環境変数 HF に設定）
-#hf_token = os.environ.get("HF")
-load_dotenv()
-hf_token = os.getenv("HF")
-if hf_token is None:
-    raise ValueError("HUGGINGFACE_HUB_TOKEN が設定されていません。")
-
-# キャッシュディレクトリの作成（書き込み可能な /tmp を利用）
-cache_dir = "/tmp/hf_cache"
-os.makedirs(cache_dir, exist_ok=True)
-
-# pyannote モデルの読み込み
-model = Model.from_pretrained("pyannote/embedding", use_auth_token=hf_token, cache_dir=cache_dir)
-inference = Inference(model)
+from process import AudioProcessor
 
+process=AudioProcessor()
 app = Flask(__name__)
 
-app.config['SECRET_KEY'] = os.urandom(24)
-
-def cosine_similarity(vec1, vec2):
-    vec1 = vec1 / np.linalg.norm(vec1)
-    vec2 = vec2 / np.linalg.norm(vec2)
-    return np.dot(vec1, vec2)
-
-def segment_audio(path, target_path='/tmp/setup_voice', seg_duration=1.0):
-    """
-    音声を指定秒数ごとに分割する。
-    target_path に分割したファイルを保存し、元の音声の総長（ミリ秒）を返す。
-    """
-    os.makedirs(target_path, exist_ok=True)
-    base_sound = AudioSegment.from_file(path)
-    duration_ms = len(base_sound)
-    seg_duration_ms = int(seg_duration * 1000)
-    
-    for i, start in enumerate(range(0, duration_ms, seg_duration_ms)):
-        end = min(start + seg_duration_ms, duration_ms)
-        segment = base_sound[start:end]
-        segment.export(os.path.join(target_path, f'{i}.wav'), format="wav")
-    
-    return target_path, duration_ms
-
-def calculate_similarity(path1, path2):
-    embedding1 = inference(path1)
-    embedding2 = inference(path2)
-    return float(cosine_similarity(embedding1.data.flatten(), embedding2.data.flatten()))
-
-def process_audio(reference_path, input_path, output_folder='/tmp/data/matched_segments', seg_duration=1.0, threshold=0.5):
-    """
-    入力音声ファイルを seg_duration 秒ごとに分割し、各セグメントと参照音声の類似度を計算。
-    類似度が threshold を超えたセグメントを output_folder にコピーし、マッチした時間（ms）と
-    マッチしなかった時間（ms）を返す。
-    """
-    os.makedirs(output_folder, exist_ok=True)
-    segmented_path, total_duration_ms = segment_audio(input_path, seg_duration=seg_duration)
-    
-    matched_time_ms = 0
-    for file in sorted(os.listdir(segmented_path)):
-        segment_file = os.path.join(segmented_path, file)
-        similarity = calculate_similarity(segment_file, reference_path)
-        if similarity > threshold:
-            shutil.copy(segment_file, output_folder)
-            matched_time_ms += len(AudioSegment.from_file(segment_file))
-    
-    unmatched_time_ms = total_duration_ms - matched_time_ms
-    return matched_time_ms, unmatched_time_ms
-
-def generate_random_string(length):
-    letters = string.ascii_letters + string.digits
-    return ''.join(random.choice(letters) for i in range(length))
-
-def generate_filename(random_length):
-    random_string = generate_random_string(random_length)
-    current_time = datetime.now().strftime("%Y%m%d%H%M%S")
-    filename = f"{current_time}_{random_string}.wav"
-    return filename
-
 # トップページ（テンプレート: index.html）
 @app.route('/')
-def top():
-    return redirect('index')
-
+@app.route('/index', methods=['GET', 'POST'])
+def index():
+    return render_template('index.html')
 
 # フィードバック画面（テンプレート: feedback.html）
 @app.route('/feedback', methods=['GET', 'POST'])
 def feedback():
-    #ログイン問題解決しだい戻す
-    """
-    if 'google_id' not in session:
-        return redirect(url_for('login'))
-    user_info = {
-        'name': session.get('name'),
-        'email': session.get('email')
-    }
-    """
     return render_template('feedback.html')
 
 # 会話詳細画面（テンプレート: talkDetail.html）
 @app.route('/talk_detail', methods=['GET', 'POST'])
 def talk_detail():
-    """
-    if 'google_id' not in session:
-        return redirect(url_for('login'))
-    user_info = {
-        'name': session.get('name'),
-        'email': session.get('email')
-   }
-   """
     return render_template('talkDetail.html')
 
-   # インデックス画面（テンプレート: index.html）
-@app.route('/index', methods=['GET', 'POST'])
-def index():
-    """
-   if 'google_id' not in session:
-        return redirect(url_for('login'))
-   user_info = {
-        'name': session.get('name'),
-        'email': session.get('email')
-   }
-   """
-    return render_template('index.html')
-
-@app.before_request
-def before_request():
-    # リクエストのたびにセッションの寿命を更新する
-    session.permanent = True
-    app.permanent_session_lifetime = timedelta(minutes=15)
-    session.modified = True
-
 # 音声アップロード＆解析エンドポイント
 @app.route('/upload_audio', methods=['POST'])
 def upload_audio():
@@ -215,8 +93,6 @@ def upload_base_audio():
         print("Error in /upload_base_audio:", str(e))
         return jsonify({"error": "サーバーエラー", "details": str(e)}), 500
 
-
-
 if __name__ == '__main__':
     port = int(os.environ.get("PORT", 7860))
     app.run(debug=True, host="0.0.0.0", port=port)