Update app.py

app.py

@@ -112,8 +112,10 @@ def plot_confusion_matrix(y_true, y_pred, title):
     plt.savefig(f"{title}.png")
     return Image.open(f"{title}.png")
 
-# Function to process the uploaded HDF5 file and perform classification using the custom model
-def process_hdf5_file(uploaded_file, percentage_idx):
+# Store the original working directory when the app starts
+original_dir = os.getcwd()
+
+def process_p_file(uploaded_file, percentage_idx, complexity_idx):
     capture = PrintCapture()
     sys.stdout = capture  # Redirect print statements to capture
     
@@ -127,14 +129,15 @@ def process_hdf5_file(uploaded_file, percentage_idx):
             subprocess.run(["git", "clone", model_repo_url, model_repo_dir], check=True)
 
         # Step 2: Verify the repository was cloned and change the working directory
-        if os.path.exists(model_repo_dir):
-            os.chdir(model_repo_dir)
+        repo_work_dir = os.path.join(original_dir, model_repo_dir)
+        if os.path.exists(repo_work_dir):
+            os.chdir(repo_work_dir)  # Change the working directory only once
             print(f"Changed working directory to {os.getcwd()}")
             print(f"Directory content: {os.listdir(os.getcwd())}")  # Debugging: Check repo content
         else:
-            print(f"Directory {model_repo_dir} does not exist.")
+            print(f"Directory {repo_work_dir} does not exist.")
             return
-
+            
         # Step 3: Dynamically load lwm_model.py, input_preprocess.py, and inference.py
         lwm_model_path = os.path.join(os.getcwd(), 'lwm_model.py')
         input_preprocess_path = os.path.join(os.getcwd(), 'input_preprocess.py')
@@ -181,6 +184,8 @@ def process_hdf5_file(uploaded_file, percentage_idx):
         return str(e), str(e), capture.get_output()
 
     finally:
+        # Always return to the original working directory after processing
+        os.chdir(original_dir)
         sys.stdout = sys.__stdout__  # Reset print statements
 
 # Function to handle logic based on whether a file is uploaded or not