Update main.py

main.py

@@ -6,7 +6,7 @@ from langchain.chains.summarize import load_summarize_chain
 #https://python.langchain.com/docs/use_cases/summarization
 from langchain import HuggingFaceHub
 from huggingface_hub import InferenceClient
-#from bs4 import BeautifulSoup
+from bs4 import BeautifulSoup
 import requests
 import sys
 
@@ -30,27 +30,24 @@ app = FastAPI()
 @app.post('/')
 async def home_api(request: Request):
     data = await request.json()
-    user_query = data['user_question']
-    print(user_query)
+    target_url = data['target_url']
+    print(target_url)
     return {"Message": "FastAPI Home API Deploy Success on HF"}
 
 @app.post('/api/chat')
 async def chat(request: Request):
     data = await request.json()
-    user_query = data['user_question']
-    print(user_query)
-    if user_query.strip() != "":
-        try:
-            loader = WebBaseLoader(user_query)
-            print(user_query)
-            docs = loader.load()
-            result = chain.run(docs)
-            print("AI Summarization: " + result)
-            #return {'response': result}  #FastAPI方式下，这个返回形式，有问题？
-            return JSONResponse({'response': result})
-        except Exception as e:
-            err_msg = "Wrong URL or URL not parsable."
-            print(err_msg)
-            raise HTTPException(status_code=400, detail=err_msg)
-    else:
-        raise HTTPException(status_code=400, detail="Invalid user question")
+    target_url = data['target_url']
+    print(target_url)
+    try:
+        loader = WebBaseLoader(target_url)
+        print(target_url)
+        docs = loader.load()
+        result = chain.run(docs)
+        print("AI Summarization: " + result)
+        #return {'response': result}  #FastAPI方式下，这个返回形式，有问题？
+        return JSONResponse({'response': result})
+    except Exception as e:
+        err_msg = "Wrong URL or URL not parsable."
+        print(err_msg)
+        raise HTTPException(status_code=400, detail=err_msg)