Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import requests
+import json
 
 st.title("PDF Extraction App")
 st.write("Upload a PDF file to extract correct answers and options using the backend service.")
@@ -25,6 +26,15 @@ if uploaded_file is not None:
                     result = response.json()
                     st.success("Extraction successful!")
                     st.json(result)
+                    
+                    # Create a download button for the JSON result
+                    json_data = json.dumps(result, indent=2)
+                    st.download_button(
+                        label="Download JSON",
+                        data=json_data,
+                        file_name="extraction_result.json",
+                        mime="application/json"
+                    )
                 else:
                     st.error(f"Error: {response.text}")
             except Exception as e: