CMM_API/apis/main.py

from fastapi import FastAPI
from fastapi import Request
from fastapi import Form
import urllib
import sys
import os
import codecs
import html
from typing import Optional

from fastapi import FastAPI
from pydantic import BaseModel
import rpyc
import zhtest


class FName(BaseModel):
    fname: str


class TTS(BaseModel):
    txt: str
    num: str
#    description: Optional[str] = None
#    price: float
#    tax: Optional[float] = None

app = FastAPI()

@app.post("/tts/audio_to_video")
async def audio_to_video(item: FName):
    print(item.fname)
    fname=zhtest.to_video.delay(item.fname)
    print(fname)    
    return {"OK": "200"}



@app.post("/tts")
async def run_tts(item: TTS):
    print(item.txt)
    print(item.num)
#    import zhtest
    fname=zhtest.to_wav.delay(item.txt)
    print(fname)
    return {"OK": "200"}



@app.get("/gpu")
async def get_gpu():
    conn = rpyc.classic.connect("139.162.121.30", 12333)
    conn.execute('import os')
    ros = conn.modules.os
#    lst=ros.listdir('/var/wav')
    return {'ok':'200'}


@app.get("/tts/files")
async def get_tts_files():
    conn = rpyc.classic.connect("139.162.121.30", 12333)
    conn.execute('import os')
    ros = conn.modules.os
    lst=ros.listdir('/var/wav')
    return {'files':lst}


@app.get("/tts/videos")
async def get_video_files():
    conn = rpyc.classic.connect("139.162.121.30", 12333)
    conn.execute('import os')
    ros = conn.modules.os
    lst=ros.listdir('/var/video')
    newlst=[]
    for l in lst:
        newlst.append('http://139.162.121.30:54321/'+l)    
    return {'files':newlst}



@app.get("/tts/s1_videos")
async def s1_videos():
    conn = rpyc.classic.connect("139.162.121.30", 12333)
    conn.execute('import os')
    ros = conn.modules.os
    lst=ros.listdir('/var/video/s1')
    newlst=[]
    for l in lst:
        newlst.append('http://139.162.121.30:54321/s1/'+l)    
    return {'files':newlst}



@app.get("/")
async def root():
    return {"message": "Hello World"}