Update main_demo.py

main_demo.py

@@ -1,13 +1,22 @@
-from graph_pipeline import textdoctor_app, TextDoctorState
-
-if __name__ == "__main__":
-    state: TextDoctorState = {
-        "input_text": "bro i need 2 more days for this, hope thats okay, i had some issues so couldnt finish on time"
-    }
-
-    final_state = textdoctor_app.invoke(state)
-
-    print("Original:", final_state["input_text"])
-    print("\nFinal:", final_state["review_result"]["final_text"])
-    print("\nDecision explanation:")
-    print(final_state["review_result"]["decision_explanation"])
+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)