main.py 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. from concurrent.futures import ThreadPoolExecutor
  2. import uvicorn
  3. from fastapi import FastAPI
  4. from fastapi.middleware.cors import CORSMiddleware
  5. from loguru import logger
  6. from routers import get_related_words
  7. from routers import script_config
  8. from routers import tengxunyun_server
  9. from script_score.lstm_network import SentimentNet
  10. from script_score.evaluate_script import router as evaluate_script_router
  11. logger.remove() # 删去 import logger之后自动产生的handler,不删除的话会出现重复输出的现象
  12. logger.add("/data/pythonProject/video_to_word/logs/main_server.{time:YYYY-MM-DD}.log",
  13. rotation="00:00",
  14. format="{time:YYYY-MM-DD HH:mm:ss,SSS} [{process}] [{thread}] {level} {file} {line} - {message}",
  15. level="INFO")
  16. threadPool = ThreadPoolExecutor(max_workers=4)
  17. app = FastAPI()
  18. origins = [
  19. "http://192.168.0.195:9001",
  20. "http://192.168.6.220:3000",
  21. "http://192.168.1.91:3000",
  22. "http://192.168.1.34",
  23. "http://192.168.1.34:8000",
  24. "http://192.168.1.105",
  25. "http://192.168.1.105:3000",
  26. "http://111.206.86.186",
  27. "http://111.206.86.186:3000",
  28. "http://adsp.tjyourong.com.cn",
  29. "http://adsp.tjyourong.com.cn:3000",
  30. "http://adsp.c-top.com.cn",
  31. "http://adsp.c-top.com.cn:3000"
  32. ]
  33. app.add_middleware(
  34. CORSMiddleware,
  35. allow_origins=origins,
  36. allow_credentials=True,
  37. allow_methods=["*"],
  38. allow_headers=["*"],
  39. )
  40. app.include_router(get_related_words.router)
  41. app.include_router(tengxunyun_server.router)
  42. app.include_router(script_config.router)
  43. app.include_router(evaluate_script_router)
  44. if __name__ == '__main__':
  45. uvicorn.run(app='main:app', host="0.0.0.0", port=31013, reload=True, debug=True)
  46. # gunicorn main:app -w 4 -k uvicorn.workers.UvicornWorker #线上启动命令