Found error in reading of output files. Need to edit how to output the files

Data_Generation/Dataset_Generation_Functions.py

@@ -7,7 +7,7 @@ import json
 from Data_Generation.Piecewise_Box_Functions import basic_box_array, back_slash_array, forward_slash_array, hamburger_array, hot_dog_array
 # from Piecewise_Box_Functions import basic_box_array, back_slash_array, forward_slash_array, hamburger_array, hot_dog_array
 import matplotlib.pyplot as plt
-
+from json import JSONEncoder
 
 
 
@@ -77,8 +77,17 @@ boxes = make_boxes(image_size, densities)
 
 box_arrays, box_density, basic_box_thickness, forward_slash_box_thickness, back_slash_box_thickness,hot_dog_box_thickness, hamburger_box_thickness\
     = list(zip(*boxes))[0], list(zip(*boxes))[1], list(zip(*boxes))[2], list(zip(*boxes))[3], list(zip(*boxes))[4], list(zip(*boxes))[5], list(zip(*boxes))[6]
+print(type(box_arrays[0]))
+
+
+class NumpyArrayEncoder(JSONEncoder):
+    def default(self, obj):
+        if isinstance(obj, np.ndarray):
+            return obj.tolist()
+        return JSONEncoder.default(self, obj)
 
 
+box_arrays = [json.dumps(x, cls=NumpyArrayEncoder) for x in box_arrays] # find argument for json dumps for numpy
 
 # Create a dataframe to convert the data to a csv file
 dataframe = (pd.DataFrame((box_arrays, box_density, basic_box_thickness, forward_slash_box_thickness, back_slash_box_thickness,hot_dog_box_thickness, hamburger_box_thickness)).T).astype(str)
@@ -86,17 +95,17 @@ dataframe = (pd.DataFrame((box_arrays, box_density, basic_box_thickness, forward
 # Rename the columns to the desired outputs
 dataframe = dataframe.rename(columns={0: "Array", 1: "Density", 2:"Basic Box Thickness", 3:"Forward Slash Strut Thickness", 4:"Back Slash Strut Thickness", 5:"Vertical Strut Thickness", 6:"Horizontal Strut Thickness"})
 
-csv = dataframe.to_csv('2D_Lattice')
+csv = dataframe.to_csv('2D_Lattice.csv')
 
 
-
-df = pd.read_csv('2D_Lattice')
+df = pd.read_csv('2D_Lattice.csv')
 # print(df)
 row = 0
 box = df.iloc[row, 1]
 print(box)
 array = np.array(json.loads(box))
-
+plt.imshow(array)
+plt.show()
 
 # import matplotlib.pyplot as plt
 # plt.imshow(np.array(box))

app.py

@@ -7,7 +7,7 @@ from Data_Generation.Dataset_Generation_Functions import make_boxes
 from Data_Generation.Piecewise_Box_Functions import basic_box_array, forward_slash_array, combine_arrays, add_thickness
 ########################################################################################################################
 # User Inputs
-image_size = st.slider('Select a value for the image size', min_value=9, max_value=28)
+image_size = st.slider('Select a value for the image size', min_value=9, max_value=16)
 # st.write(x, 'squared is', x * x)
 
 density_selection = st.slider('Select a value for the number of equally spaced density values (0, 1]', min_value=1, max_value=10)