Refactor custom component code

src/backend/langflow/template/frontend_node/custom_components.py

@@ -1,10 +1,48 @@
 from typing import Optional
 
-from langflow.interface.custom.constants import DEFAULT_CUSTOM_COMPONENT_CODE
 from langflow.template.field.base import TemplateField
 from langflow.template.frontend_node.base import FrontendNode
 from langflow.template.template.base import Template
 
+DEFAULT_CUSTOM_COMPONENT_CODE = """from langflow import CustomComponent
+from typing import Optional, List, Dict, Union
+from langflow.field_typing import (
+    AgentExecutor,
+    BaseChatMemory,
+    BaseLanguageModel,
+    BaseLLM,
+    BaseLoader,
+    BaseMemory,
+    BaseOutputParser,
+    BasePromptTemplate,
+    BaseRetriever,
+    Callable,
+    Chain,
+    ChatPromptTemplate,
+    Data,
+    Document,
+    Embeddings,
+    NestedDict,
+    Object,
+    PromptTemplate,
+    TextSplitter,
+    Tool,
+    VectorStore,
+)
+
+
+class Component(CustomComponent):
+    display_name: str = "Custom Component"
+    description: str = "Create any custom component you want!"
+
+    def build_config(self):
+        return {"param": {"display_name": "Parameter"}}
+
+    def build(self, param: Data) -> Data:
+        return param
+
+"""
+
 
 class CustomComponentFrontendNode(FrontendNode):
     name: str = "CustomComponent"