modified return value

main.py

@@ -49,7 +49,7 @@ async def ocr(file: UploadFile = File(...)):
     return out_df.to_dict(orient='records')
 
 TMP_DIR = Path(__file__).parent / "tmp_uploads"
-# TMP_DIR.mkdir(parents=True, exist_ok=True)
+TMP_DIR.mkdir(parents=True, exist_ok=True)
 @app.post("/ocr_table")    
 async def ocr_table(file: UploadFile = File(...)):
     # ------------------- ① 参数校验 -------------------
@@ -89,15 +89,15 @@ async def ocr_table(file: UploadFile = File(...)):
         table_engine = RapidTable(input_args)
         table_results = table_engine(tmp_path)
 
-        #print(table_results.pred_htmls)
-        return table_results
+        print(table_results.pred_htmls)
+        return {"pred_htmls": table_results.pred_htmls, "cell_bboxes": table_results.cell_bboxes}
     finally:
         # ------------------- ⑥ 清理临时文件 -------------------
         # 为了防止磁盘被塞满，尽量在请求结束后删除文件
-        # try:
-        #     if tmp_path.exists():
-        #         tmp_path.unlink()
+        try:
+            if tmp_path.exists():
+                tmp_path.unlink()
                 print(f"已删除临时文件 {tmp_path}")
-        # except Exception as exc:
-        #     print(f"删除临时文件 {tmp_path} 失败: {exc}")
+        except Exception as exc:
+            print(f"删除临时文件 {tmp_path} 失败: {exc}")
     return None