Update main.py

main.py

@@ -1,3 +1,4 @@
+import time
 from fastapi import FastAPI, HTTPException, Query
 from pydantic import BaseModel
 import subprocess
@@ -188,10 +189,16 @@ 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()  # 记录请求接受的时间
     try:
         # 调用 SMILES 生成逻辑
-        print("123123")
+        print("/generate请求开始时间:", start_time)
         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  # 计算用时
+
+        print("/generate请求结束时间:", end_time)
+        print(f"请求处理用时: {duration:.2f}秒")
         return result
     except Exception as e:
         # 捕获异常并记录详细的错误信息，包括堆栈追踪