diff --git a/src/bot.py b/src/bot.py index f484890..e643e3e 100644 --- a/src/bot.py +++ b/src/bot.py @@ -51,7 +51,10 @@ def process_audio(message: Message): # Draw results with draw_result(str(file), result) as buf: f, m, o = get_result_percentages(result) - msg = f"分析结果: {f*100:.0f}% 🙋‍♀️ | {m*100:.0f}% 🙋‍♂️ | {o*100:.0f}% 🚫" + msg = f"分析结果: {f*100:.0f}% 🙋‍♀️ | {m*100:.0f}% 🙋‍♂️ | {o*100:.0f}% 🚫\n" \ + f"(结果仅供参考, 如果结果不是你想要的,那就是模型的问题,欢迎反馈)\n" \ + f"" \ + f"(因为这个模型基于法语数据, 和中文发音习惯有差异, 所以这个识别结果可能不准)" bot.send_photo(message.chat_id, photo=buf, caption=msg, reply_to_message_id=message.message_id) diff --git a/src/ina_main.py b/src/ina_main.py index da3c1bf..331d355 100644 --- a/src/ina_main.py +++ b/src/ina_main.py @@ -7,6 +7,8 @@ import sys import tempfile import time import wave +from dataclasses import dataclass + from PIL import Image from subprocess import Popen, PIPE from typing import NamedTuple, Callable @@ -94,8 +96,6 @@ def to_wav(file: str, callback: Callable, start_sec: float = 0, stop_sec: float output, error = p.communicate() assert p.returncode == 0, error - shutil.copy2(tmp_wav, './test_leohearts.wav') - return callback(tmp_wav) @@ -190,6 +190,6 @@ def get_result_percentages(result: Result) -> tuple[float, float, float]: # # print(process(seg, ['../test.mp3'])) if __name__ == '__main__': - to_wav('../audio_tmp/2021-12-22 05-32 leph1art5.mp3', print) + # to_wav('../audio_tmp/2021-12-22 05-32 leph1art5.mp3', print) # test() pass