download button

app.py

@@ -17,7 +17,8 @@ def save_data(data):
     print(file_path.split(".json")[0])
     with open(f"{file_path.split('.json')[0]}_graded.json", 'w') as file:
         json.dump(data, file, indent=4)
-
+def download_json(data):
+    return json.dumps(data, indent=4)
 data = load_data()
 
 for query in data:
@@ -59,7 +60,7 @@ def display_query():
 
     st.session_state.graded_queries = sum(query.get('status', None) is not None for query in st.session_state.data)
     print(f"Current Query Index: {st.session_state.current_query_index} | Graded Queries: {st.session_state.graded_queries} | Total Queries: {len(st.session_state.data)} | Current Query Status {current_query.get('status', None)}")
-    col1, col2 = st.columns([5, 1], gap="small")
+    col1, col2 = st.columns([4, 2], gap="small")
     with col1:
         if st.button('Previous'):
             if st.session_state.current_query_index > 0:
@@ -67,7 +68,7 @@ def display_query():
                 st.rerun()
         st.progress((st.session_state.current_query_index + 1) / len(st.session_state.data))
     with col2:
-        col1, col2, col3 = st.columns([1, 1, 1], gap = "small")
+        col1, col2, col3, col4 = st.columns([1, 1, 1, 1], gap = "small")
         with col1:
             if st.button('Next'):
                 if st.session_state.current_query_index < len(st.session_state.data) - 1:
@@ -81,6 +82,14 @@ def display_query():
             if st.button('Junk'):
                 current_query['status'] = 'nonsense'
                 next_query()
+        with col4:
+            # Example button for downloading data
+            st.download_button(
+                label="Download",
+                data=download_json(st.session_state.data),
+                file_name="graded_data.json",
+                mime="application/json"
+            )
 
         st.markdown(f"<p>At index {st.session_state.current_query_index + 1}. Graded Queries: {st.session_state.graded_queries}/{len(st.session_state.data)}</p>", unsafe_allow_html=True)