Update app.py

app.py

@@ -45,16 +45,18 @@ def main():
         try:
             # Process the image with the OCR pipeline
             ocr_result = image_pipeline(image)[0]['generated_text'].replace(" ", "").lower()
-            verified_url = st.text_input("Recognized URL", ocr_result)
+            # Store the verified URL in session state for access later
+            st.session_state['verified_url'] = st.text_input("Recognized URL", ocr_result)
         except Exception as e:
             st.error(f"An error occurred during image processing: {e}")
 
     if st.button('Detect Phishing'):
-        if 'verified_url' in st.session_state:
+        # Check for 'verified_url' in session state instead of local variable
+        if 'verified_url' in st.session_state and st.session_state['verified_url']:
             result = check_phishing(phishing_model, phishing_tokenizer, st.session_state['verified_url'])
             st.write(result)
         else:
-            st.write("Please upload an image to detect the URL and check for phishing.")
+            st.error("Please upload an image to detect the URL and check for phishing.")
 
 # Run the main function
 if __name__ == "__main__":