Update app.py

app.py

@@ -82,11 +82,12 @@ def process_p_file(uploaded_file, percentage_idx, complexity_idx):
         model_repo_url = "https://huggingface.co/sadjadalikhani/LWM"
         model_repo_dir = "./LWM"
 
+        # Step 1: Clone the repository if not already done
         if not os.path.exists(model_repo_dir):
             print(f"Cloning model repository from {model_repo_url}...")
             subprocess.run(["git", "clone", model_repo_url, model_repo_dir], check=True)
 
-        ## Debugging: Check if the directory exists and print contents
+        # Step 2: Verify the repository was cloned
         if os.path.exists(model_repo_dir):
             os.chdir(model_repo_dir)
             print(f"Changed working directory to {os.getcwd()}")
@@ -95,30 +96,40 @@ def process_p_file(uploaded_file, percentage_idx, complexity_idx):
             print(f"Directory {model_repo_dir} does not exist.")
             return
 
-        # Step 2: Add the cloned repo to sys.path for imports
+        # Step 3: Add the cloned repo to sys.path for imports
         if model_repo_dir not in sys.path:
             sys.path.append(model_repo_dir)
-        
-        # Debugging: Print sys.path to ensure the cloned repo is in the path
+
+        # Step 4: Debugging - Print sys.path to ensure the cloned repo is in the path
         print(f"sys.path: {sys.path}")
-        #sys.path.append('./')
-        
+
+        # Ensure the 'LWM' directory is in the path
         lwm_model_dir = os.path.join(os.getcwd(), 'LWM')
         if lwm_model_dir not in sys.path:
             sys.path.append(lwm_model_dir)
-            
+
+        # Step 5: Verify if lwm_model.py exists in the directory
+        lwm_model_path = os.path.join(lwm_model_dir, 'lwm_model.py')
+        if not os.path.exists(lwm_model_path):
+            print(f"Error: lwm_model.py not found at {lwm_model_path}")
+            return f"Error: lwm_model.py not found at {lwm_model_path}"
+        else:
+            print(f"lwm_model.py found at {lwm_model_path}")
+
+        # Step 6: Import lwm_model after ensuring path is set
         from lwm_model import LWM
         device = 'cpu'
         print(f"Loading the LWM model on {device}...")
         model = LWM.from_pretrained(device=device)
 
+        # Step 7: Import tokenizer and load data
         from input_preprocess import tokenizer
-
         with open(uploaded_file.name, 'rb') as f:
             manual_data = pickle.load(f)
 
         preprocessed_chs = tokenizer(manual_data=manual_data)
 
+        # Step 8: Perform inference
         from inference import lwm_inference, create_raw_dataset
         output_emb = lwm_inference(preprocessed_chs, 'channel_emb', model)
         output_raw = create_raw_dataset(preprocessed_chs, device)