added main

app.py

 import warnings
-
+import sys
 import numpy as np
 import pandas as pd
 from loguru import logger
@@ -253,18 +253,24 @@ def start_prediction(raw_path, viscosity_path, index_no, raw_skip_rows, viscosit
 
 if __name__ == "__main__":
     try:
-        logger.info("Starting the model")
-        index_number = 1250
-        raw_file_path = 'FY676-A-WO_Visc.xlsx'
-        raw_file_skip_rows = 0
-        viscosity_file_path = 'viscosity_natural_rubber_data.xlsx'
-        viscosity_file_skip_rows = 3
-        start_prediction(raw_file_path, viscosity_file_path, index_number, raw_file_skip_rows, viscosity_file_skip_rows)
-        index_number = 3294
-        raw_file_path = 'fy664g_raw.csv'
-        raw_file_skip_rows = 0
-        viscosity_file_path = 'fy664g-viscosity.xlsx'
-        viscosity_file_skip_rows = 2
+        # logger.info("Starting the model")
+        # index_number = 1250
+        # raw_file_path = 'FY676-A-WO_Visc.xlsx'
+        # raw_file_skip_rows = 0
+        # viscosity_file_path = 'viscosity_natural_rubber_data.xlsx'
+        # viscosity_file_skip_rows = 3
+        # start_prediction(raw_file_path, viscosity_file_path, index_number, raw_file_skip_rows, viscosity_file_skip_rows)
+        # index_number = 3294
+        # raw_file_path = 'fy664g_raw.csv'
+        # raw_file_skip_rows = 0
+        # viscosity_file_path = 'fy664g-viscosity.xlsx'
+        # viscosity_file_skip_rows = 2
+        # start_prediction(raw_file_path, viscosity_file_path, index_number, raw_file_skip_rows, viscosity_file_skip_rows)
+        raw_file_path = sys.argv[sys.argv.index('-r') + 1]
+        viscosity_file_path = sys.argv[sys.argv.index('-v') + 1]
+        raw_file_skip_rows = int(sys.argv[sys.argv.index('-sr') + 1])
+        viscosity_file_skip_rows = int(sys.argv[sys.argv.index('-sv') + 1])
+        index_number = int(sys.argv[sys.argv.index('-index') + 1])
         start_prediction(raw_file_path, viscosity_file_path, index_number, raw_file_skip_rows, viscosity_file_skip_rows)
     except Exception as e:
         logger.exception(f"Module failed because of error {e}")