Update app.py

app.py

@@ -46,28 +46,30 @@ if url !="" and not url.strip().isspace() and not url == "" and not url.strip()
     try:
         #loader = WebBaseLoader("https://www.usinoip.com/")
         with st.spinner("AI Thinking...Please wait a while to Cheers!"):
-            print("1")
+            print("Website to Chat: "+url)             
             loader = WebBaseLoader(url)
-            print("2")
             docs = loader.load()
-            print(docs)
             split_docs = text_splitter_rcs.split_documents(docs)
-            print("3")
             print(split_docs)
+            print()
             #result=chain.run(docs)   #这个result的格式比较特殊，可以直接print，但不可以和其他字符串联合print输出 - this step errors!
-            result=chain.run(split_docs)
-            print("4")
-            print(url)
-            ai_response=str(result)       
+            result=chain.run(split_docs)   #找到之前总是POST Error的原因：chain.run(docs)的结果，格式不是str，导致程序错误
+            result=str(result)  
+            cleaned_initial_ai_response = remove_context(result)
+            print(cleaned_initial_ai_response)
+            print()
+            final_ai_response = cleaned_initial_ai_response.split('<|end|>')[0].strip().replace('\n\n', '\n').replace('<|end|>', '').replace('<|user|>', '').replace('<|system|>', '').replace('<|assistant|>', '')
+            new_final_ai_response = final_ai_response.split('Unhelpful Answer:')[0].strip()
+            final_result = new_final_ai_response.split('Note:')[0].strip()   
             
             #print("AI Summarization: "+result)   #这个会出错，原因见上方
             print("AI Summarization:")
             #print(result)
-            print(ai_response)     
+            print(final_result)     
             
             st.write("AI Summarization:")
             #st.write(result)
-            st.write(ai_response)
+            st.write(final_result)
             
     except Exception as e:
         st.write("Wrong URL or URL not parsable.")