# intent_parser.py - Extracts purpose and domain of the robotics app idea

import os
import google.generativeai as genai

# Use Gemini API Key from environment variable
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
if not GEMINI_API_KEY:
    raise EnvironmentError("GEMINI_API_KEY not set in environment variables.")

genai.configure(api_key=GEMINI_API_KEY)
model = genai.GenerativeModel("gemini-pro")

# Categories the system understands
INTENT_CATEGORIES = [
    "educational",
    "assistive",
    "entertainment",
    "industrial",
    "home automation",
    "healthcare",
    "retail",
    "creative"
]

# Classify robot idea using Gemini

def classify_robot_idea(user_input: str) -> str:
    prompt = f"""
    Classify this user idea into one of the following categories:
    {', '.join(INTENT_CATEGORIES)}.
    Only return the category word. If none fits, return 'creative'.

    Idea: {user_input}
    Category:
    """

    response = model.generate_content(prompt)
    return response.text.strip().lower()

# Example
if __name__ == "__main__":
    idea = "Build a robot that reminds elderly people to take medicine."
    print("Predicted Intent:", classify_robot_idea(idea))