RPCS3/discord-bot
1
0
Fork 0
mirror of https://github.com/RPCS3/discord-bot.git synced 2025-01-07 10:50:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

improve reactions logic

Browse Source
This commit is contained in:
13xforever 2020-05-23 15:16:24 +05:00
parent c4ce030221
commit a16106e59a
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
CompatBot/Commands/Vision.cs
View File

@ -80,7 +80,8 @@ namespace CompatBot.Commands
var client = new ComputerVisionClient(new ApiKeyServiceClientCredentials(Config.AzureComputerVisionKey)) {Endpoint = Config.AzureComputerVisionEndpoint}; var client = new ComputerVisionClient(new ApiKeyServiceClientCredentials(Config.AzureComputerVisionKey)) {Endpoint = Config.AzureComputerVisionEndpoint};
var result = await client.AnalyzeImageAsync(imageUrl, new List<VisualFeatureTypes> {VisualFeatureTypes.Description}, cancellationToken: Config.Cts.Token).ConfigureAwait(false); var result = await client.AnalyzeImageAsync(imageUrl, new List<VisualFeatureTypes> {VisualFeatureTypes.Description}, cancellationToken: Config.Cts.Token).ConfigureAwait(false);
var description = await GetDescriptionAsync(ctx, result.Description, result.Description.Tags).ConfigureAwait(false); var description = GetDescription(result.Description);
await ReactToTagsAsync(ctx.Message, result.Description.Tags).ConfigureAwait(false);
await ctx.RespondAsync(description).ConfigureAwait(false); await ctx.RespondAsync(description).ConfigureAwait(false);
} }
catch (Exception e) catch (Exception e)
@ -151,7 +152,7 @@ namespace CompatBot.Commands
}, },
cancellationToken: Config.Cts.Token cancellationToken: Config.Cts.Token
).ConfigureAwait(false); ).ConfigureAwait(false);
var description = await GetDescriptionAsync(ctx, result.Description, result.Objects.Select(o => o.ObjectProperty).Concat(result.Description.Tags)).ConfigureAwait(false); var description = GetDescription(result.Description);
var objects = result.Objects.OrderBy(c => c.Confidence).ToList(); var objects = result.Objects.OrderBy(c => c.Confidence).ToList();
var scale = Math.Max(1.0f, img.Width / 400.0f); var scale = Math.Max(1.0f, img.Width / 400.0f);
if (objects.Count > 0) if (objects.Count > 0)
@ -286,10 +287,14 @@ namespace CompatBot.Commands
resultStream.Seek(0, SeekOrigin.Begin); resultStream.Seek(0, SeekOrigin.Begin);
quality--; quality--;
} while (resultStream.Length > Config.AttachmentSizeLimit); } while (resultStream.Length > Config.AttachmentSizeLimit);
await ctx.RespondWithFileAsync(Path.GetFileNameWithoutExtension(imageUrl) + "_tagged.jpg", resultStream, description).ConfigureAwait(false); var respondMsg = await ctx.RespondWithFileAsync(Path.GetFileNameWithoutExtension(imageUrl) + "_tagged.jpg", resultStream, description).ConfigureAwait(false);
await ReactToTagsAsync(respondMsg, result.Objects.Select(o => o.ObjectProperty).Concat(result.Description.Tags)).ConfigureAwait(false);
} }
else else
{
await ctx.RespondAsync(description).ConfigureAwait(false); await ctx.RespondAsync(description).ConfigureAwait(false);
await ReactToTagsAsync(ctx.Message, result.Description.Tags).ConfigureAwait(false);
}
} }
catch (Exception e) catch (Exception e)
{ {
@ -306,7 +311,7 @@ namespace CompatBot.Commands
//|| a.FileName.EndsWith(".webp", StringComparison.InvariantCultureIgnoreCase) //|| a.FileName.EndsWith(".webp", StringComparison.InvariantCultureIgnoreCase)
); );
private static async Task<string> GetDescriptionAsync(CommandContext ctx, ImageDescriptionDetails description, IEnumerable<string> tags) private static string GetDescription(ImageDescriptionDetails description)
{ {
var captions = description.Captions.OrderByDescending(c => c.Confidence).ToList(); var captions = description.Captions.OrderByDescending(c => c.Confidence).ToList();
string msg; string msg;
@ -333,10 +338,14 @@ namespace CompatBot.Commands
} }
else else
msg = "An image so weird, I have no words to describe it"; msg = "An image so weird, I have no words to describe it";
return msg;
}
private static async Task ReactToTagsAsync(DiscordMessage reactMsg, IEnumerable<string> tags)
{
foreach (var t in tags) foreach (var t in tags)
if (Reactions.TryGetValue(t, out var emojiList)) if (Reactions.TryGetValue(t, out var emojiList))
await ctx.Message.ReactWithAsync(DiscordEmoji.FromUnicode(emojiList[new Random().Next(emojiList.Length)])).ConfigureAwait(false); await reactMsg.ReactWithAsync(DiscordEmoji.FromUnicode(emojiList[new Random().Next(emojiList.Length)])).ConfigureAwait(false);
return msg;
} }
private static async Task<string> GetImageUrlAsync(CommandContext ctx, string imageUrl) private static async Task<string> GetImageUrlAsync(CommandContext ctx, string imageUrl)