Update main.py

main.py

@@ -8,8 +8,9 @@ from generate_image import generate_image
 from fastapi.staticfiles import StaticFiles
 
 app = FastAPI()
+os.mkdir("tmp")
 # 挂载 tmp 文件夹下的静态文件
-app.mount("/static", StaticFiles(directory="/tmp"), name="static")
+app.mount("/static", StaticFiles(directory="tmp"), name="static")
 # 添加跨域支持
 origins = [
     "*"
@@ -31,7 +32,7 @@ class Markdown(BaseModel):
 @app.post("/generate_image")
 async def generate_image_endpoint(markdown: Markdown):
     # 生成随机文件名
-    image_filename = '/tmp/'+str(uuid.uuid4()) + ".png"
+    image_filename = 'tmp/'+str(uuid.uuid4()) + ".png"
     # 调用 generate_image 函数来生成图片
     generate_image(markdown.markdown, image_filename)
     # 返回图片的路径
@@ -41,8 +42,8 @@ async def generate_image_endpoint(markdown: Markdown):
 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)
+    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)