JhonSmith0x7b commited on
Commit
557137d
·
1 Parent(s): add5a10
Files changed (1) hide show
  1. app/main.py +7 -5
app/main.py CHANGED
@@ -9,6 +9,10 @@ from typing import List, Optional
9
  import io
10
  import os
11
  import soundfile as sf
 
 
 
 
12
 
13
 
14
  app = FastAPI()
@@ -38,14 +42,14 @@ tts_helper = TTSHelper()
38
  @app.middleware("http")
39
  async def log_request_data(request: Request, call_next):
40
  # 打印请求方法和路径
41
- print(f"Request: {request.method} {request.url}")
42
 
43
  # # 打印请求头
44
  # print(f"Headers: {request.headers}")
45
 
46
  # 打印请求体(如果有)
47
  body = await request.body()
48
- print(f"Body: {body.decode('utf-8') if body else 'No Body'}")
49
 
50
  # 调用下一个中间件或路由
51
  response = await call_next(request)
@@ -62,9 +66,8 @@ def generate_audio(text: str = Form(...), background_tasks: BackgroundTasks = No
62
  duration = len(audio.samples) / audio.sample_rate
63
  elapsed_seconds = end - start
64
  rtf = elapsed_seconds / duration
65
- print(
66
  f"RTF: {rtf:.2f}, Duration: {duration:.2f} seconds, Elapsed: {elapsed_seconds:.2f} seconds")
67
- # filename = f"{uuid.uuid4()}.wav"
68
  audio_buffer = io.BytesIO()
69
  sf.write(
70
  audio_buffer,
@@ -74,7 +77,6 @@ def generate_audio(text: str = Form(...), background_tasks: BackgroundTasks = No
74
  subtype="PCM_16",
75
  )
76
  audio_buffer.seek(0)
77
- # background_tasks.add_task(os.remove, filename)
78
  response = responses.StreamingResponse(
79
  audio_buffer, media_type="audio/wav")
80
  return response
 
9
  import io
10
  import os
11
  import soundfile as sf
12
+ import logging
13
+
14
+ logging.basicConfig(level=logging.INFO)
15
+ LOGGER = logging.getLogger()
16
 
17
 
18
  app = FastAPI()
 
42
  @app.middleware("http")
43
  async def log_request_data(request: Request, call_next):
44
  # 打印请求方法和路径
45
+ logging.info(f"Request: {request.method} {request.url}")
46
 
47
  # # 打印请求头
48
  # print(f"Headers: {request.headers}")
49
 
50
  # 打印请求体(如果有)
51
  body = await request.body()
52
+ logging.info(f"Body: {body.decode('utf-8') if body else 'No Body'}")
53
 
54
  # 调用下一个中间件或路由
55
  response = await call_next(request)
 
66
  duration = len(audio.samples) / audio.sample_rate
67
  elapsed_seconds = end - start
68
  rtf = elapsed_seconds / duration
69
+ logging.info(
70
  f"RTF: {rtf:.2f}, Duration: {duration:.2f} seconds, Elapsed: {elapsed_seconds:.2f} seconds")
 
71
  audio_buffer = io.BytesIO()
72
  sf.write(
73
  audio_buffer,
 
77
  subtype="PCM_16",
78
  )
79
  audio_buffer.seek(0)
 
80
  response = responses.StreamingResponse(
81
  audio_buffer, media_type="audio/wav")
82
  return response