Spaces:
Running
Running
Commit
·
63f1335
1
Parent(s):
d840e96
corrected dowload behaviour
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ import numpy as np
|
|
| 4 |
import streamlit as st
|
| 5 |
import torch
|
| 6 |
import json
|
|
|
|
| 7 |
|
| 8 |
from doctr.io import DocumentFile
|
| 9 |
from doctr.utils.visualization import visualize_page
|
|
@@ -116,13 +117,16 @@ def main(det_archs, reco_archs):
|
|
| 116 |
print('\n')
|
| 117 |
|
| 118 |
# Display Text
|
| 119 |
-
st.markdown("\n
|
| 120 |
st.write(all_text)
|
| 121 |
|
| 122 |
# Display JSON
|
| 123 |
-
json_string = json.dumps(page_export)
|
| 124 |
-
st.markdown("\n
|
| 125 |
-
|
|
|
|
|
|
|
|
|
|
| 126 |
st.json(page_export, expanded=False)
|
| 127 |
|
| 128 |
|
|
|
|
| 4 |
import streamlit as st
|
| 5 |
import torch
|
| 6 |
import json
|
| 7 |
+
import base64
|
| 8 |
|
| 9 |
from doctr.io import DocumentFile
|
| 10 |
from doctr.utils.visualization import visualize_page
|
|
|
|
| 117 |
print('\n')
|
| 118 |
|
| 119 |
# Display Text
|
| 120 |
+
st.markdown("\n### **Here is your text:**")
|
| 121 |
st.write(all_text)
|
| 122 |
|
| 123 |
# Display JSON
|
| 124 |
+
# json_string = json.dumps(page_export)
|
| 125 |
+
st.markdown("\n### **Here is your document structure in JSON format:**")
|
| 126 |
+
encoded_data = base64.b64encode(json.dumps(page_export).encode("utf-8")).decode("utf-8")
|
| 127 |
+
download_link = f"data:application/json;base64,{encoded_data}"
|
| 128 |
+
st.markdown(f"[Download JSON]( {download_link} )")
|
| 129 |
+
# st.download_button(label="Download JSON", data=json_string, file_name='data.json', mime='application/json')
|
| 130 |
st.json(page_export, expanded=False)
|
| 131 |
|
| 132 |
|