allignement

scrapegraphai/__init__.py

@@ -1,3 +1,8 @@
 """
 __init__.py file for scrapegraphai folder
 """
+
+from .utils.logging import get_logger, set_verbosity_info
+
+logger = get_logger(__name__)
+set_verbosity_info()

scrapegraphai/graphs/abstract_graph.py

@@ -5,6 +5,7 @@
 import asyncio
 import uuid
 import warnings
+import time
 from abc import ABC, abstractmethod
 from typing import Optional, Type
 
@@ -14,8 +15,13 @@
 
 from ..helpers import models_tokens
 from ..models import CLoD, DeepSeek, OneApi, XAI
-from ..utils.logging import set_verbosity_info, set_verbosity_warning
+from ..utils.logging import set_verbosity_info, set_verbosity_warning, get_logger
+from ..telemetry import log_graph_execution
 
+logger = get_logger(__name__)
+
+# ANSI escape sequence for hyperlink
+CLICKABLE_URL = "\033]8;;https://scrapegraphai.com\033\\https://scrapegraphai.com\033]8;;\033\\"
 
 class AbstractGraph(ABC):
     """
@@ -316,6 +322,10 @@ def run(self) -> str:
         """
         Abstract method to execute the graph and return the result.
         """
+        inputs = {"user_prompt": self.prompt, self.input_key: self.source}
+        self.final_state, self.execution_info = self.graph.execute(inputs)
+        result = self.final_state.get("answer", "No answer found.")
+        return result
 
     async def run_safe_async(self) -> str:
         """

scrapegraphai/graphs/base_graph.py

@@ -8,7 +8,12 @@
 
 from ..telemetry import log_graph_execution
 from ..utils import CustomLLMCallbackManager
+from ..utils.logging import get_logger
 
+logger = get_logger(__name__)
+
+# ANSI escape sequence for hyperlink
+CLICKABLE_URL = "\033]8;;https://scrapegraphai.com\033\\https://scrapegraphai.com\033]8;;\033\\"
 
 class BaseGraph:
     """
@@ -353,9 +358,24 @@ def execute(self, initial_state: dict) -> Tuple[dict, list]:
 
             bridge = BurrBridge(self, self.burr_config)
             result = bridge.execute(initial_state)
-            return (result["_state"], [])
+            state, exec_info = (result["_state"], [])
         else:
-            return self._execute_standard(initial_state)
+            state, exec_info = self._execute_standard(initial_state)
+
+        # Print the result first
+        if "answer" in state:
+            print(state["answer"])
+        elif "parsed_doc" in state:
+            print(state["parsed_doc"])
+        elif "generated_code" in state:
+            print(state["generated_code"])
+        elif "merged_script" in state:
+            print(state["merged_script"])
+
+        # Then show the message ONLY ONCE
+        print(f"✨ Try enhanced version of ScrapegraphAI at {CLICKABLE_URL} ✨")
+
+        return state, exec_info
 
     def append_node(self, node):
         """