import { ask, say  } from "../shared/cli.js";
import { gptPrompt } from "../shared/openai.js";
import chalk from 'chalk';
import boxen from 'boxen';

async function main() {
    console.log(chalk.bold.yellowBright("Welcome to the Meal Suggestion App!"));
    
    const groceries = await ask(chalk.green.bold("What groceries do you have? List them separated by commas: "));
    
    // Generate dish suggestions based on the provided groceries
    const dishSuggestionsPrompt = `Given these groceries: ${groceries}, suggest three dishes that can be made. List them in a format suitable for selection.`;
    
    // Define max_tokens
    const maxTokens = 400; 
    const dishSuggestionsString = await gptPrompt(dishSuggestionsPrompt, { max_tokens: maxTokens });
    
    let dishSuggestions = dishSuggestionsString.split('\n').filter(line => line.trim().length > 0 && line.trim().match(/^\d\./)).map(line => line.trim().substring(2).trim());
    
    if (dishSuggestions.length === 0) {
        say(`No dish suggestions could be parsed from the response: "${dishSuggestionsString}"`);
        return; 
    }
    
    say(chalk.green.bold("Based on your groceries, you could make:"));
    const formattedSuggestions = dishSuggestions.map((dish, index) => `${index + 1}. ${dish}`).join('\n');
    say(boxen(formattedSuggestions, { padding: 1 }));
    
    const dishChoice = await ask(chalk.green.bold("Which dish would you like to make? (Enter the number corresponding to the dish): "));
    const chosenDish = dishSuggestions[parseInt(dishChoice) - 1];
    
    if (!chosenDish) {
        say("Invalid dish selection lol.");
        return; // Exit if no dish selected
    }
    
    // Generate recipe for the chosen dish
    const recipePrompt = `Provide a simple recipe for making ${chosenDish} using the following groceries: ${groceries}.`;
    const recipe = await gptPrompt(recipePrompt, { max_tokens: maxTokens });
    
    say(chalk.blue.bold(`Here is a simple recipe for ${chosenDish}:`));

    // Format recipe text using chalk
    const formattedRecipe = chalk.yellow(recipe);
    
    say(formattedRecipe);
    
    
}

main();