From 060b9f918279ecde797dc26c91c664bcbfa66365 Mon Sep 17 00:00:00 2001
From: Nikola Petrov <nikola@petrovv.com>
Date: Wed, 22 Oct 2025 19:07:59 +0200
Subject: [PATCH] s

---
 view/inc/Vapp.hpp |  1 +
 view/src/Vapp.cpp | 11 ++++-------
 2 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/view/inc/Vapp.hpp b/view/inc/Vapp.hpp
index bff36f7..506ccee 100644
--- a/view/inc/Vapp.hpp
+++ b/view/inc/Vapp.hpp
@@ -11,6 +11,7 @@ enum DrawingStage
   drawTree,
   drawBig,
   calSim,
+  save,
   done,
 };
 
diff --git a/view/src/Vapp.cpp b/view/src/Vapp.cpp
index 8a96ace..1c5b812 100644
--- a/view/src/Vapp.cpp
+++ b/view/src/Vapp.cpp
@@ -107,9 +107,12 @@ void Vapp::update()
     simil[3] = Similarity::calc_similarity(manager.vector, Similarity::hamming_distance);
     simil[4] = Similarity::calc_similarity(manager.vector, Similarity::hamming_distance_without_seeds);
     simil[5] = Similarity::calc_similarity(manager.vector, Similarity::levenshtein_distance);
+    stageOfDrawing = DrawingStage::save;
+    break;
+  case DrawingStage::save:
+    drawToFile();
     stageOfDrawing = DrawingStage::done;
     break;
-
   case DrawingStage::done:
     enableAll = true;
     break;
@@ -137,10 +140,6 @@ void Vapp::draw()
     {
       setUpTable();
     }
-    if (ImGui::MenuItem("DrawToFile", nullptr, false, enableAll))
-    {
-      drawToFile();
-    }
     ImGui::EndMainMenuBar();
   }
 
@@ -363,12 +362,10 @@ void Vapp::setUpTable()
 
 void Vapp::drawToFile()
 {
-  enableAll = false;
   int64_t id = ids[selected_id_index];
   char buff[50];
   sprintf(buff, "%ld_%d.png", id, selected_gen);
   Image image = LoadImageFromTexture(bigTexture.texture);
   ExportImage(image, buff);
   UnloadImage(image);
-  enableAll = true;
 }
\ No newline at end of file