3 files changed, 8 insertions, 4 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()) 
diff --git a/frontend/src/app/_pages/home/home.component.html b/frontend/src/app/_pages/home/home.component.html
index 374cb324..7245f602 100644
--- a/frontend/src/app/_pages/home/home.component.html
+++ b/frontend/src/app/_pages/home/home.component.html
@@ -1,4 +1,4 @@
-<div class="d-flex flex-column align-items-center">
+<div class="d-flex flex-column align-items-center bg-light">
     <img src="../../../assets/svg/logo.svg" class="bi me-2" width="256" height="256" role="img">
     <div *ngIf="shared.loggedIn" class="d-flex flex-column align-items-center">
         <h2 class="my-4">Započnite sa treniranjem!</h2>