Update app.py

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 import requests
+import timeit
+import datetime
 
 st.set_page_config(page_title="AI Chat Client", layout="wide")
 st.title("AI Chat Client")
@@ -18,17 +20,26 @@ if st.button('Get AI Response'):
             # 构造请求
             data = {"url": url, "question": question}
             # 发送请求到 FastAPI 服务器
-            response = requests.post(fastapi_server_url, json=data)
+            current_datetime_0 = datetime.datetime.now()
+            print(f'API调用请求发送开始 @ {current_datetime}')             
+            start_1 = timeit.default_timer() # Start timer               
+            response = requests.post(fastapi_server_url, json=data)                       
+            end_1 = timeit.default_timer() # Start timer   
+            print(f'API调用请求发送结束，共耗时： @ {end_1 - start_1}')  
             
             if response.status_code == 200:
                 # 显示 AI 的回答
                 #ai_response = response.json().get("AI Response", "No response received.")
+                print(f'获取API调用结果开始 @ {current_datetime}')
+                start_2 = timeit.default_timer() # Start timer   
                 ai_response = response.json()
                 ai_response_output=ai_response['AIResponse']
+                end_2 = timeit.default_timer() # Start timer   
+                print(f'获取API调用结果完毕，共耗时： @ {end_2 - start_2}') 
                 st.write("AI Response:", ai_response)
                 st.write("AI Response:", ai_response_output)
             else:
                 # 显示错误信息
                 st.error("Error in fetching response from AI server.")
     else:
-        st.warning("Please enter both URL and a question.")
+        st.warning("Please enter both URL and a question.")