TextDoctor / main_demo.py
FurqanIshaq's picture
Update main_demo.py
36935a5 verified
raw
history blame
672 Bytes
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)