"""
Python Code Generating Wizard
Deterministic, extensible, Codex-compatible
"""

from typing import List

class CodeWizard:
    def __init__(self):
        self.spec = {}

    def ask(self, prompt: str, options: List[str] | None = None) -> str:
        print("\n" + prompt)
        if options:
            for i, opt in enumerate(options, 1):
                print(f"  {i}. {opt}")
            choice = int(input("Select option: ")) - 1
            return options[choice]
        return input("> ")

    def collect_intent(self):
        self.spec["program_type"] = self.ask(
            "What type of program do you want?",
            ["function", "class", "script"]
        )

        self.spec["task"] = self.ask(
            "Describe what the program should do (one sentence):"
        )

        self.spec["inputs"] = self.ask(
            "What inputs does it take? (comma separated, or 'none')"
        )

        self.spec["outputs"] = self.ask(
            "What should it return or produce?"
        )

        self.spec["logic"] = self.ask(
            "Select a logic pattern:",
            ["calculation", "iteration", "condition", "data processing"]
        )

    def generate_code(self) -> str:
        name = "generated_function"

        inputs = self.spec["inputs"]
        params = "" if inputs == "none" else inputs

        body = "    pass\n"

        if self.spec["logic"] == "calculation":
            body = "    result = 0\n    return result\n"
        elif self.spec["logic"] == "iteration":
            body = "    for item in []:\n        pass\n"
        elif self.spec["logic"] == "condition":
            body = "    if True:\n        pass\n"
        elif self.spec["logic"] == "data processing":
            body = "    data = []\n    return data\n"

        if self.spec["program_type"] == "function":
            return f"""def {name}({params}):
    \"\"\"{self.spec['task']}\"\"\"
{body}"""

        if self.spec["program_type"] == "class":
            return f"""class GeneratedClass:
    \"\"\"{self.spec['task']}\"\"\"

    def __init__(self):
        pass
"""

        return f"""# {self.spec['task']}
def main():
{body}

if __name__ == "__main__":
    main()
"""

    def run(self):
        print("🧙 Python Code Generating Wizard")
        self.collect_intent()
        code = self.generate_code()
        print("\n=== GENERATED CODE ===\n")
        print(code)
        return code


if __name__ == "__main__":
    CodeWizard().run()