reorg content graph

open_notebook/graphs/content_processing/text.py

@@ -0,0 +1,28 @@
+from loguru import logger
+
+from open_notebook.graphs.content_processing.state import SourceState
+
+
+def extract_txt(state: SourceState):
+    """
+    Parse the text file and print its content.
+    """
+    return_dict = {}
+    if (
+        state.get("file_path") is not None
+        and state.get("identified_type") == "text/plain"
+    ):
+        logger.debug(f"Extracting text from {state.get('file_path')}")
+        file_path = state.get("file_path")
+        if file_path is not None:
+            try:
+                with open(file_path, "r", encoding="utf-8") as file:
+                    content = file.read()
+                    logger.debug(f"Extracted: {content[:100]}")
+                    return_dict["content"] = content
+            except FileNotFoundError:
+                raise FileNotFoundError(f"File not found at {file_path}")
+            except Exception as e:
+                raise Exception(f"An error occurred: {e}")
+
+    return return_dict