from graph_pipeline import textdoctor_app, TextDoctorState

if __name__ == "__main__":
    try:
        print("=== TextDoctor CLI ===")
        user_text = input("\nEnter your text: ").strip()

        if not user_text:
            print("❌ No text entered.")
            exit()

        state: TextDoctorState = {"input_text": user_text}
        final_state = textdoctor_app.invoke(state)

        print("\n--- Final Output ---")
        print(final_state["review_result"]["final_text"])

        print("\n--- Explanation ---")
        print(final_state["review_result"]["decision_explanation"])

    except Exception as e:
        print("\n❌ An error occurred:", e)