jonaszbigda/Conceptor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Upgraded to new Comfy API

Browse Source
This commit is contained in:
Jonasz Bigda
2025-08-28 22:50:08 +02:00
parent 4b41ba82e8
commit d72a3ad054
3 changed files with 55 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
Content/Conceptor.uasset
View File

Binary file not shown.

69
Source/Conceptor/Private/ConceptorWidgetBase.cpp
View File

@@ -388,7 +388,6 @@ void UConceptorWidgetBase::UpdateModels() {
FHttpModule* Http = &FHttpModule::Get(); FHttpModule* Http = &FHttpModule::Get();
TSharedRef<IHttpRequest, ESPMode::ThreadSafe> Request = Http->CreateRequest(); TSharedRef<IHttpRequest, ESPMode::ThreadSafe> Request = Http->CreateRequest();
Checkpoints.Empty(); Checkpoints.Empty();
ControlNets.Empty();
Request->OnProcessRequestComplete().BindLambda([this](FHttpRequestPtr Request, FHttpResponsePtr Response, bool bWasSuccessful) Request->OnProcessRequestComplete().BindLambda([this](FHttpRequestPtr Request, FHttpResponsePtr Response, bool bWasSuccessful)
{ {
@@ -400,23 +399,20 @@ void UConceptorWidgetBase::UpdateModels() {
if (FJsonSerializer::Deserialize(JsonReader, OutValue)) { if (FJsonSerializer::Deserialize(JsonReader, OutValue)) {
MainObject = OutValue->AsObject(); MainObject = OutValue->AsObject();
TArray<TSharedPtr<FJsonValue>> FilesArray = MainObject->GetArrayField("files"); TSharedPtr<FJsonObject> CheckpointLoader = MainObject->GetObjectField("CheckpointLoaderSimple");
TSharedPtr<FJsonObject> InputField = CheckpointLoader->GetObjectField("input");
TSharedPtr<FJsonObject> RequiredField = InputField->GetObjectField("required");
TArray<TSharedPtr<FJsonValue>> CkeckpointsArray = RequiredField->GetArrayField("ckpt_name");
for (TSharedPtr<FJsonValue>& Value : FilesArray) { for (TSharedPtr<FJsonValue>& Checkpoint : CkeckpointsArray) {
TSharedPtr<FJsonObject> FileObject = Value->AsObject(); TArray<TSharedPtr<FJsonValue>> FileObject = Checkpoint->AsArray();
FString FileName; if (FileObject.IsEmpty()) continue;
bool IsCheckpointsFile = FileObject->GetStringField("path").Split("checkpoints\\", nullptr, &FileName);
bool IsControlNetFile = FileObject->GetStringField("path").Split("controlnet\\", nullptr, &FileName);
if ((IsCheckpointsFile || IsControlNetFile) && (FileName.Contains(".safetensors") || FileName.Contains(".pth"))) { FString FileName = FileObject[0]->AsString();
if (IsCheckpointsFile) {
Checkpoints.Add(FileName); if (FileName.Contains(".safetensors") || FileName.Contains(".pth")) {
UE_LOG(LogTemp, Warning, TEXT("%s added to Checkpoints"), *FileName); Checkpoints.Add(FileName);
} UE_LOG(LogTemp, Warning, TEXT("%s added to Checkpoints"), *FileName);
else {
ControlNets.Add(FileName);
UE_LOG(LogTemp, Warning, TEXT("%s added to ControlNets"), *FileName);
}
} }
} }
} }
@@ -424,7 +420,46 @@ void UConceptorWidgetBase::UpdateModels() {
OnModelsUpdated(); OnModelsUpdated();
}); });
Request->SetURL("http://127.0.0.1:8188/internal/files?directory=models"); Request->SetURL("http://127.0.0.1:8188/api/object_info");
Request->SetVerb("GET");
Request->ProcessRequest();
}
void UConceptorWidgetBase::UpdateControlNets() {
FHttpModule* Http = &FHttpModule::Get();
TSharedRef<IHttpRequest, ESPMode::ThreadSafe> Request = Http->CreateRequest();
ControlNets.Empty();
Request->OnProcessRequestComplete().BindLambda([this](FHttpRequestPtr Request, FHttpResponsePtr Response, bool bWasSuccessful)
{
if (bWasSuccessful && Response.IsValid())
{
TSharedRef<TJsonReader<TCHAR>> JsonReader = TJsonReaderFactory<TCHAR>::Create(Response->GetContentAsString());
TSharedPtr<FJsonValue> OutValue;
TSharedPtr<FJsonObject> MainObject;
if (FJsonSerializer::Deserialize(JsonReader, OutValue)) {
MainObject = OutValue->AsObject();
TSharedPtr<FJsonObject> CNLoader = MainObject->GetObjectField("ControlNetLoader");
TSharedPtr<FJsonObject> InputField = CNLoader->GetObjectField("input");
TSharedPtr<FJsonObject> RequiredField = InputField->GetObjectField("required");
TArray<TSharedPtr<FJsonValue>> CNArray = RequiredField->GetArrayField("control_net_name");
TArray<TSharedPtr<FJsonValue>> FilenameArray = CNArray[0]->AsArray();
for (TSharedPtr<FJsonValue>& ControlNet : FilenameArray) {
FString FileName = ControlNet->AsString();
if (FileName.Contains(".safetensors") || FileName.Contains(".pth")) {
ControlNets.Add(FileName);
UE_LOG(LogTemp, Warning, TEXT("%s added to ControlNets"), *FileName);
}
}
}
}
OnModelsUpdated();
});
Request->SetURL("http://127.0.0.1:8188/api/object_info");
Request->SetVerb("GET"); Request->SetVerb("GET");
Request->ProcessRequest(); Request->ProcessRequest();
} }

3
Source/Conceptor/Public/ConceptorWidgetBase.h
View File

@@ -154,6 +154,9 @@ public:
UFUNCTION(BlueprintCallable, Category = "Conceptor") UFUNCTION(BlueprintCallable, Category = "Conceptor")
void UpdateModels(); void UpdateModels();
UFUNCTION(BlueprintCallable, Category = "Conceptor")
void UpdateControlNets();
UFUNCTION(BlueprintNativeEvent, Category = "Conceptor") UFUNCTION(BlueprintNativeEvent, Category = "Conceptor")
void OnModelsUpdated(); void OnModelsUpdated();
virtual void OnModelsUpdated_Implementation(); virtual void OnModelsUpdated_Implementation();