Update app.py

app.py

@@ -85,10 +85,11 @@ def human_feedback(entities: Dict):
 # Generate JSON Output
 def generate_json_output(entities: Dict) -> str:
     output_path = "resume_entities.json"
+    json_data = json.dumps(entities, indent=4)  # Create JSON string from the entities dictionary
     with open(output_path, 'w') as f:
-        json.dump(entities, f, indent=4)
+        f.write(json_data)
     st.write(f"Output saved as {output_path}")
-    return output_path
+    return json_data  # Return the JSON data instead of file path
 
 # Main function to orchestrate workflow
 def main():
@@ -122,8 +123,10 @@ def main():
         st.write("Final entities:", entities)
         
         # Generate JSON output
-        json_path = generate_json_output(entities)
-        st.download_button("Download JSON Output", file_name=json_path)
+        json_data = generate_json_output(entities)  # Use JSON data instead of file path
+
+        # Pass the JSON data to st.download_button
+        st.download_button("Download JSON Output", data=json_data, file_name="resume_entities.json")
 
 # Streamlit app entry point
 if __name__ == "__main__":