Update app.py

app.py

@@ -22,7 +22,18 @@ def load_LLR(uniprot_id):
   with ZipFile(LLR_FILE) as myzip:
     data = myzip.open(myzip.namelist()[0]+uniprot_id+'_LLR.csv')
   return pd.read_csv(data,index_col=0)
-
+  
+def meltLLR(LLR,gene_prefix=None,ignore_pos=False):
+  vars = LLR.melt(ignore_index=False)
+  vars['variant'] = [''.join(i.split(' '))+j for i,j in zip(vars['variable'],vars.index)]
+  vars['score'] = vars['value']
+  vars = vars.set_index('variant')
+  if not ignore_pos:
+    vars['pos'] = [int(i[1:-1]) for i in vars.index]
+  del vars['variable'],vars['value']
+  if gene_prefix is not None:
+    vars.index=gene_prefix+'_'+vars.index
+  return vars
 
 def plot_interactive(uniprot_id, show_clinvar=False):
   primaryLLR = load_LLR(uniprot_id)
@@ -105,4 +116,11 @@ show_clinvar = st.checkbox('show ClinVar annotations (red: pathogenic, green: be
 
 fig = plot_interactive(uid,show_clinvar=show_clinvar)
 fig.update_layout(width = 800, height = 600, autosize = False)
-st.plotly_chart(fig, use_container_width=True)
+st.plotly_chart(fig, use_container_width=True)
+
+st.download_button(
+     label="Download data as CSV",
+     data=meltLLR(load_LLR(uid)),
+     file_name=selection+'.csv',
+     mime='text/csv',
+ )