feat(NoteForm): 增加文件上传状态反馈

backend/app/transcriber/groq.py

@@ -6,15 +6,31 @@ from app.models.transcriber_model import TranscriptResult, TranscriptSegment
 from app.services.provider import ProviderService
 from app.transcriber.base import Transcriber
 from openai import OpenAI
+import ffmpeg
+import tempfile
 from dotenv import load_dotenv
 load_dotenv()
+MAX_SIZE_MB = 18
+MAX_SIZE_BYTES = MAX_SIZE_MB * 1024 * 1024
+def compress_audio(input_path: str, target_bitrate='64k') -> str:
+    output_fd, output_path = tempfile.mkstemp(suffix=".mp3")  # 临时输出文件
+    os.close(output_fd)  # 关闭文件描述符，ffmpeg 会用路径操作
+    ffmpeg.input(input_path).output(output_path, audio_bitrate=target_bitrate).run(quiet=True, overwrite_output=True)
+    return output_path
 
 class GroqTranscriber(Transcriber, ABC):
 
 
     @timeit
     def transcript(self, file_path: str) -> TranscriptResult:
+        file_size = os.path.getsize(file_path)
+        if file_size > MAX_SIZE_BYTES:
+            print(f"文件超过 {MAX_SIZE_MB}MB，开始压缩（当前 {round(file_size / (1024 * 1024), 2)}MB）...")
+            file_path = compress_audio(file_path)
+            print(f"压缩完成，临时路径：{file_path}")
         provider = ProviderService.get_provider_by_id('groq')
+
+
         if not provider:
             raise Exception("Groq 供应商未配置,请配置以后使用。")
         client = OpenAI(