#pragma warning disable SKEXP0070

using Microsoft.Extensions.DependencyInjection;
using Microsoft.SemanticKernel;
using Microsoft.SemanticKernel.ChatCompletion;
using Microsoft.SemanticKernel.Connectors.AzureAIInference;
using OllamaApiFacade.Extensions;
using SemanticFlow.DemoWebApi;

var azureKeyVaultHelper = new AzureKeyVaultHelper("https://XXX.vault.azure.net");
var endpoint = await azureKeyVaultHelper.GetSecretAsync("AZURE-MINISTRAL-NEMO-ENDPOINT");
var apiKey = await azureKeyVaultHelper.GetSecretAsync("AZURE-MINISTRAL-NEMO-KEY");

// Kernel setup
var builder = Kernel.CreateBuilder()
    .AddAzureAIInferenceChatCompletion("Mistral-Nemo-oclsi", apiKey, new Uri(endpoint));

// Use Burp Suite proxy for analysis backend communication
// builder.Services.AddProxyForDebug();

var kernel = builder.Build();

// Import plugin functions
kernel.ImportPluginFromFunctions("HelperFunctions",
[
    kernel.CreateFunctionFromMethod(() => new List<string> { "Squirrel Steals Show", "Dog Wins Lottery" },
        "GetLatestNewsTitles", "Retrieves latest news titles."),
    kernel.CreateFunctionFromMethod(() => DateTime.UtcNow.ToString("R"), 
        "GetCurrentUtcDateTime", "Retrieves the current date time in UTC."),
    kernel.CreateFunctionFromMethod((string cityName, string currentDateTime) =>
    {
        if (string.IsNullOrEmpty(cityName) || string.IsNullOrEmpty(currentDateTime))
        {
            throw new ArgumentException("cityName and currentDateTime are required.");
        }
        return cityName switch
        {
            "Boston" => "61 and rainy",
            "London" => "55 and cloudy",
            "Miami" => "80 and sunny",
            "Paris" => "60 and rainy",
            "Tokyo" => "50 and sunny",
            "Sydney" => "75 and sunny",
            "Tel Aviv" => "80 and sunny",
            _ => "31 and snowing",
        };
    }, "GetWeatherForCity", "Gets the current weather for the specified city, using the city name and current UTC date/time.")
]);

// Settings with Streaming Enabled
var settings = new AzureAIInferencePromptExecutionSettings { FunctionChoiceBehavior = FunctionChoiceBehavior.Auto() };

var chatHistory = new ChatHistory();
chatHistory.AddUserMessage("What is the weather in Tokyo based on the current date and time?");

// Streaming with IChatCompletionService
var chatCompletionService = kernel.Services.GetRequiredService<IChatCompletionService>();

try
{
    // Stream the response
    await foreach (var message in chatCompletionService.GetStreamingChatMessageContentsAsync(chatHistory, settings, kernel))
    {
        if (message.Role.HasValue)
        {
            Console.Write($"{message.Role.Value}: ");
        }
        if (!string.IsNullOrEmpty(message.Content))
        {
            Console.Write(message.Content);
        }
    }
    Console.WriteLine("\nStreaming completed.");
}
catch (Exception ex)
{
    // Log any exceptions
    Console.WriteLine($"Error: {ex.Message}");
}