Update patent_app.py

patent_app.py

@@ -44,10 +44,12 @@ patent_num = st.selectbox("Select a patent based on its number",  val_set['paten
 if "button_clicked" not in st.session_state:    
     st.session_state.button_clicked = False
 
-button_1 = st.button('Get Data to predict!')
+# The button was clicked
+def callback():
+    st.session_state.button_clicked = True
 
 # Get the abstract and claims data to predict
-if patent_num and (button_1 or st.session_state.button_clicked):
+if patent_num and (st.button('Get Data to predict!', on_click = callback) or st.session_state.button_clicked):
   # Display the abstract and claims
   val_set = val_set.filter(lambda e: e['patent_number'] == patent_num)