File size: 2,444 Bytes
f236a9e 0c4b8aa f236a9e f8b49ee f236a9e facdb79 f236a9e f8b49ee f236a9e f8b49ee f236a9e f8b49ee f236a9e 0c4b8aa f8b49ee 0c4b8aa f8b49ee f236a9e b34f902 c06de5c f8b49ee 0c4b8aa b34f902 5e092ec 0c4b8aa f8b49ee f236a9e 78043f5 f236a9e 0c4b8aa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
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")
current_datetime_0= datetime.datetime.now()
print(f"Anything happens, this ST app will execute from top down. @ {current_datetime_0}")
print()
# FastAPI 服务器的 URL
fastapi_server_url = "https://binqiangliu-wechatarticleloaderfastapi.hf.space/get_ai_response"
# 用户输入
url = st.text_input("Enter the URL to chat with:")
print(f"URL Entered")
print()
question = st.text_input("Enter your question:")
print(f"Question Entered")
print()
# 当用户点击按钮时
if st.button('Get AI Response'):
if url and question:
print(f"URL & Question Entered & Button clicked.")
print()
with st.spinner('Fetching AI response...'):
# 构造请求
data = {"url": url, "question": question}
# 发送请求到 FastAPI 服务器
current_datetime_0 = datetime.datetime.now()
print(f'API调用请求发送开始 @ {current_datetime_0}')
print()
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}')
print()
if response.status_code == 200:
# 显示 AI 的回答
#ai_response = response.json().get("AI Response", "No response received.")
current_datetime_1 = datetime.datetime.now()
print(f'获取API调用结果开始 @ {current_datetime_1}')
print()
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}')
print()
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.") |