From 1ae1410a30a1cd70b8e9578afb7746f268c8571c Mon Sep 17 00:00:00 2001
From: Rodrigo <rodrigosilvanader@gmail.com>
Date: Tue, 18 Jun 2024 10:58:13 -0300
Subject: [PATCH] Add conditional router

---
 .../experimental/ConditionalRouter.py         | 87 +++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 src/backend/base/langflow/components/experimental/ConditionalRouter.py

diff --git a/src/backend/base/langflow/components/experimental/ConditionalRouter.py b/src/backend/base/langflow/components/experimental/ConditionalRouter.py
new file mode 100644
index 000000000..77c06d158
--- /dev/null
+++ b/src/backend/base/langflow/components/experimental/ConditionalRouter.py
@@ -0,0 +1,87 @@
+from langflow.custom import Component
+from langflow.inputs import BoolInput, DropdownInput, TextInput, MessageInput
+from langflow.template import Output
+from langflow.schema.message import Message
+
+
+class ConditionalRouterComponent(Component):
+    display_name = "Conditional Router"
+    description = "Routes an input message to a corresponding output based on text comparison."
+    icon = "equal"
+
+    inputs = [
+        TextInput(
+            name="input_text",
+            display_name="Input Text",
+            info="The primary text input for the operation.",
+        ),
+        TextInput(
+            name="match_text",
+            display_name="Match Text",
+            info="The text input to compare against.",
+        ),
+        DropdownInput(
+            name="operator",
+            display_name="Operator",
+            options=["equals", "not equals", "contains", "starts with", "ends with"],
+            info="The operator to apply for comparing the texts.",
+            value="equals",
+            advanced=True
+        ),
+        BoolInput(
+            name="case_sensitive",
+            display_name="Case Sensitive",
+            info="If true, the comparison will be case sensitive.",
+            value=False,
+            advanced=True,
+        ),
+        MessageInput(
+            name="message",
+            display_name="Message",
+            info="The message to pass through either route.",
+        ),
+    ]
+
+    outputs = [
+        Output(display_name="True Route", name="true_result", method="true_response"),
+        Output(display_name="False Route", name="false_result", method="false_response"),
+    ]
+
+    def evaluate_condition(self, input_text: str, match_text: str, operator: str, case_sensitive: bool) -> bool:
+        if not case_sensitive:
+            input_text = input_text.lower()
+            match_text = match_text.lower()
+
+        if operator == "equals":
+            return input_text == match_text
+        elif operator == "not equals":
+            return input_text != match_text
+        elif operator == "contains":
+            return match_text in input_text
+        elif operator == "starts with":
+            return input_text.startswith(match_text)
+        elif operator == "ends with":
+            return input_text.endswith(match_text)
+        return False
+
+    def true_response(self) -> Message:
+        result = self.evaluate_condition(self.input_text, self.match_text, self.operator, self.case_sensitive)
+        if result:
+            self.stop("false_result")
+            response = self.message if self.message else self.input_text
+            self.status = response
+            return response
+        else:
+            self.stop("true_result")
+            return ""
+
+    def false_response(self) -> Message:
+        result = self.evaluate_condition(self.input_text, self.match_text, self.operator, self.case_sensitive)
+        if not result:
+            self.stop("true_result")
+            response = self.message if self.message else self.input_text
+            self.status = response
+            return response
+        else:
+            self.stop("false_result")
+            return ""