2 files changed, 7 insertions, 3 deletions
diff --git a/frontend/src/app/_pages/add-model/add-model.component.html b/frontend/src/app/_pages/add-model/add-model.component.html
index 5c4411b9..f5270127 100644
--- a/frontend/src/app/_pages/add-model/add-model.component.html
+++ b/frontend/src/app/_pages/add-model/add-model.component.html
@@ -30,11 +30,11 @@
     
         <div class="my-5 justify-content-center">
             <h2>Izvor podataka:</h2>
-            <app-dataset-load id="dataset"></app-dataset-load>
+            <app-dataset-load id="dataset" (loaded)="datasetLoaded = true"></app-dataset-load>
         </div>
 
-        <div *ngIf="datasetLoadComponent" class="mt-2">
-            <div class="row">
+        <div *ngIf="datasetLoaded" class="mt-2">
+            <div *ngIf="datasetLoadComponent" class="row">
               <div class="col d-flex justify-content-center">
                 <h3>Izaberite ulazne kolone:</h3> 
                 <div id="divInputs" class="form-check">
diff --git a/frontend/src/app/_pages/add-model/add-model.component.ts b/frontend/src/app/_pages/add-model/add-model.component.ts
index 3c8c32c6..c18ad324 100644
--- a/frontend/src/app/_pages/add-model/add-model.component.ts
+++ b/frontend/src/app/_pages/add-model/add-model.component.ts
@@ -13,6 +13,7 @@ import { ModelsService } from 'src/app/_services/models.service';
 export class AddModelComponent implements OnInit {
 
   @ViewChild(DatasetLoadComponent) datasetLoadComponent?: DatasetLoadComponent;
+  datasetLoaded: boolean = false;
 
   newModel: Model;
 
@@ -31,6 +32,9 @@ export class AddModelComponent implements OnInit {
   }
 
   addModel() {
+    if (this.datasetLoadComponent)
+      this.models.addDataset(this.datasetLoadComponent?.dataset);
+
     this.getCheckedInputCols();
     this.getCheckedOutputCol();
     if (this.validationInputsOutput())