Update app.py

app.py

@@ -253,7 +253,7 @@ if submit and len(x) != 0:
   for e in edccanbis:
     if e[-1]== ["B-EDC","B-CANCER"]:
       edccann += [[e[0],e[1],e[2]]]
-      st.write("am in edcann")
+      
       
     elif e[-1]== ["B-EDC","B-HORMONE"]:
       edchorm += [[e[0],e[1],e[2]]]
@@ -280,7 +280,19 @@ if submit and len(x) != 0:
   hormcancsv = hormcandf.to_csv('hormcan.csv')
   hormrecepcsv = hormrecepdf.to_csv('hormrecep.csv')
   
+  
+  
+  for e in edccanbis:
+    edccan += [[e[3],e[0]+" ["+e[-1][0][2:]+"]", e[1]+" ["+e[-1][1][2:]+"]",e[2]]]
+   
+  edccandf = pd.DataFrame(edccan, columns= ["Sentence", "Entity 1", "Entity 2", "Relation"] )
+  
+  edccandf.to_csv("table.csv")
+  
+  
   with zipfile.ZipFile("allcsvs.zip", "w") as zipf:
+    if len(edccan)!=0:
+      zipf.write('table.csv')
     if len(edccann)!=0:
       zipf.write('edccan.csv')
       
@@ -292,31 +304,31 @@ if submit and len(x) != 0:
       zipf.write('hormcan.csv')
     if len(hormrecep)!=0:
       zipf.write('hormrecep.csv')
+      
+      
   zipf.close()
   
-  for e in edccanbis:
-    edccan += [[e[3],e[0]+" ["+e[-1][0][2:]+"]", e[1]+" ["+e[-1][1][2:]+"]",e[2]]]
-   
-  edccandf = pd.DataFrame(edccan, columns= ["Sentence", "Entity 1", "Entity 2", "Relation"] )
-  
-  
   st.table(edccandf)
   csv = edccandf.to_csv(index=False).encode('utf-8')
-  st.download_button(
-       label="Download all data as CSV",
-       data=csv,
-       file_name='Relation_triples.csv',
-       mime='text/csv',
-  )
-  
-  with open("allcsvs.zip", "rb") as fp:
-    btn = st.download_button(
-        label="Download ZIP",
-        data=fp,
-        file_name="SeperateCsvs.zip",
-        mime="application/zip"
+  st.sidebar:
+    st.write("You can only choose one download!!")
+    st.write("we recommed ZIP file.")
+    st.write("Download table only:")
+    st.download_button(
+         label="Download CSV",
+         data=csv,
+         file_name='Relation_triples_table.csv',
+         mime='text/csv',
     )
-   
+    st.write("Download table plus separate csvs for each family of pairs:")
+    with open("allcsvs.zip", "rb") as fp:
+      btn = st.download_button(
+          label="Download ZIP",
+          data=fp,
+          file_name="SeparateCsvs.zip",
+          mime="application/zip"
+      )
+   )