Update app.py

app.py

@@ -1,52 +1,36 @@
-# app.py
-import logging
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import JSONResponse
-from predictor import Predictor
 import uvicorn
+from predictor import Predictor
 
-# 日志设置
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# 创建 FastAPI app
+# 创建 FastAPI 应用
 app = FastAPI()
 
-# 模型持有者（懒加载）
-model_holder = {"predictor": None}
+# 加载模型（启动时只加载一次）
+predictor = Predictor()
+
+@app.get("/")
+async def root():
+    return {"message": "API is running. Use POST /evaluate to upload files."}
 
 @app.post("/evaluate")
 async def evaluate_file(file: UploadFile = File(...)):
-    # 第一次请求时加载模型
-    if model_holder["predictor"] is None:
-        logger.info("模型实例为空，开始加载模型...")
-        try:
-            model_holder["predictor"] = Predictor()
-            logger.info("模型加载成功。")
-        except Exception as e:
-            logger.critical(f"模型加载失败: {e}", exc_info=True)
-            return JSONResponse(status_code=500, content={"error": f"Model loading failed: {str(e)}"})
-
-    predictor = model_holder["predictor"]
-
     try:
-        # 读取文件内容（假设是UTF-8文本）
-        text = (await file.read()).decode("utf-8")
-
-        if not text.strip():
-            return JSONResponse(status_code=400, content={"error": "File content is empty."})
+        # 读取上传文件内容
+        content = await file.read()
+        text = content.decode("utf-8", errors="ignore")
 
-        logger.info(f"收到文件 {file.filename}，开始处理...")
+        # 调用 predictor 进行推理
         result = predictor.predict(text)
-        logger.info(f"处理完成，返回结果: {result}")
-        return result
+
+        return JSONResponse(content=result)
 
     except Exception as e:
-        logger.error(f"处理文件时发生错误: {e}", exc_info=True)
-        return JSONResponse(status_code=500, content={"error": str(e)})
+        return JSONResponse(content={"error": str(e)}, status_code=500)
+
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run("app:app", host="0.0.0.0", port=7860)