Enabled faster hat training and reset of the AI iteration

Author: csharpfritz

Fritz.Chatbot/Commands/PredictHatCommand.cs

@@ -24,7 +24,7 @@ public class PredictHatCommand : IBasicCommand
 		private string _TwitchChannel = "";
 		private Guid _AzureProjectId;
 
-		private static string _IterationName = "";
+		internal static string IterationName = "";
 		private ScreenshotTrainingService _TrainHat;
 
 		public PredictHatCommand(IConfiguration configuration, ScreenshotTrainingService service)
@@ -42,7 +42,7 @@ public PredictHatCommand(IConfiguration configuration, ScreenshotTrainingService
 		public async Task Execute(IChatService chatService, string userName, ReadOnlyMemory<char> rhs)
 		{
 
-			if (string.IsNullOrEmpty(_IterationName)) {
+			if (string.IsNullOrEmpty(IterationName)) {
 				await IdentifyIterationName();
 			}
 
@@ -59,7 +59,7 @@ public async Task Execute(IChatService chatService, string userName, ReadOnlyMem
 			ImagePrediction result;
 			try
 			{
-				result = await client.DetectImageWithNoStoreAsync(_AzureProjectId, _IterationName, obsImage);
+				result = await client.DetectImageWithNoStoreAsync(_AzureProjectId, IterationName, obsImage);
 			} catch (CustomVisionErrorException ex) {
 
 
@@ -94,7 +94,7 @@ private async Task IdentifyIterationName()
 			};
 
 			var iterations = await client.GetIterationsAsync(_AzureProjectId);
-			_IterationName = iterations
+			IterationName = iterations
 				.Where(i => !string.IsNullOrEmpty(i.PublishName) && i.Status == "Completed")
 				.OrderByDescending(i => i.LastModified).First().PublishName;
 

Fritz.Chatbot/Commands/ResetHatAiCommand.cs

@@ -0,0 +1,31 @@
+using Fritz.StreamLib.Core;
+using System;
+using System.Collections.Generic;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Fritz.Chatbot.Commands
+{
+	public class ResetHatAiCommand : IBasicCommand2
+	{
+
+		public string Trigger => "resethatai";
+		public string Description => "Reset the Hat detection AI after adding new hats";
+		public TimeSpan? Cooldown { get; } = TimeSpan.FromSeconds(30);
+
+		public async Task Execute(IChatService chatService, string userName, bool isModerator, bool isVip, bool isBroadcaster, ReadOnlyMemory<char> rhs)
+		{
+
+			if (!(isModerator || isBroadcaster)) return;
+
+			PredictHatCommand.IterationName = string.Empty;
+			await chatService.SendMessageAsync("Reset the AI iteration and will detect the latest for Hat identification next time !hat is called");
+
+		}
+
+		public Task Execute(IChatService chatService, string userName, ReadOnlyMemory<char> rhs)
+		{
+			return Task.CompletedTask;
+		}
+	}
+}

Fritz.Chatbot/Commands/TrainHatCommand.cs

@@ -26,7 +26,7 @@ public async Task Execute(IChatService chatService, string userName, bool isMode
 			if (!(isModerator || isBroadcaster)) return;
 
 			_TrainHat.StartTraining();
-			await chatService.SendMessageAsync("Started taking screenshots, 1 per minute for the next 15 minutes");
+			await chatService.SendMessageAsync("Started taking screenshots, 1 every ten seconds for the next 150 seconds");
 
 		}
 

Fritz.Chatbot/ScreenshotTrainingService.cs

@@ -79,7 +79,7 @@ private async Task Train(CancellationToken token)
 				{
 
 					await AddScreenshot(true);
-					await Task.Delay(TimeSpan.FromMinutes(1));
+					await Task.Delay(TimeSpan.FromSeconds(10));
 
 				}
 				else