support all rife 4.x models, reinstall backend button

the-database · Jul 29, 2024 · 383ebe9 · 383ebe9
1 parent 8b416c3
commit 383ebe9
Show file tree

Hide file tree

Showing 8 changed files with 406 additions and 49 deletions.
diff --git a/VideoJaNai/Services/IPythonService.cs b/VideoJaNai/Services/IPythonService.cs
@@ -1,11 +1,13 @@
 using Avalonia.Collections;
+using System;
 
 namespace AnimeJaNaiConverterGui.Services
 {
  public interface IPythonService
  {
  bool IsPythonInstalled();
  bool AreModelsInstalled();
+ bool IsFfmpegInstalled();
  string PythonDirectory { get; }
  string ModelsDirectory { get; }
  string BackendDirectory { get; }
@@ -14,6 +16,8 @@ public interface IPythonService
  string PythonPath { get; }
  string VspipePath { get; }
  string VapourSynthPluginsPath { get; }
+ string VsmlrtModelsPath { get; }
+ Version VsmlrtMinVersion { get; }
  string AnimeJaNaiDirectory { get; }
  string InstallUpdatePythonDependenciesCommand { get; }
  string InstallVapourSynthPluginsCommand { get; }

diff --git a/VideoJaNai/Services/PythonService.cs b/VideoJaNai/Services/PythonService.cs
@@ -41,19 +41,22 @@ public PythonService(IUpdateManagerService? updateManagerService = null)
  _updateManagerService = updateManagerService ?? Locator.Current.GetService<IUpdateManagerService>()!;
  }
 
- public string ModelsDirectory => Path.Join(AnimeJaNaiDirectory, "onnx");
  public string BackendDirectory => (_updateManagerService?.IsInstalled ?? false) ? Path.Join(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData), @"VideoJaNai") : Path.GetFullPath(@".\backend");
+ public string ModelsDirectory => Path.Join(BackendDirectory, "onnx");
  public string PythonDirectory => Path.Join(BackendDirectory, "python");
  public string FfmpegDirectory => Path.Join(BackendDirectory, "ffmpeg");
  public string AnimeJaNaiDirectory => Path.GetFullPath("./backend/animejanai");
  public string PythonPath => Path.GetFullPath(Path.Join(PythonDirectory, PYTHON_DOWNLOADS["win32"].Path));
  public string VapourSynthPluginsPath => Path.Combine(PythonDirectory, "vs-plugins");
+ public string VsmlrtModelsPath => Path.Combine(VapourSynthPluginsPath, "models");
  public string FfmpegPath => Path.GetFullPath(Path.Join(FfmpegDirectory, "ffmpeg.exe"));
  public string VspipePath => Path.GetFullPath(Path.Join(PythonDirectory, "VSPipe.exe"));
  public string VsrepoPath => Path.GetFullPath(Path.Join(PythonDirectory, "vsrepo.py"));
+ public Version VsmlrtMinVersion => new("3.20.4"); // vsmlrt v14.test3
 
  public bool IsPythonInstalled() => File.Exists(PythonPath);
  public bool AreModelsInstalled() => Directory.Exists(ModelsDirectory) && Directory.GetFiles(ModelsDirectory).Length > 0;
+ public bool IsFfmpegInstalled() => File.Exists(FfmpegPath);
 
  public class PythonDownload
  {