Update app.py

app.py

@@ -220,7 +220,6 @@ def app_main(app_mode,example_files,model_name_files,clus_types):
 
   try:
       
-      
       with st.form('twc_form'):
 
         step1_line = "Upload text file(one sentence in a line) or choose an example text file below"
@@ -245,14 +244,6 @@ def app_main(app_mode,example_files,model_name_files,clus_types):
                     options = list(dict.keys(cluster_types)), index=0,  key = "twc_cluster_types")
         st.write("")
         submit_button = st.form_submit_button('Run')
-        st.download_button(
-         label="Download results as json",
-         data= st.session_state["download_ready"] if st.session_state["download_ready"] != None else "",
-         disabled = False if st.session_state["download_ready"] != None else True,
-         file_name= (st.session_state["model_name"] + "_" +  str(st.session_state["threshold"]) + "_" + st.session_state["overlapped"] + "_" +  '_'.join(st.session_state["file_name"].split(".")[:-1]) + ".json").replace("/","_"),
-         mime='text/json',
-         key ="download" 
-        )
 
         
         input_status_area = st.empty()
@@ -288,13 +279,35 @@ def app_main(app_mode,example_files,model_name_files,clus_types):
                         st.info("Custom model overrides model selection in step 2 above. So please clear the custom model text box to choose models from step 2")
                     display_results(sentences,results,response_info,app_mode,run_model)
                     #st.json(results)
-
+        
+    if submit_button:
+        st.download_button(
+            label="Download results as JSON",
+            data=st.session_state["download_ready"] if st.session_state["download_ready"] is not None else "",
+            disabled=not st.session_state["download_ready"],
+            file_name=(st.session_state["model_name"] + "_" + str(st.session_state["threshold"]) + "_" +
+                       st.session_state["overlapped"] + "_" + '_'.join(st.session_state["file_name"].split(".")[:-1]) +
+                       ".json").replace("/", "_"),
+            mime='text/json',
+            key="download"
+    )
       
       
 
   except Exception as e:
     st.error("Some error occurred during loading" + str(e))
-    st.stop()  
+    if submit_button:
+        st.download_button(
+            label="Download results as JSON",
+            data=st.session_state["download_ready"] if st.session_state["download_ready"] is not None else "",
+            disabled=not st.session_state["download_ready"],
+            file_name=(st.session_state["model_name"] + "_" + str(st.session_state["threshold"]) + "_" +
+                       st.session_state["overlapped"] + "_" + '_'.join(st.session_state["file_name"].split(".")[:-1]) +
+                       ".json").replace("/", "_"),
+            mime='text/json',
+            key="download"
+    )
+    #st.stop()  
 	
   st.markdown(markdown_str, unsafe_allow_html=True)