|
@@ -8,8 +8,8 @@ import requests
|
|
|
import uuid
|
|
|
import os
|
|
|
import uvicorn
|
|
|
-from asr_client import send_asr_request,send_task_request
|
|
|
-from database import insert,update,query,Task
|
|
|
+from asr_client import send_asr_request, send_task_request
|
|
|
+from database import insert, update, query, Task
|
|
|
from datetime import datetime
|
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
import hashlib
|
|
@@ -39,37 +39,43 @@ app.add_middleware(
|
|
|
allow_headers=["*"],
|
|
|
)
|
|
|
|
|
|
+
|
|
|
class QueryItem():
|
|
|
- md5:Optional[str]=None
|
|
|
- url:Optional[str]=None
|
|
|
+ md5: Optional[str] = None
|
|
|
+ url: Optional[str] = None
|
|
|
+
|
|
|
|
|
|
@app.get('/')
|
|
|
def index():
|
|
|
return {'message': '你已经正确创建 FastApi 服务!'}
|
|
|
|
|
|
+
|
|
|
@app.post('/asr/task/submit')
|
|
|
-def task_submit(md5:str,url:str):
|
|
|
+def task_submit(md5: str, url: str):
|
|
|
json = send_asr_request(url)
|
|
|
- task = Task(md5=md5,task_id=json.Data.TaskId,task_result=json.to_json_string(),task_status=1)
|
|
|
+ task = Task(md5=md5, task_id=json.Data.TaskId, task_result=json.to_json_string(), task_status=1)
|
|
|
insert(task)
|
|
|
- return {'code':0,'taskId':json.Data.TaskId}
|
|
|
+ return {'code': 0, 'taskId': json.Data.TaskId}
|
|
|
+
|
|
|
|
|
|
@app.post('/asr/task/result')
|
|
|
-def task_submit(task_id:int):
|
|
|
+def task_submit(task_id: int):
|
|
|
json = send_task_request(task_id)
|
|
|
- task = query(None,None,task_id)[0]
|
|
|
- task.task_status=json.Data.Status
|
|
|
+ task = query(None, None, task_id)[0]
|
|
|
+ task.task_status = json.Data.Status
|
|
|
task.task_result = json.to_json_string()
|
|
|
if json.Data.Status == 2:
|
|
|
task.word_text = json.Data.ResultDetail[0].FinalSentence
|
|
|
task.word_split = json.Data.ResultDetail[0].SliceSentence
|
|
|
update(task)
|
|
|
- return {'code':0,'status':json.Data.StatusStr}
|
|
|
+ return {'code': 0, 'status': json.Data.StatusStr}
|
|
|
+
|
|
|
|
|
|
@app.post('/asr/task/list')
|
|
|
-def task_submit(task_status:int):
|
|
|
- task = query(None,task_status,None)
|
|
|
- return {'code':0,'data':task}
|
|
|
+def task_submit(task_status: int):
|
|
|
+ task = query(None, task_status, None)
|
|
|
+ return {'code': 0, 'data': task}
|
|
|
+
|
|
|
|
|
|
# @app.post('/asr/task/result')
|
|
|
# def retry(item:Item):
|
|
@@ -97,7 +103,7 @@ def task_submit(task_status:int):
|
|
|
|
|
|
# return {'code':0,'data':{'taskId': task.id}}
|
|
|
|
|
|
-if __name__ == '__main__':
|
|
|
- uvicorn.run(app='main:app', host="0.0.0.0", port=8000, reload=True, debug=True)
|
|
|
|
|
|
-#gunicorn main:app -w 4 -k uvicorn.workers.UvicornWorker #线上启动命令
|
|
|
+if __name__ == '__main__':
|
|
|
+ uvicorn.run(app='main:app', host="0.0.0.0", port=31012, reload=True, debug=True)
|
|
|
+# gunicorn main:app -w 4 -k uvicorn.workers.UvicornWorker #线上启动命令
|