main.py 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. from fastapi import FastAPI
  2. from fastapi import Request
  3. from fastapi import Form
  4. import urllib
  5. import sys
  6. import os
  7. import codecs
  8. import html
  9. from typing import Optional
  10. from fastapi import FastAPI
  11. from pydantic import BaseModel
  12. import rpyc
  13. import zhtest
  14. class FName(BaseModel):
  15. fname: str
  16. class TTS(BaseModel):
  17. txt: str
  18. num: str
  19. # description: Optional[str] = None
  20. # price: float
  21. # tax: Optional[float] = None
  22. app = FastAPI()
  23. @app.post("/tts/audio_to_video")
  24. async def audio_to_video(item: FName):
  25. print(item.fname)
  26. fname=zhtest.to_video.delay(item.fname)
  27. print(fname)
  28. return {"OK": "200"}
  29. @app.post("/tts")
  30. async def run_tts(item: TTS):
  31. print(item.txt)
  32. print(item.num)
  33. # import zhtest
  34. fname=zhtest.to_wav.delay(item.txt)
  35. print(fname)
  36. return {"OK": "200"}
  37. @app.get("/tts/files")
  38. async def get_tts_files():
  39. conn = rpyc.classic.connect("139.162.121.30", 12333)
  40. conn.execute('import os')
  41. ros = conn.modules.os
  42. lst=ros.listdir('/var/wav')
  43. return {'files':lst}
  44. @app.get("/tts/videos")
  45. async def get_video_files():
  46. conn = rpyc.classic.connect("139.162.121.30", 12333)
  47. conn.execute('import os')
  48. ros = conn.modules.os
  49. lst=ros.listdir('/var/video')
  50. return {'files':lst}
  51. @app.get("/")
  52. async def root():
  53. return {"message": "Hello World"}