app.py

import sys
from core import CodeExplainer, LLMService, PromptManager, FileManager
from enums.models import OllamaModel
import questionary # type: ignore

def model_switcher() -> OllamaModel:
    escolhas = [model.value for model in OllamaModel]

    resposta = questionary.select(
        "🧠 Selecione o modelo desejado:",
        choices=escolhas
    ).ask()

    for model in OllamaModel:
        if model.value == resposta:
            return model

    return OllamaModel.CODELLAMA_LATEST

def app():
    if len(sys.argv) != 2:
        print("Uso: python main.py <arquivo>")
        sys.exit(1)

    file_path = sys.argv[1]
    code = FileManager.read_file(file_path)

    modelo_escolhido = model_switcher()
    print(f"\n✅ Modelo selecionado: {modelo_escolhido.value}\n")

    explainer = CodeExplainer(
        llm_service=LLMService(model_name=modelo_escolhido.value),
        prompt_manager=PromptManager()
    )

    explainer.explain_code(code)

if __name__ == "__main__":
    app()