Added download button

app.py

@@ -1,3 +1,4 @@
+from io import BytesIO
 import streamlit as st
 import base64
 from transformers import AutoModel, AutoTokenizer
@@ -40,7 +41,14 @@ def display_tree(output):
         """, unsafe_allow_html=True)
     
     #st.image('syntax_tree.' + format, use_column_width=True)
-        
+
+def display_download(disp_string):
+    to_download = BytesIO(disp_string.encode())
+    st.download_button(label="⬇️ Download text file", 
+                    data=to_download, 
+                    file_name="parsed_output.txt", 
+                    mime="text/plain")
+
 # Streamlit app title
 st.title('DictaBERT-Joint Visualizer')
 
@@ -80,6 +88,8 @@ if sentence:
             tree.append(dict(word=parts[1], dep_head_idx=int(parts[6]) - 1, dep_func=parts[7]))
         display_tree(tree)
 
+        display_download('\n'.join(ud_output))
+
         # Construct the table as a Markdown string
         table_md = "<div dir='rtl' style='text-align: right;'>\n\n"  # Start with RTL div
         
@@ -105,7 +115,10 @@ if sentence:
         tree = [w['syntax'] for w in output['tokens']]
         display_tree(tree)
         
+        json_output = json.dumps(output, ensure_ascii=False, indent=2)
+        display_download(json_output)
+
         # and the full json
-        st.markdown("```json\n" + json.dumps(output, ensure_ascii=False, indent=2) + "\n```")
+        st.markdown("```json\n" + json_output + "\n```")