// This is an autogenerated file from Firebase Studio.
'use server';

/**
 * @fileOverview This file defines a Genkit flow for classifying the complexity of project features.
 *
 * - classifyComplexity - An async function that takes a feature description and returns its complexity classification.
 * - ClassifyComplexityInput - The input type for the classifyComplexity function.
 * - ClassifyComplexityOutput - The output type for the classifyComplexity function.
 */

import {ai} from '@/ai/genkit';
import {z} from 'genkit';

const ClassifyComplexityInputSchema = z.object({
  featureDescription: z
    .string()
    .describe('A detailed description of the feature to be classified.'),
});
export type ClassifyComplexityInput = z.infer<typeof ClassifyComplexityInputSchema>;

const ClassifyComplexityOutputSchema = z.object({
  complexity: z
    .enum(['simple', 'medium', 'complex'])
    .describe('The complexity classification of the feature.'),
});
export type ClassifyComplexityOutput = z.infer<typeof ClassifyComplexityOutputSchema>;

export async function classifyComplexity(
  input: ClassifyComplexityInput
): Promise<ClassifyComplexityOutput> {
  return classifyComplexityFlow(input);
}

const prompt = ai.definePrompt({
  name: 'classifyComplexityPrompt',
  input: {schema: ClassifyComplexityInputSchema},
  output: {schema: ClassifyComplexityOutputSchema},
  prompt: `You are an expert project manager classifying the complexity of software features. Classify the following feature description as either 'simple', 'medium', or 'complex'.\n\nFeature Description: {{{featureDescription}}}\n\nComplexity:`,
});

const classifyComplexityFlow = ai.defineFlow(
  {
    name: 'classifyComplexityFlow',
    inputSchema: ClassifyComplexityInputSchema,
    outputSchema: ClassifyComplexityOutputSchema,
  },
  async input => {
    const {output} = await prompt(input);
    return output!;
  }
);