Update app.py

app.py

@@ -722,12 +722,22 @@ gbc = GradientBoostingClassifier(max_depth=4,learning_rate=0.7)
 gbc.fit(X_train,y_train)  
 
 import streamlit as st
+st.title("Phishing Website Detection")
+
+# User input for URL
 url = st.text_input("Enter the Url:", key="url_input")
 #can provide any URL. this URL was taken from PhishTank
-obj = FeatureExtraction(url)
-x = np.array(obj.getFeaturesList()).reshape(1,30) 
-y_pred =gbc.predict(x)[0]
-if y_pred==1:
-  print("We guess it is a safe website")
-else:
-  print("Caution! Suspicious website detected")
+
+# Predict and display the result
+if st.button("Check"):
+    if url:
+        obj = FeatureExtraction(url)
+        x = np.array(obj.getFeaturesList()).reshape(1, 30)
+        y_pred = gbc.predict(x)[0]
+        if y_pred == 1:
+            st.write("We guess it is a safe website")
+        else:
+            st.write("Caution! Suspicious website detected")
+        st.write(y_pred)
+    else:
+        st.write("Please enter a URL.")