Update main.py

main.py

@@ -1,4 +1,4 @@
-# import time
+import time
 from fastapi import FastAPI, HTTPException, Query
 from pydantic import BaseModel
 import subprocess
@@ -189,17 +189,17 @@ async def fragmentize(smiles: str = Query(..., description="SMILES string of the
 
 @app.post("/generate", response_model=List[MoleculeOutput])
 async def generate_molecules(request: GenerateRequest):
-    # start_time = time.time()  # 记录请求接受的时间
+    start_time = time.time()  # 记录请求接受的时间
     try:
         # 调用 SMILES 生成逻辑
-        # print("/generate请求开始时间:", start_time)
+        print("/generate请求开始时间:", start_time)
         print("--------------/generate start---------------")
         result = run_generate_runner(request.constSmiles, request.varSmiles, request.mainCls, request.minorCls, request.deltaValue, request.num)
-        # end_time = time.time()  # 记录生成结束的时间
-        # duration = end_time - start_time  # 计算用时
+        end_time = time.time()  # 记录生成结束的时间
+        duration = end_time - start_time  # 计算用时
 
-        # print("/generate请求结束时间:", end_time)
-        # print(f"请求处理用时: {duration:.2f}秒，本次处理分子数量为 {request.num}")
+        print("/generate请求结束时间:", end_time)
+        print(f"请求处理用时: {duration:.2f}秒，本次处理分子数量为 {request.num}")
         return result
     except Exception as e:
         # 捕获异常并记录详细的错误信息，包括堆栈追踪