Update main.py

main.py

@@ -1,5 +1,3 @@
-import os
-from dotenv import load_dotenv
 from fastapi import FastAPI, Request, HTTPException
 from langchain.document_loaders import WebBaseLoader
 from langchain.chains.summarize import load_summarize_chain
@@ -9,8 +7,10 @@ from huggingface_hub import InferenceClient
 from bs4 import BeautifulSoup
 import requests
 import sys
-
+import os
+from dotenv import load_dotenv
 load_dotenv()
+
 hf_token = os.environ.get('HUGGINGFACEHUB_API_TOKEN')
 repo_id=os.environ.get('repo_id')
 #port = os.getenv('port')
@@ -26,6 +26,8 @@ llm = HuggingFaceHub(repo_id=repo_id,  # for StarChat
 chain = load_summarize_chain(llm, chain_type="refine")
 
 app = FastAPI()
+class ChatRequest(BaseModel):
+    target_url: str
 
 @app.post('/')
 async def home_api(request: Request):
@@ -35,9 +37,15 @@ async def home_api(request: Request):
     return {"Message": "FastAPI Home API Deploy Success on HF"}
 
 @app.post('/api/chat')
-async def chat(request: Request):
-    data = await request.json()
-    target_url = data['target_url']
+async def chat(request: Request, chat_request: ChatRequest):
+    target_url = chat_request.target_url
+
+#@app.post('/api/chat')
+#async def chat(request: Request):
+#    data = await request.json()
+#    target_url = data['target_url']
+#以上四行代码，可能有问题？出在data = await request.json()？？？理论上似乎也没有问题，因为后面的print(target_url)，能够成功打印
+    
     print(target_url)
     try:
         loader = WebBaseLoader(target_url)