1 files changed, 27 insertions, 8 deletions
diff --git a/frontend/src/app/training/training.component.ts b/frontend/src/app/training/training.component.ts
index c82a6b79..1b85b947 100644
--- a/frontend/src/app/training/training.component.ts
+++ b/frontend/src/app/training/training.component.ts
@@ -4,15 +4,17 @@ import Shared from '../Shared';
 import Experiment from '../_data/Experiment';
 import Model, { ProblemType } from '../_data/Model';
 import { ModelLoadComponent } from '../_elements/model-load/model-load.component';
+import { AuthService } from '../_services/auth.service';
 import { ExperimentsService } from '../_services/experiments.service';
 import { ModelsService } from '../_services/models.service';
+import { SignalRService } from '../_services/signal-r.service';
 
 @Component({
   selector: 'app-training',
   templateUrl: './training.component.html',
   styleUrls: ['./training.component.css']
 })
-export class TrainingComponent implements OnInit{
+export class TrainingComponent implements OnInit {
 
   @ViewChild(ModelLoadComponent) modelLoadComponent?: ModelLoadComponent;
 
@@ -24,22 +26,39 @@ export class TrainingComponent implements OnInit{
 
   term: string = "";
 
-  constructor(private modelsService: ModelsService, private route: ActivatedRoute, private experimentsService: ExperimentsService) { 
+  constructor(private modelsService: ModelsService, private route: ActivatedRoute, private experimentsService: ExperimentsService, private authService: AuthService, private signalRService: SignalRService) {
+    if (this.signalRService.hubConnection) {
+      this.signalRService.hubConnection.on("NotifyEpoch", (mName: string, mId: string, stat: string, totalEpochs: number, currentEpoch: number) => {
+        if (this.selectedModel?._id == mId) {
+          this.trainingResult = stat;
+        }
+      });
+    }
   }
 
-  ngOnInit(): void { 
+  ngOnInit(): void {
     this.route.queryParams.subscribe(params => {
-      let experimentId =this.route.snapshot.paramMap.get("id");
+      let experimentId = this.route.snapshot.paramMap.get("id");
 
-      this.experimentsService.getMyExperiments().subscribe((experiments) => {
-        this.myExperiments = experiments;
+      this.fetchExperiments(experimentId);
 
-        this.selectedExperiment = this.myExperiments.filter(x => x._id == experimentId)[0];
-        console.log("selektovan exp u training comp: ", this.selectedExperiment);
+      this.authService.loggedInEvent.subscribe(_ => {
+        this.fetchExperiments(experimentId);
+
+        this.signalRService.startConnection()
       });
     });
   }
 
+  fetchExperiments(andSelectWithId: string | null = '') {
+    this.experimentsService.getMyExperiments().subscribe((experiments) => {
+      this.myExperiments = experiments;
+
+      this.selectedExperiment = this.myExperiments.filter(x => x._id == andSelectWithId)[0];
+      console.log("selektovan exp u training comp: ", this.selectedExperiment);
+    });
+  }
+
   selectThisExperiment(experiment: Experiment) {
     this.selectedExperiment = experiment;
     this.modelLoadComponent!.newModel.type = this.selectedExperiment.type;