From da33d2064c00b355a933f52a200020af813a4527 Mon Sep 17 00:00:00 2001
From: Brandon Rodriguez <brodriguez8774@gmail.com>
Date: Thu, 5 Dec 2019 13:27:38 -0500
Subject: [PATCH] Add ability for user to select algorithm

---
 main.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 3745b43..1a0453b 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.')
-- 
GitLab