ai import ui improvements

1 files changed, 16 insertions, 4 deletions

diff --git a/src/ui/ui_expenses.cpp b/src/ui/ui_expenses.cpp
index 6ac043f..64dbc18 100644
--- a/src/ui/ui_expenses.cpp
+++ b/src/ui/ui_expenses.cpp
@@ -20,6 +20,7 @@
 
 #include "ui.hpp"
 #include "memops.hpp"
+#include "assets.hpp"
 #include "strops.hpp"
 #include "locales.hpp"
 #include "importer.hpp"
@@ -431,6 +432,9 @@ static void draw_import_request()
         }
     }
 
+    ai_service ai = administration::get_active_ai_service();
+    importer::ai_provider_impl ai_impl = importer::get_ai_provider_implementation(ai.provider);
+
     ImGui::PushFont(ui::fontBig);
 
     ImVec2 windowSize = ImGui::GetWindowSize();
@@ -439,17 +443,25 @@ static void draw_import_request()
     const char* text = importer::status_to_string(active_import_request->status);
     if (active_import_request->error != I_ERR_SUCCESS) text = importer::error_to_string(active_import_request->error);
     ImVec2 textSize = ImGui::CalcTextSize(text);
-    ImGui::SetCursorPos(ImVec2((windowSize.x - textSize.x) * 0.5f,
+    
+    ImGui::SetCursorPos(ImVec2((windowSize.x - textSize.x) * 0.5f - 24,
                     (windowSize.y) * 0.5f - radius - 40.0f));
+    ImGui::Image(assets::load_image(ai_impl.icon_name), ImVec2(24, 24));
+    
+    
+    ImGui::SetCursorPos(ImVec2((windowSize.x - textSize.x) * 0.5f + 10,
+                    (windowSize.y) * 0.5f - radius - 40.0f - 2));
     ImGui::Text(text);
 
     if (active_import_request->error == I_ERR_SUCCESS) {
         ImGui::SetCursorPos(ImVec2((windowSize.x - radius*2) * 0.5f,
                         (windowSize.y - radius*2) * 0.5f));
 
-        const ImVec4 col = ImGui::GetStyleColorVec4(ImGuiCol_ButtonHovered);
-        const ImVec4 bg = ImGui::GetStyleColorVec4(ImGuiCol_Button);
-        ImGui::LoadingIndicatorCircle(radius, bg, col, 10, 4.0f);
+        const ImVec4 col = ImGui::GetStyleColorVec4(ImGuiCol_LoadingIndicatorFg);
+        const ImVec4 bg = ImGui::GetStyleColorVec4(ImGuiCol_LoadingIndicatorBg);
+        ImGui::LoadingIndicatorCircle(radius, bg, col, 10, 4.0f, 0.0f);
+
+        ImGui::Dummy(ImVec2(200,200));
     }
 
     ImGui::PopFont();