fix: improve live updater control flow with error handling

Author: lemorage

examples/live_updates/main.py

@@ -1,16 +1,9 @@
-import dataclasses
 import datetime
 
 import cocoindex
 from dotenv import load_dotenv
 
 
-@dataclasses.dataclass
-class Document:
-    filename: str
-    content: str
-
-
 @cocoindex.flow_def(name="LiveUpdates")
 def live_update_flow(
     flow_builder: cocoindex.FlowBuilder, data_scope: cocoindex.DataScope
@@ -49,7 +42,10 @@ def main():
         print("Live updater started. Watching for changes in the 'data' directory.")
         print("Try adding, modifying, or deleting files in the 'data' directory.")
         print("Press Ctrl+C to stop.")
-        updater.wait()
+        try:
+            updater.wait()
+        except KeyboardInterrupt:
+            print("Stopping live updater...")
 
 
 if __name__ == "__main__":