Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import requests
 import timeit
 import datetime
+from dotenv import load_dotenv
+load_dotenv()
 
 st.set_page_config(page_title="AI Chat Client", layout="wide")
 st.title("AI Chat Client")
@@ -10,8 +12,7 @@ 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"
+HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
 # 用户输入
 url = st.text_input("Enter the URL to chat with:")
@@ -28,13 +29,21 @@ if st.button('Get AI Response'):
         print()
         with st.spinner('Fetching AI response...'):
             # 构造请求
+            # FastAPI 服务器的 URL
+            fastapi_server_url = "https://binqiangliu-wechatarticleloaderfastapi.hf.space/get_ai_response"
+            headers = {
+                "Content-Type": "application/json",
+                "Authorization": f"Bearer {HUGGINGFACEHUB_API_TOKEN}"
+                #保险起见，建议始终采用f''的形式以及配合使用{}
+            }
             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)                       
+            #response = requests.post(fastapi_server_url, json=data)   
+            response = requests.post(fastapi_server_url, headers=headers, json=data)            
             end_1 = timeit.default_timer() # Start timer   
             print(f'API调用请求发送结束，共耗时： @ {end_1 - start_1}')  
             print()