Update app.py

app.py

@@ -45,15 +45,40 @@ if process.returncode == 0:
 else:
     print("Houve um erro na execução do comando:", process.stderr)
 
+
+def find_and_move_library(library_name, destination):
+    # Search for the library in the entire system
+    try:
+        command = f"find / -name '{library_name}' 2>/dev/null"
+        process = subprocess.run(command, shell=True, text=True, capture_output=True)
+        
+        if process.returncode == 0:
+            found_paths = process.stdout.strip().split('\n')
+            if found_paths:
+                for path in found_paths:
+                    print(f"Found {library_name} at: {path}")
+                    # Move the library to the destination
+                    try:
+                        subprocess.run(['mv', path, destination], check=True)
+                        print(f"Moved {library_name} to {destination}")
+                        break  # Exit after moving the first valid match
+                    except subprocess.CalledProcessError as e:
+                        print(f"Failed to move {path} to {destination}: {e}")
+            else:
+                print(f"No {library_name} found.")
+        else:
+            print("Error during search:", process.stderr)
+    except Exception as e:
+        print(f"Error finding {library_name}: {e}")
+
 source_path = '/usr/local/lib/python3.10/site-packages/nvidia/cublas/lib/libcublasLt.so.12'
 destination_path = '/usr/local/lib/python3.10/site-packages/nvidia/cudnn/lib/'
 command = ['mv', source_path, destination_path]
 subprocess.run(command, check=True)
 
-command = ['mv', "/usr/local/lib/python3.10/site-packages/nvidia/cublas/lib/libcublas.so.12", destination_path]
-subprocess.run(command, check=True)
+find_and_move_library('libnvrtc.so.12', destination_path)
 
-command = ['mv', "/usr/local/lib/python3.10/site-packages/nvidia/cublas/lib/libnvrtc.so.12", destination_path]
+command = ['mv', "/usr/local/lib/python3.10/site-packages/nvidia/cublas/lib/libcublas.so.12", destination_path]
 subprocess.run(command, check=True)
 
 command = ['mv', "/usr/local/lib/python3.10/site-packages/nvidia/cufft/lib/libcufft.so.11", destination_path]