Add localization

1 files changed, 5 insertions, 3 deletions

diff --git a/src/handlers/export.rs b/src/handlers/export.rs
index 0b64a73..a0a4451 100644
--- a/src/handlers/export.rs
+++ b/src/handlers/export.rs
@@ -18,24 +18,26 @@ use std::path::PathBuf;
 use crate::AppModel;
 use crate::actions::ImageFormat;
 use crate::dialogs;
+use crate::tr;
+use crate::stages::LocalizedStageType;
 use wmap_renderer::{render_to_png, render_to_svg};
 
 /// Exports the map to an image file in the specified format.
 pub fn export_to_file(model: &AppModel, path: PathBuf, format: ImageFormat) {
     let result: Result<(), String> = match format {
         ImageFormat::Png => {
-            render_to_png(&model.map, &model.stage_type, &model.render_configuration)
+            render_to_png(&model.map, &model.stage_type.to_localized(), &model.render_configuration)
                 .map_err(|e| e.to_string())
                 .and_then(|data| std::fs::write(&path, data).map_err(|e| e.to_string()))
         }
         ImageFormat::Svg => {
-            render_to_svg(&model.map, &model.stage_type, &model.render_configuration)
+            render_to_svg(&model.map, &model.stage_type.to_localized(), &model.render_configuration)
                 .map_err(|e| e.to_string())
                 .and_then(|data| std::fs::write(&path, data).map_err(|e| e.to_string()))
         }
     };
 
     if let Err(error) = result {
-        dialogs::show_error(&model.window, "Error Exporting Image", &error);
+        dialogs::show_error(&model.window, &tr!("error.file.export"), &error);
     }
 }