Update app.py

app.py

@@ -32,16 +32,39 @@ class PrintCapture(io.StringIO):
         return ''.join(self.output)
 
 # Function to load and display predefined images based on user selection
+#def display_predefined_images(percentage_idx, complexity_idx):
+#    percentage = percentage_values[percentage_idx]
+#    complexity = complexity_values[complexity_idx]
+#    raw_image_path = os.path.join(RAW_PATH, f"percentage_{percentage}_complexity_{complexity}.png")
+#    embeddings_image_path = os.path.join(EMBEDDINGS_PATH, f"percentage_{percentage}_complexity_{complexity}.png")
+    
+#    raw_image = Image.open(raw_image_path)
+#    embeddings_image = Image.open(embeddings_image_path)
+    
+#    return raw_image, embeddings_image
+
 def display_predefined_images(percentage_idx, complexity_idx):
+    # Map the slider index to the actual value
     percentage = percentage_values[percentage_idx]
     complexity = complexity_values[complexity_idx]
+    
+    # Generate the paths to the images
     raw_image_path = os.path.join(RAW_PATH, f"percentage_{percentage}_complexity_{complexity}.png")
     embeddings_image_path = os.path.join(EMBEDDINGS_PATH, f"percentage_{percentage}_complexity_{complexity}.png")
     
+    # Check if the images exist
+    if not os.path.exists(raw_image_path):
+        return None, None  # Or handle the error appropriately
+    if not os.path.exists(embeddings_image_path):
+        return None, None  # Or handle the error appropriately
+
+    # Load images using PIL
     raw_image = Image.open(raw_image_path)
     embeddings_image = Image.open(embeddings_image_path)
     
+    # Return the loaded images
     return raw_image, embeddings_image
+    
 
 # Function to load the pre-trained model from your cloned repository
 def load_custom_model():
@@ -63,14 +86,14 @@ def process_p_file(uploaded_file, percentage_idx, complexity_idx):
             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
-        if os.path.exists(model_repo_dir):
-            os.chdir(model_repo_dir)
-            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.")
-            return
+        ## Debugging: Check if the directory exists and print contents
+        #if os.path.exists(model_repo_dir):
+        #    os.chdir(model_repo_dir)
+        #    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.")
+        #    return
 
         ## Step 2: Add the cloned repo to sys.path for imports
         #if model_repo_dir not in sys.path:
@@ -78,7 +101,7 @@ def process_p_file(uploaded_file, percentage_idx, complexity_idx):
         
         ## Debugging: Print sys.path to ensure the cloned repo is in the path
         #print(f"sys.path: {sys.path}")
-        sys.path.append('./')
+        #sys.path.append('./')
         
         from lwm_model import LWM
         device = 'cpu'