Spaces:
Runtime error
Runtime error
File size: 3,544 Bytes
81c4606 b6eb6e4 a2dcf49 81c4606 0e2022b 81c4606 0e2022b 81c4606 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
from flask import Flask, request, jsonify, send_file
import requests
import os
import uuid
from pathlib import Path
import threading
import time
from pydub import AudioSegment
def audio_overlay(audio1, audio2, save_filename):
audio_raw1 = AudioSegment.from_file(audio1)
audio_raw2 = AudioSegment.from_file(audio2)
min_length = min(len(audio_raw1), len(audio_raw2))
audio_raw1 = audio_raw1[:min_length]
audio_raw2 = audio_raw2[:min_length]
overlayed_audio = audio_raw1.overlay(audio_raw2)
save_path = os.path.join("/tmp", save_filename)
overlayed_audio.export(save_path, format="mp3")
# 任务队列和状态存储
tasks = {}
running_threads = 0
condition = threading.Condition()
def process_audio_overlay(task_id, audio_url1, audio_url2):
global running_threads
with condition:
while running_threads >= 1:
tasks[task_id] = {"status": "queue"}
condition.wait()
running_threads += 1
tasks[task_id] = {"status": "processing"}
try:
# 下载音频文件到本地
response1 = requests.get(audio_url1)
if response1.status_code != 200:
raise Exception("无效的 URL")
audio_filename1 = audio_url1.split('/')[-1]
with open("/tmp/" + audio_filename1, 'wb') as f:
f.write(response1.content)
response2 = requests.get(audio_url2)
if response2.status_code != 200:
raise Exception("无效的 URL")
audio_filename2 = audio_url2.split('/')[-1]
with open("/tmp/" + audio_filename2, 'wb') as f:
f.write(response2.content)
# 执行音频合并操作
audio = audio_overlay("/tmp/" + audio_filename1, "/tmp/" + audio_filename2, os.path.join("/tmp", audio_filename1 + "_with_" + audio_filename2 + "_overlayed.mp3"))
# 提供文件的永久直链
result_audio_url = "/download/" + audio_filename1 + "_with_" + audio_filename2 + "_overlayed.mp3"
# 更新任务状态
tasks[task_id] = {
"status": "completed",
"result_url": result_audio_url
}
except Exception as e:
tasks[task_id] = {
"status": "error",
"message": str(e)
}
with condition:
running_threads -= 1
condition.notify_all()
app = Flask(__name__)
@app.route('/', methods=['GET'])
def hello():
return "Hello! This is an api server, and it is running successfully. For usage, please contact the person who hosted this api server."
@app.route('/api/audio_overlay', methods=['GET'])
def audio_separation():
audio_url1 = request.args.get('url1')
if not audio_url1:
return jsonify({"error": "URL1 parameter is required"}), 400
audio_url2 = request.args.get('url2')
if not audio_url2:
return jsonify({"error": "URL2 parameter is required"}), 400
task_id = str(uuid.uuid4())
tasks[task_id] = {"status": "processing"}
# 异步执行任务
threading.Thread(target=process_audio_overlay, args=(task_id, audio_url1, audio_url2)).start()
return jsonify({"task_id": task_id}), 202
@app.route('/api/tasks/<task_id>', methods=['GET'])
def get_task_status(task_id):
task = tasks.get(task_id)
if task:
return jsonify(task)
else:
return jsonify({"error": "Task not found"}), 404
@app.route('/download/<filename>', methods=['GET'])
def download(filename):
return send_file("/tmp/" + filename, as_attachment=True)
if __name__ == '__main__':
app.run(debug=False) |