Update app.py

app.py

@@ -17,11 +17,6 @@ HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
 # 用户输入
 url = st.text_input("Enter the URL to chat with:")
-# 检查 URL 是否以 "https://mp.weixin.qq.com" 开始
-if not url.startswith("https://mp.weixin.qq.com"):
-    # 如果不是，则显示错误信息并终止程序
-    st.error("URL is incorrect. Please enter a URL that starts with 'https://mp.weixin.qq.com'.")
-    st.stop()
 print(f"URL Entered: "+url)
 print()
 question = st.text_input("Enter your question:")
@@ -30,6 +25,11 @@ print()
 
 # 当用户点击按钮时
 if st.button('Get AI Response'):
+    # 检查 URL 是否以 "https://mp.weixin.qq.com" 开始
+    if not url.startswith("https://mp.weixin.qq.com"):
+        # 如果不是，则显示错误信息并终止程序
+        st.write("URL is incorrect. Please enter a URL that starts with 'https://mp.weixin.qq.com'.")
+        st.stop()
     if url and question:
         print(f"URL & Question Entered & Button clicked.")
         print()