adding suppport for multiple files

Author: ArturU043

servicex_analysis_utils/file_peeking.py

@@ -144,21 +144,36 @@ def print_structure_from_str(deliver_dict, filter_branch="", save_to_txt=False )
                         print(f"   │   ├── {branch_line[8:]}")
 
 
-def get_structure(dataset_dict, **kwargs):
+def get_structure(dataset, **kwargs):
     """
     Utility function. 
-    Creates and sends the ServiceX request from user input.
+    Creates and sends the ServiceX request from user inputed datasets to retrieve file stucture.
     Calls print_structure_from_str()
 
     Parameters:
-      dataset_dict (dict): The datasets to print the structures from, with the associated sample name for readability 
-                            note - should add default sample names and option to add just did or list of dids 
+      dataset_dict (dict,str,[str]): The datasets from which to print the file structures.
+                                     A custom sample name per dataset can be given in a dict form: {'sample_name':'dataset_id'}
       kwargs : Arguments to be propagated to print_structure_from_str 
     """
     #Servicex query using the PythonFunction backend
     query_PythonFunction = servicex.query.PythonFunction().with_uproot_function(run_query)
-    sample_list=[]
 
+    #Create a dict with sample name for ServiceX query & datasetID
+    dataset_dict={}
+    user_in=type(dataset)
+    if user_in == str:
+        dataset_dict.update({"Sample":dataset})
+    elif user_in == list and type(dataset[0]) is str:
+        for i in range(len(dataset)):
+            name="Sample"+str(i+1) #write for humans
+            dataset_dict.update({name:dataset[i]})
+    elif user_in == dict:
+        dataset_dict=dataset
+    else:
+        raise ValueError(f"Unsupported dataset input type: {user_in}.\nInput must be dict ('sample_name':'dataset_id'), str or list of str")
+    
+    sample_list=[]
+
     for name, did in dataset_dict.items():
         tmp_dict={
             "NFiles":1,