File size: 1,820 Bytes
7acc049 934814f 7acc049 e80241e 7acc049 5be5932 7acc049 e949e36 7acc049 934814f 7acc049 e949e36 7acc049 3e55f8d 7acc049 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
from fastapi import FastAPI, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
import os
import time
import uuid
from generate_image import generate_image
from fastapi.staticfiles import StaticFiles
app = FastAPI()
# 挂载 tmp 文件夹下的静态文件
app.mount("/tmp", StaticFiles(directory="tmp"), name="static")
host=os.environ.get("HOST", "https://99i-t2c.hf.space")
# 添加跨域支持
origins = [
"*"
]
app.add_middleware(
CORSMiddleware,
allow_origins=origins,
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# 创建一个模型，用于接收 markdown 字符串
class Markdown(BaseModel):
markdown: str
@app.get("/")
def hello():
return "hello world"
# 创建一个路由，用于接收 markdown 字符串并生成图片
@app.post("/t2c")
async def generate_image_endpoint(markdown: Markdown):
# 生成随机文件名
image_filename = 'tmp/'+str(uuid.uuid4()) + ".png"
# 调用 generate_image 函数来生成图片
generate_image(markdown.markdown, image_filename)
# 返回图片的路径
return {"url": f"{host}/{image_filename}"}
# 创建一个函数，用于清理过期的图片
def clean_expired_images():
expiration_time = int(os.environ.get("EXPIRATION_TIME", "3600")) # 默认过期时间为1小时
current_time = time.time()
for filename in os.listdir("tmp"):
file_path = os.path.join("tmp", filename)
if current_time - os.path.getmtime(file_path) > expiration_time:
os.remove(file_path)
# 创建一个路由，用于触发清理过期的图片
@app.get("/ci")
async def clean_images_endpoint():
# 调用 clean_expired_images 函数来清理过期的图片
clean_expired_images()
return {"message": "Images cleaned"}
|