Update app.py

app.py

@@ -15,6 +15,19 @@ load_dotenv()
 hf_token = os.environ.get('HUGGINGFACEHUB_API_TOKEN')
 repo_id=os.environ.get('repo_id')
 
+print(f"定义处理多余的Context文本的函数")
+def remove_context(text):
+    # 检查 'Context:' 是否存在
+    if 'Context:' in text:
+        # 找到第一个 '\n\n' 的位置
+        end_of_context = text.find('\n\n')
+        # 删除 'Context:' 到第一个 '\n\n' 之间的部分
+        return text[end_of_context + 2:]  # '+2' 是为了跳过两个换行符
+    else:
+        # 如果 'Context:' 不存在，返回原始文本
+        return text
+print(f"处理多余的Context文本函数定义结束")    
+
 #port = os.getenv('port')
 
 #OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
@@ -45,27 +58,25 @@ 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(url)            
             loader = WebBaseLoader(url)
-            print("2")
             docs = loader.load()
-            print(docs)
             split_docs = text_splitter_rcs.split_documents(docs)
-            print("3")
-            print("split_docs")
             #result=chain.run(docs)   #这个result的格式比较特殊，可以直接print，但不可以和其他字符串联合print输出 - this step errors!
             result=chain.run(split_docs)
-            print("4")
-            print(url)
-            ai_response=str(result)       
-            
+            initial_ai_response=str(result) #找到之前总是POST Error的原因：chain.run(docs)的结果，格式不是str，导致程序错误
+            cleaned_initial_ai_response = remove_context(initial_ai_response)
+            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.")