74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
from flask import Flask, request, jsonify, send_file, Response, stream_with_context
|
||
from flask_cors import CORS
|
||
import io
|
||
import requests
|
||
import uuid
|
||
from datetime import datetime
|
||
import os
|
||
|
||
app = Flask(__name__)
|
||
CORS(app, supports_credentials=True)
|
||
|
||
def transcribe_audio(audio, language):
|
||
"""调用WeNet API，用户提问转文字"""
|
||
url = "http://127.0.0.1:9991/transcribe"
|
||
files = {'audio': open(audio, 'rb')}
|
||
data = {'language': language}
|
||
response = requests.post(url, files=files, data=data)
|
||
if response.status_code == 200:
|
||
return response.json()['text']
|
||
else:
|
||
raise Exception("Error transcribing audio")
|
||
|
||
|
||
def generate_response_audio(transcription, language, audio):
|
||
"""调用生成回答音频API"""
|
||
url = "http://127.0.0.1:9992/tts"
|
||
data = {
|
||
"text": transcription,
|
||
'language': language
|
||
}
|
||
files = {'audio': open(audio, 'rb')}
|
||
response_audio = requests.post(url, data=data, files=files)
|
||
if response_audio.status_code == 200:
|
||
return io.BytesIO(response_audio.content)
|
||
else:
|
||
raise Exception("Error generating response audio")
|
||
|
||
@app.route('/process_audio', methods=['POST'])
|
||
def process_audio():
|
||
# 检查语言表单项
|
||
if 'audio' not in request.files or 'language' not in request.form:
|
||
return jsonify({"error": "Audio file and language must be provided"}), 400
|
||
|
||
# 从请求中获取语言设置和音频
|
||
request_audio = request.files['audio']
|
||
language = request.form['language']
|
||
|
||
if language not in ['chinese', 'english']:
|
||
return jsonify({"error": "Unsupported language"}), 400
|
||
|
||
unique_id = str(uuid.uuid4())
|
||
timestamp = datetime.now().strftime('%Y%m%d%H%M%S')
|
||
os.makedirs('temp', exist_ok=True)
|
||
input_audio_filename = f"input_{timestamp}_{unique_id}.wav"
|
||
input_audio_path = os.path.join('temp', input_audio_filename)
|
||
request_audio.save(input_audio_path)
|
||
|
||
try:
|
||
# 使用WeNet，音频转文本
|
||
transcription = transcribe_audio(input_audio_path, language)
|
||
# 生成回答音频
|
||
response_audio = generate_response_audio(transcription, language, input_audio_path)
|
||
|
||
# 返回生成的音频文件
|
||
return send_file(response_audio, as_attachment=True, download_name='response.wav', mimetype='audio/wav')
|
||
except Exception as e:
|
||
return jsonify({"error": str(e)}), 500
|
||
finally:
|
||
if os.path.exists(input_audio_path):
|
||
os.remove(input_audio_path)
|
||
|
||
if __name__ == '__main__':
|
||
app.run()
|