添加模型对外API

api/service/main.py

@@ -0,0 +1,73 @@
+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()

api/service/run_service.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+export FLASK_APP=main.py
+export FLASK_ENV=development
+flask run \
+    -h 0.0.0.0 \
+    -p 9990