@Agent(groupName = "SimpleAction", groupDescription = "This is a simple action which can be used to test the action processor")
public class SimpleAction {

    @Action(description = "Provide persons name and then find out what does that person like")
    public String whatFoodDoesThisPersonLike(String name) {
        if("vishal".equalsIgnoreCase(name))
            return "Paneer Butter Masala";
        else if ("vinod".equalsIgnoreCase(name)) {
            return "aloo kofta";
        }else
            return "something yummy";
    }
}

public class SimpleActionExample {
    public static void main(String[] args) {
        System.setProperty("tools4ai.properties.path", "io/github/vishalmysore/simple/tools4ai.properties");
        AIProcessor processor = PredictionLoader.getInstance().createOrGetAIProcessor();
        try {
            String result = (String) processor.processSingleAction("what does Vishal like to eat");
            System.out.println(result); // Output: Paneer Butter Masala
        } catch (AIProcessingException e) {
            e.printStackTrace();
        }
    }
}

@Agent(groupName = "personal", groupDescription = "all personal actions are here")
public class MyAgent {
    // ...
}

@Action(description = "Handles travel preferences")
public String preferTravel(String name, String preference) {
    // implementation
}

public class Customer {
    @Prompt(describe = "convert this to Hindi")
    private String reasonForCalling;

    @Prompt(ignore = true)
    private String location;

    @Prompt(dateFormat = "yyyy-MM-dd", describe = "if you dont find date provide todays date")
    private Date dateJoined;
}

@Action
public String[] myTaskList(String[] myTaskList) {
    return myTaskList;
}

@Action
public Map<Integer, String> addSports(@MapKeyType(Integer.class) @MapValueType(String.class) Map<Integer, String> mapOfSportsName) {
    return mapOfSportsName;
}

@Agent
public class TaskListAction {
    @Action
    public String[] myTaskList(String[] myTaskList) {
        return myTaskList;
    }
}

// Usage
String[] taskList = (String[]) processor.processSingleAction("Vishal would like to go to gym, then play football and finally watch a cricket match.");

@Agent
public class SportsDirectory {
    @Action(description = "add new Sports into the map")
    public Map<Integer, String> addSports(@MapKeyType(Integer.class) @MapValueType(String.class) Map<Integer, String> mapOfSportsName) {
        return mapOfSportsName;
    }
}

// Usage
Map<Integer, String> sportsMap = (Map<Integer, String>) processor.processSingleAction("Vishal like cricket its number one, then comes soccer which is number 3, since in between he likes chess as well");

@Agent
public class CustomerWithQueryAction {
    @Action(description = "Customer has problem create ticket for him")
    public String customerComplaintWithNameandFullInfo(Customer customer, @Prompt(dateFormat = "yyyy-MM-dd") Date dateOfComp, @Prompt(describe = "this is customer complaint") String query) {
        return customer.toString();
    }
}

// Process multiple actions in sequence
String result1 = (String) processor.processSingleAction("vishal wants to travel and wants adventure");
String result2 = (String) processor.processSingleAction("vishals computer is broken he needs to fix it");

ScriptProcessor script = new ScriptProcessor(processor);
String commands = "Vishal wants to eat food first find his food preference\n then find his favorite restaurants and book a table for dinner tonight at 7 PM";
ScriptResult result = script.process(commands);

ScriptProcessor script = new ScriptProcessor(processor);
String complexScript = "can you reserve the flight for Vishal from Toronto to Bangalore for 3 Days on 7th december " +
    "If flight booking is successful, can you reserve the car for Vishal from Bangalore to Toronto for 10 Days on 17th december " +
    "if car booking is successful and flight cost are less than $1000 then book the sight seeing attraction called 5 star palace " +
    "if car booking is successful and flight cost are more than $1000 then book the sight seeing attraction called peanut palace";
ScriptResult result = script.process(complexScript);
String summary = script.summarize(result);

PromptTransformer transformer = PredictionLoader.getInstance().createOrGetPromptTransformer();

// Transform to POJO
Organization org = (Organization) transformer.transformIntoPojo(promptText, Organization.class.getName(), "", "");

// Transform to JSON
String jsonString = transformer.transformIntoJson(jsonTemplate, promptText, "Customer", "get Customer details");

@Agent
public class CustomExample {
    @Action
    public Organization addOrganization(Organization organization) {
        return organization;
    }
}

@Agent(actionName = "restartTheECOMServer", description = "will be used to restart the server", riskLevel = ActionRisk.HIGH, groupName = "customer support", groupDescription = "actions related to customer support")
public class ServerRestartAction implements JavaMethodAction {
    public String restartTheECOMServer(String reasonForRestart, String requestedBy) {
        return "Server has been restarted by " + requestedBy + " due to following reason " + reasonForRestart;
    }
}

GeminiImageActionProcessor processor = new GeminiImageActionProcessor();
String imageDescription = processor.imageToText(imageBytes);
Object result = actionProcessor.processSingleAction(imageDescription);

SeleniumProcessor processor = new SeleniumProcessor(driver);
processor.processWebAction("go to website https://the-internet.herokuapp.com");
boolean buttonPresent = processor.trueFalseQuery("do you see Add/Remove Elements?");

SpringOpenAIProcessor springProcessor = new SpringOpenAIProcessor(applicationContext);

@NoArgsConstructor
@AllArgsConstructor
@Getter
@Setter
@ToString
public class Organization {
    private String name;
    @ListType(Employee.class)
    private List<Employee> em;
    @ListType(String.class)
    private List<String> locations;
}

@NoArgsConstructor
@AllArgsConstructor
@Getter
@Setter
@ToString
public class Customer {
    private String firstName;
    private String lastName;
    @Prompt(describe = "convert this to Hindi")
    private String reasonForCalling;
    @Prompt(ignore = true)
    private String location;
    @Prompt(dateFormat = "yyyy-MM-dd", describe = "if you dont find date provide todays date")
    private Date dateJoined;
}

System.setProperty("tools4ai.properties.path", "path/to/tools4ai.properties");