Update main.py

main.py

@@ -78,7 +78,9 @@ async def home():
 
 @app.post("/fastapi_file_upload_process")
 #async def upload_file(user_question: str, file_to_process: FileToProcess = Depends()):
-async def pdf_file_qa_process(user_question: str, request: Request, file_to_process: FileToProcess = Depends()):    
+async def pdf_file_qa_process(user_question: str, request: Request, file_to_process: FileToProcess = Depends()):
+    print("API Call Triggered.")
+    start_0 = timeit.default_timer()
     uploaded_file = file_to_process.uploaded_file
     print("File received:"+uploaded_file.filename)
 
@@ -174,7 +176,9 @@ async def pdf_file_qa_process(user_question: str, request: Request, file_to_proc
     new_final_ai_response = final_ai_response.split('Unhelpful Answer:')[0].strip()
     new_final_ai_response = new_final_ai_response.split('Note:')[0].strip()
     new_final_ai_response = new_final_ai_response.split('Please provide feedback on how to improve the chatbot.')[0].strip() 
-    print(new_final_ai_response)  
-
+    print(new_final_ai_response) 
+    end_0 = timeit.default_timer()  
+    print("API Call Ended.")  
+    print(f'API Call共耗时： @ {end_0 - start_0}')
     return {"AIResponse": new_final_ai_response}
     #return JSONResponse({"AIResponse": new_final_ai_response})