From 41b0660dfd6cb01640e8d10b0694da88de750da4 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Wed, 5 Jun 2024 16:51:30 -0300
Subject: [PATCH] Styled Output Type

---
 poetry.lock                                   | 45 ++++++++++++++++++-
 pyproject.toml                                |  1 +
 .../components/OutputComponent/index.tsx      | 23 ++++++----
 3 files changed, 60 insertions(+), 9 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index 5d85d6b81..521a9e42b 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -2488,6 +2488,38 @@ benchmarks = ["httplib2", "httpx", "requests", "urllib3"]
 dev = ["dpkt", "pytest", "requests"]
 examples = ["oauth2"]
 
+[[package]]
+name = "gitdb"
+version = "4.0.11"
+description = "Git Object Database"
+optional = false
+python-versions = ">=3.7"
+files = [
+    {file = "gitdb-4.0.11-py3-none-any.whl", hash = "sha256:81a3407ddd2ee8df444cbacea00e2d038e40150acfa3001696fe0dcf1d3adfa4"},
+    {file = "gitdb-4.0.11.tar.gz", hash = "sha256:bf5421126136d6d0af55bc1e7c1af1c397a34f5b7bd79e776cd3e89785c2b04b"},
+]
+
+[package.dependencies]
+smmap = ">=3.0.1,<6"
+
+[[package]]
+name = "gitpython"
+version = "3.1.43"
+description = "GitPython is a Python library used to interact with Git repositories"
+optional = false
+python-versions = ">=3.7"
+files = [
+    {file = "GitPython-3.1.43-py3-none-any.whl", hash = "sha256:eec7ec56b92aad751f9912a73404bc02ba212a23adb2c7098ee668417051a1ff"},
+    {file = "GitPython-3.1.43.tar.gz", hash = "sha256:35f314a9f878467f5453cc1fee295c3e18e52f1b99f10f6cf5b1682e968a9e7c"},
+]
+
+[package.dependencies]
+gitdb = ">=4.0.1,<5"
+
+[package.extras]
+doc = ["sphinx (==4.3.2)", "sphinx-autodoc-typehints", "sphinx-rtd-theme", "sphinxcontrib-applehelp (>=1.0.2,<=1.0.4)", "sphinxcontrib-devhelp (==1.0.2)", "sphinxcontrib-htmlhelp (>=2.0.0,<=2.0.1)", "sphinxcontrib-qthelp (==1.0.3)", "sphinxcontrib-serializinghtml (==1.1.5)"]
+test = ["coverage[toml]", "ddt (>=1.1.1,!=1.4.3)", "mock", "mypy", "pre-commit", "pytest (>=7.3.1)", "pytest-cov", "pytest-instafail", "pytest-mock", "pytest-sugar", "typing-extensions"]
+
 [[package]]
 name = "google-ai-generativelanguage"
 version = "0.6.4"
@@ -8151,6 +8183,17 @@ files = [
     {file = "six-1.16.0.tar.gz", hash = "sha256:1e61c37477a1626458e36f7b1d82aa5c9b094fa4802892072e49de9c60c4c926"},
 ]
 
+[[package]]
+name = "smmap"
+version = "5.0.1"
+description = "A pure Python implementation of a sliding window memory map manager"
+optional = false
+python-versions = ">=3.7"
+files = [
+    {file = "smmap-5.0.1-py3-none-any.whl", hash = "sha256:e6d8668fa5f93e706934a62d7b4db19c8d9eb8cf2adbb75ef1b675aa332b69da"},
+    {file = "smmap-5.0.1.tar.gz", hash = "sha256:dceeb6c0028fdb6734471eb07c0cd2aae706ccaecab45965ee83f11c8d3b1f62"},
+]
+
 [[package]]
 name = "sniffio"
 version = "1.3.1"
@@ -10054,4 +10097,4 @@ local = ["ctransformers", "llama-cpp-python", "sentence-transformers"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.10,<3.13"
-content-hash = "83c94ed0fa28b968553221385251b871139a7440ab0420f867efbe16568b8411"
+content-hash = "8fd6622a9bdd88dcac9ab9ca8136b6b7c7939f0abeb6cd3f4ee754cc38527648"
diff --git a/pyproject.toml b/pyproject.toml
index 18be560c7..dd3057bff 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -86,6 +86,7 @@ youtube-transcript-api = "^0.6.2"
 markdown = "^3.6"
 langchain-chroma = "^0.1.1"
 upstash-vector = "^0.4.0"
+gitpython = "^3.1.43"
 
 
 [tool.poetry.group.dev.dependencies]
diff --git a/src/frontend/src/customNodes/genericNode/components/OutputComponent/index.tsx b/src/frontend/src/customNodes/genericNode/components/OutputComponent/index.tsx
index dfa5ac350..f684a902f 100644
--- a/src/frontend/src/customNodes/genericNode/components/OutputComponent/index.tsx
+++ b/src/frontend/src/customNodes/genericNode/components/OutputComponent/index.tsx
@@ -11,6 +11,7 @@ import useFlowStore from "../../../../stores/flowStore";
 import { outputComponentType } from "../../../../types/components";
 import { NodeDataType } from "../../../../types/flow";
 import { cn } from "../../../../utils/utils";
+import { Button } from "../../../../components/ui/button";
 
 export default function OutputComponent({
   selected,
@@ -28,16 +29,21 @@ export default function OutputComponent({
   }
 
   return (
-    <div className="nocopy nopan nodelete nodrag noundo flex gap-2 ">
-      <span>{name}</span>
+    <div className="nocopy nopan nodelete nodrag noundo flex items-center gap-2 ">
       <DropdownMenu>
-        <DropdownMenuTrigger>
-          <span
-            className={cn(frozen ? " text-ice" : "", "flex items-center gap-1")}
+        <DropdownMenuTrigger asChild>
+          <Button
+            disabled={frozen}
+            variant="primary"
+            size="xs"
+            className={cn(
+              frozen ? "text-ice" : "",
+              "items-center gap-1 pl-2 pr-1.5 align-middle text-xs font-normal",
+            )}
           >
-            {selected}
-            <ForwardedIconComponent name="ChevronDown" className="h-4 w-4" />
-          </span>
+            <span className="pb-px">{selected}</span>
+            <ForwardedIconComponent name="ChevronDown" className="h-3 w-3" />
+          </Button>
         </DropdownMenuTrigger>
         <DropdownMenuContent>
           {types.map((type) => (
@@ -58,6 +64,7 @@ export default function OutputComponent({
           ))}
         </DropdownMenuContent>
       </DropdownMenu>
+      <span>{name}</span>
     </div>
   );
 }