Merge pull request #151 from PolicyEngine/nikhilwoodruff/issue150

Simulation API data files are being corrupted after download

Author: anth-volk

libs/policyengine-simulation-api/src/policyengine_api/simulation_api/simulation.py

@@ -6,6 +6,7 @@
 from policyengine.outputs.macro.comparison.calculate_economy_comparison import (
     EconomyComparison,
 )
+from pathlib import Path
 
 
 def create_router():
@@ -16,9 +17,18 @@ def create_router():
     )
     async def simulate(parameters: SimulationOptions) -> EconomyComparison:
         model = SimulationOptions.model_validate(parameters)
-
+        print("Initialising")
         simulation = Simulation(**model.model_dump())
+        print("Calculating")
+        result = simulation.calculate_economy_comparison()
+        # Clear data files
+
+        for file in Path(".").glob("*.csv"):
+            file.unlink()
+
+        for file in Path(".").glob("*.h5"):
+            file.unlink()
 
-        return simulation.calculate_economy_comparison()
+        return result
 
     return router