Uploading Python Code as Custom Component in Langflow

[Tuguldur-Tserenbaljir]

Hi,

I’m currently working on a project with Langflow and have a Python script titled "Sum of Two Numbers" that I copied from Langflow’s own examples. I know that the Python code works properly with Langflow, as there are no issues on that front.

I also understand that Langflow allows the uploading of JSON files as custom components. This got me wondering if it’s possible to upload Python code as a custom component as well. I tried to upload the script, but I’m receiving the following response: Response details: {"detail":"Method Not Allowed"}.

Could you confirm if uploading Python code is supported? If not, is there an alternative approach to integrating Python code into Langflow?

def upload_python_component(
    python_file_path: str,
    flow_id: str,
    position: Dict[str, int] = {"x": 100, "y": 100},
    langflow_api_url: str = "http://localhost:7860/api"
) -> Optional[dict]:
    """
    Upload a Python custom component to Langflow
    
    Args:
        python_file_path (str): Path to the Python file
        flow_id (str): ID of the existing flow
        position (dict): Position where to place the component (x, y coordinates)
        langflow_api_url (str): Base Langflow API URL
        
    Returns:
        dict: Response from the API
    """
    print("\n" + "="*50)
    print(f"🚀 Uploading Python component at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
    print("="*50)
    
    # Read the Python file
    print(f"\n📂 Reading Python file from: {python_file_path}")
    with open(python_file_path, 'r') as file:
        python_code = file.read()
        
    # Encode the Python code in base64
    encoded_content = base64.b64encode(python_code.encode()).decode()
    
    # Create the component upload payload
    upload_payload = {
        "code": encoded_content,
        "file_name": python_file_path.split('/')[-1]
    }
    
    # First, upload the custom component
    upload_endpoint = f"{langflow_api_url.rstrip('/')}/custom_components/upload"
    print(f"\n📡 Uploading component to: {upload_endpoint}")
    
    upload_response = requests.post(
        upload_endpoint,
        json=upload_payload,
        headers={'Content-Type': 'application/json'}
    )
    upload_response.raise_for_status()
    
    # Get the component details from the response
    component_info = upload_response.json()
    
    # Now add the component to the flow
    flow_endpoint = f"{langflow_api_url.rstrip('/')}/flows/{flow_id}"
    print(f"\n📡 Fetching existing flow from: {flow_endpoint}")
    
    response = requests.get(flow_endpoint)
    response.raise_for_status()
    flow_data = response.json()
    
    # Create the node for the custom component
    component_type = component_info.get("name", "CustomComponent")
    new_id = f"custom-{str(uuid4())[:6]}"
    
    new_node = {
        "id": new_id,
        "type": "genericNode",
        "position": position,
        "data": {
            "node": {
                "template": component_info.get("template", {}),
                "description": component_info.get("description", "Custom Python Component"),
                "base_classes": component_info.get("base_classes", []),
                "display_name": component_info.get("display_name", component_type),
                "custom_component": True
            },
            "id": new_id,
            "type": component_type,
            "value": None
        }
    }
    
    # Add the component to the flow's data
    if "data" in flow_data and "nodes" in flow_data["data"]:
        flow_data["data"]["nodes"].append(new_node)
    else:
        print("\n❌ Error: Invalid flow data structure")
        return None
    
    # Update the flow with the new component
    print(f"\n📡 Updating flow with new component")
    update_endpoint = f"{langflow_api_url.rstrip('/')}/flows/{flow_id}"
    update_response = requests.patch(
        update_endpoint,
        json=flow_data,
        headers={'Content-Type': 'application/json'}
    )
    
    update_response.raise_for_status()
    return update_response.json()

if __name__ == "__main__":
    # Example usage
    python_file_path = r"C:\\Users\\btuud\\Downloads\\Component_generator\\sum.py"  # Path to your Python component file
    flow_id = "793f7bed-10ee-4343-a0eb-bc14ac49000b"  # Your flow ID
    
    # Optionally specify position (default is x:100, y:100)
    custom_position = {
        "x": 200,
        "y": 150
    }