LykosAI · mohnjiles · Jun 8, 2026 · Jan 7, 2026 · Jan 8, 2026 · Jan 8, 2026
diff --git a/CHANGELOG.md b/CHANGELOG.md
diff --git a/StabilityMatrix.Avalonia/App.axaml b/StabilityMatrix.Avalonia/App.axaml
@@ -56,6 +56,7 @@
         <StyleInclude Source="avares://Avalonia.Xaml.Interactions.Draggable/Styles.axaml" />
         <StyleInclude Source="avares://FluentAvalonia.BreadcrumbBar/Styling/Styles.axaml" />
         <StyleInclude Source="Styles/ProgressRing.axaml" />
+        <StyleInclude Source="Styles/ScrollBarStyles.axaml" />
         <StyleInclude Source="Styles/ButtonStyles.axaml" />
         <StyleInclude Source="Styles/SplitButtonStyles.axaml" />
         <StyleInclude Source="Styles/ToggleButtonStyles.axaml" />
@@ -89,6 +90,7 @@
         <StyleInclude Source="Controls/Inference/SharpenCard.axaml" />
         <StyleInclude Source="Controls/Inference/FreeUCard.axaml" />
         <StyleInclude Source="Controls/Inference/ControlNetCard.axaml" />
+        <StyleInclude Source="Controls/Inference/RegionalPromptCard.axaml" />
         <StyleInclude Source="Controls/Inference/PromptExpansionCard.axaml" />
         <StyleInclude Source="Controls/Inference/ExtraNetworkCard.axaml" />
         <StyleInclude Source="Controls/Inference/LayerDiffuseCard.axaml" />

diff --git a/StabilityMatrix.Avalonia/App.axaml.cs b/StabilityMatrix.Avalonia/App.axaml.cs
@@ -51,6 +51,7 @@
 using StabilityMatrix.Avalonia.ViewModels.Progress;
 using StabilityMatrix.Avalonia.Views;
 using StabilityMatrix.Core.Api;
+using StabilityMatrix.Core.Api.Handlers;
 using StabilityMatrix.Core.Api.LykosAuthApi;
 using StabilityMatrix.Core.Api.PromptGenApi;
 using StabilityMatrix.Core.Attributes;
@@ -65,6 +66,7 @@
 using StabilityMatrix.Core.Models.Settings;
 using StabilityMatrix.Core.Python;
 using StabilityMatrix.Core.Services;
+using StabilityMatrix.Core.Services.ImageGeneration;
 using StabilityMatrix.Core.Updater;
 using ApiOptions = StabilityMatrix.Core.Models.Configs.ApiOptions;
 using Application = Avalonia.Application;
@@ -149,6 +151,11 @@ public override void OnFrameworkInitializationCompleted()
     {
         base.OnFrameworkInitializationCompleted();
 
+        if (!Debugger.IsAttached || Program.Args.DebugExceptionDialog)
+        {
+            Dispatcher.UIThread.UnhandledException += Dispatcher_UnhandledException;
+        }
+
         if (Design.IsDesignMode)
         {
             DesignData.DesignData.Initialize();
@@ -389,6 +396,7 @@ internal static void ConfigurePageViewModels(IServiceCollection services)
             {
                 provider.GetRequiredService<PackageManagerViewModel>(),
                 provider.GetRequiredService<InferenceViewModel>(),
+                provider.GetRequiredService<BananaVisionPageViewModel>(),
                 provider.GetRequiredService<CheckpointsPageViewModel>(),
                 provider.GetRequiredService<CheckpointBrowserViewModel>(),
                 provider.GetRequiredService<OutputsPageViewModel>(),
@@ -503,8 +511,21 @@ internal static IServiceCollection ConfigureServices(bool disableMessagePipeInte
         {
             services.AddSingleton<ILiteDbContext, LiteDbContext>();
             services.AddSingleton<IDisposable>(p => p.GetRequiredService<ILiteDbContext>());
+
+            // BananaVision has its own database to preserve conversations when main DB is cleared
+            services.AddSingleton<IBananaVisionDbContext, BananaVisionDbContext>();
+            services.AddSingleton<IDisposable>(p => p.GetRequiredService<IBananaVisionDbContext>());
         }
 
+        // Image generation services
+        services.AddSingleton<IImageGenerationProvider, GeminiImageGenerationProvider>();
+        services.AddSingleton<IImageGenerationProvider, Gemini31FlashImageGenerationProvider>();
+        services.AddSingleton<IImageGenerationProvider, Gemini3ProImageGenerationProvider>();
+        services.AddSingleton<IImageGenerationProvider, FluxKontextProvider>();
+        services.AddSingleton<IImageGenerationProvider, QwenImageEditProvider>();
+        services.AddSingleton<IImageGenerationProvider, Flux2KleinProvider>();
+        services.AddSingleton<IImageGenerationChatService, ImageGenerationChatService>();
+
         services.AddTransient<IGitHubClient, GitHubClient>(_ =>
         {
             var client = new GitHubClient(new ProductHeaderValue("StabilityMatrix"));
@@ -728,7 +749,7 @@ internal static IServiceCollection ConfigureServices(bool disableMessagePipeInte
                 }
             )
             .ConfigurePrimaryHttpMessageHandler(() => new HttpClientHandler { AllowAutoRedirect = false })
-            .AddPolicyHandler(retryPolicy)
+            .AddPolicyHandler(retryPolicyLonger)
             .AddHttpMessageHandler(serviceProvider => new TokenAuthHeaderHandler(
                 serviceProvider.GetRequiredService<LykosAuthTokenProvider>()
             ));
@@ -764,6 +785,19 @@ internal static IServiceCollection ConfigureServices(bool disableMessagePipeInte
             })
             .AddPolicyHandler(retryPolicy); // Assuming retryPolicy is suitable
 
+        services
+            .AddRefitClient<IGeminiApi>(defaultRefitSettings)
+            .ConfigureHttpClient(c =>
+            {
+                c.BaseAddress = new Uri("https://generativelanguage.googleapis.com");
+                c.Timeout = TimeSpan.FromMinutes(5); // Higher timeout for image generation
+            })
+            .AddHttpMessageHandler<GeminiApiKeyHandler>()
+            .AddPolicyHandler(retryPolicyLonger);
+
+        // Register GeminiApiKeyHandler
+        services.AddTransient<GeminiApiKeyHandler>();
+
         // Apizr clients
         services.AddApizrManagerFor<IOpenModelDbApi, OpenModelDbManager>(options =>
         {
@@ -1039,6 +1073,14 @@ private static void OnServiceProviderDisposing(ServiceProvider serviceProvider)
         Logger.Trace("Disposing {Count} Disposables", disposables.Count);
     }
 
+    private static void Dispatcher_UnhandledException(object? sender, DispatcherUnhandledExceptionEventArgs e)
+    {
+        if (Program.ShowExceptionDialog(e.Exception, true))
+        {
+            e.Handled = true;
+        }
+    }
+
     private static void TaskScheduler_UnobservedTaskException(
         object? sender,
         UnobservedTaskExceptionEventArgs e

diff --git a/StabilityMatrix.Avalonia/Assets.cs b/StabilityMatrix.Avalonia/Assets.cs
@@ -105,7 +105,7 @@ internal static class Assets
                 new RemoteResource
                 {
                     Url = new Uri("https://www.python.org/ftp/python/3.10.11/python-3.10.11-embed-amd64.zip"),
-                    HashSha256 = "608619f8619075629c9c69f361352a0da6ed7e62f83a0e19c63e0ea32eb7629d"
+                    HashSha256 = "608619f8619075629c9c69f361352a0da6ed7e62f83a0e19c63e0ea32eb7629d",
                 }
             ),
             (
@@ -115,7 +115,7 @@ internal static class Assets
                     Url = new Uri(
                         "https://github.com/indygreg/python-build-standalone/releases/download/20230507/cpython-3.10.11+20230507-x86_64-unknown-linux-gnu-install_only.tar.gz"
                     ),
-                    HashSha256 = "c5bcaac91bc80bfc29cf510669ecad12d506035ecb3ad85ef213416d54aecd79"
+                    HashSha256 = "c5bcaac91bc80bfc29cf510669ecad12d506035ecb3ad85ef213416d54aecd79",
                 }
             ),
             (
@@ -124,7 +124,48 @@ internal static class Assets
                 {
                     // Requires our distribution with signed dylib for gatekeeper
                     Url = new Uri("https://cdn.lykos.ai/cpython-3.10.11-macos-arm64.zip"),
-                    HashSha256 = "83c00486e0af9c460604a425e519d58e4b9604fbe7a4448efda0f648f86fb6e3"
+                    HashSha256 = "83c00486e0af9c460604a425e519d58e4b9604fbe7a4448efda0f648f86fb6e3",
+                }
+            )
+        );
+
+    /// <summary>
+    /// FFmpeg LGPL builds for video thumbnail generation.
+    /// </summary>
+    [SupportedOSPlatform("windows")]
+    [SupportedOSPlatform("linux")]
+    [SupportedOSPlatform("macos")]
+    public static RemoteResource FfmpegDownloadUrl =>
+        Compat.Switch(
+            (
+                PlatformKind.Windows | PlatformKind.X64,
+                new RemoteResource
+                {
+                    // BtbN LGPL build - ffmpeg-n7.1-latest-win64-lgpl-7.1
+                    Url = new Uri(
+                        "https://github.com/BtbN/FFmpeg-Builds/releases/download/latest/ffmpeg-n7.1-latest-win64-lgpl-7.1.zip"
+                    ),
+                    HashSha256 = "a77ecdc794d67401f3e4976f8856065f7762d74afd16f9c7b777ff0291a7bcaa",
+                }
+            ),
+            (
+                PlatformKind.Linux | PlatformKind.X64,
+                new RemoteResource
+                {
+                    // BtbN LGPL build - linux
+                    Url = new Uri(
+                        "https://github.com/BtbN/FFmpeg-Builds/releases/download/latest/ffmpeg-n7.1-latest-linux64-lgpl-7.1.tar.xz"
+                    ),
+                    HashSha256 = "d7d691dfa3a6d0a75362c02274a80a1f9635bd67908561aae31ee538853ab8ce",
+                }
+            ),
+            (
+                PlatformKind.MacOS | PlatformKind.Arm,
+                new RemoteResource
+                {
+                    // evermeet.cx build for macOS arm64
+                    Url = new Uri("https://evermeet.cx/ffmpeg/ffmpeg-7.1.1.zip"),
+                    HashSha256 = "8d7917c1cebd7a29e68c0a0a6cc4ecc3fe05c7fffed958636c7018b319afdda4",
                 }
             )
         );
@@ -135,18 +176,18 @@ internal static class Assets
             new RemoteResource
             {
                 Url = new Uri("https://cdn.lykos.ai/tags/danbooru.csv"),
-                HashSha256 = "b84a879f1d9c47bf4758d66542598faa565b1571122ae12e7b145da8e7a4c1c6"
+                HashSha256 = "b84a879f1d9c47bf4758d66542598faa565b1571122ae12e7b145da8e7a4c1c6",
             },
             new RemoteResource
             {
                 Url = new Uri("https://cdn.lykos.ai/tags/e621.csv"),
-                HashSha256 = "ef7ea148ad865ad936d0c1ee57f0f83de723b43056c70b07fd67dbdbb89cae35"
+                HashSha256 = "ef7ea148ad865ad936d0c1ee57f0f83de723b43056c70b07fd67dbdbb89cae35",
             },
             new RemoteResource
             {
                 Url = new Uri("https://cdn.lykos.ai/tags/danbooru_e621_merged.csv"),
-                HashSha256 = "ac405ebce8b0caae363a7ef91f89beb4b8f60a7e218deb5078833686da6d497d"
-            }
+                HashSha256 = "ac405ebce8b0caae363a7ef91f89beb4b8f60a7e218deb5078833686da6d497d",
+            },
         };
 
     public static Uri DiscordServerUrl { get; } = new("https://discord.com/invite/TUrgfECxHz");

diff --git a/StabilityMatrix.Avalonia/Assets/ImagePrompt.tmLanguage.json b/StabilityMatrix.Avalonia/Assets/ImagePrompt.tmLanguage.json
@@ -320,4 +320,4 @@
       ]
     }
   }
-}
+}
diff --git a/StabilityMatrix.Avalonia/Assets/hf-packages.json b/StabilityMatrix.Avalonia/Assets/hf-packages.json
@@ -1223,12 +1223,11 @@
   {
     "ModelCategory": "Vae",
     "ModelName": "Flux.1 VAE",
-    "RepositoryPath": "black-forest-labs/FLUX.1-schnell",
+    "RepositoryPath": "Comfy-Org/Lumina_Image_2.0_Repackaged",
     "Files": [
-      "ae.safetensors"
+      "split_files/vae/ae.safetensors"
     ],
-    "LicenseType": "Apache 2.0",
-    "LoginRequired": true
+    "LicenseType": "Flux.1 Dev NonCommercial"
   },
   {
     "ModelCategory": "Vae",

diff --git a/StabilityMatrix.Avalonia/Assets/linux-x64/7zzs b/StabilityMatrix.Avalonia/Assets/linux-x64/7zzs
-Original file line number
+Diff line change
@@ Expand Up / @@ -320,4 +320,4 @@ @@
           ]
         }
       }
-    }
+    }