expose file list

utils/pipelines.py

@@ -4,6 +4,7 @@ from lightning.fabric import seed_everything
 import time
 import os
 import spaces
+import subprocess
 
 
 @spaces.GPU(duration=300)  
@@ -65,12 +66,10 @@ def unconditional_generation(num_batches, num_designs_per_batch, length):
             n_batches=num_batches,      # Generate 1 batch
         )
 
-        #for idx, data in outputs.items():
-        #    return_str += f"Batch {idx}: {len(data)} structure(s)\n"
-        #    for i, struct in enumerate(data):
-        #        return_str += f"Structure {i+1}: {struct.atom_array.array_length()} Atoms\n"
-        #        #return_str += struct.atom_array
-        return directory
+        cmd = f"ls -R {directory}"
+        file_list = subprocess.check_output(cmd, shell=True).decode()
+
+        return file_list
     
     except Exception as e:
         return f"Error: {str(e)}"