diff --git a/main.py b/main.py index 3745b4314849ecd5e0e7b92e1f7ff6bb0c3d8f26..1a0453beb5f69f6a3f94676c2e83675636fac565 100644 --- a/main.py +++ b/main.py @@ -337,7 +337,14 @@ def simplex(): if __name__ == '__main__': logger.info('Starting program.') - ford_fulkerson() - simplex() + logger.info('Choose an option:') + logger.info(' 1) Solve Maximum Flow via Ford Fulkerson.') + logger.info(' 2) Solve Maximum Flow via Simplex.') + + user_input = str(input()).strip() + if user_input == '1': + ford_fulkerson() + elif user_input == '2': + simplex() logger.info('Terminating program.')