Various fixes, admin ui for media providers now able to add new provi…

…ders.
ServiceStack · Oct 9, 2024 · 5378462 · 5378462
1 parent a65dac9
commit 5378462
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 6 deletions.
diff --git a/AiServer.ServiceInterface/MediaTransform/CreateMediaTransformCommand.cs b/AiServer.ServiceInterface/MediaTransform/CreateMediaTransformCommand.cs
@@ -354,7 +354,10 @@ async Task DownloadOutputAsync(IMediaTransformProvider aiProvider,
                 "video/mp4" => "mp4",
                 "video/webm" => "webm",
                 "video/ogg" => "ogg",
-                _ => "webm"
+                "video/mov" => "mov",
+                "video/avi" => "avi",
+                "video/mkv" => "mkv",
+                _ => "mp4"
             };
 
             var imageBytes = await response.Content.ReadAsByteArrayAsync();

diff --git a/AiServer.ServiceInterface/VideoServices.cs b/AiServer.ServiceInterface/VideoServices.cs
@@ -198,8 +198,8 @@ private bool IsVideoFormat(MediaOutputFormat outputFormat)
             case MediaOutputFormat.MKV:
             case MediaOutputFormat.MOV:
             case MediaOutputFormat.WebM:
-                return true;
             case MediaOutputFormat.GIF:
+                return true;
             case MediaOutputFormat.MP3:
             case MediaOutputFormat.WAV:
             case MediaOutputFormat.FLAC:

diff --git a/AiServer.ServiceModel/GenerationAdmin.cs b/AiServer.ServiceModel/GenerationAdmin.cs
@@ -57,7 +57,7 @@ public class CreateMediaProvider : ICreateDb<MediaProvider>, IReturn<IdResponse>
     public List<string>? Models { get; set; }
 
     [Input(Type = "hidden")]
-    public int? MediaTypeId { get; set; }
+    public string MediaTypeId { get; set; }
 }
 
 [Tag(Tag.Info)]

diff --git a/AiServer/wwwroot/mjs/components/MediaProviders.mjs b/AiServer/wwwroot/mjs/components/MediaProviders.mjs
@@ -127,8 +127,11 @@ const SelectModels = {
                 }
                 return;
             } else {
-                supportedModels.value = Object.keys(props.providerType?.apiModels, [])
-                qualifiedModelMappings.value = props.providerType?.apiModels
+                if (props.providerType?.apiModels) {
+                    console.log(props.providerType?.apiModels)
+                    supportedModels.value = Object.keys(props.providerType?.apiModels, [])
+                    qualifiedModelMappings.value = props.providerType?.apiModels
+                }
             }
             comfyModels.value = []
             isConnectionTested.value = false
@@ -212,7 +215,7 @@ export default {
                   <span class="flex flex-1">
                     <Icon :src="type.icon" class="w-5 h-5 mr-2" />
                     <span class="flex flex-col">
-                      <span class="block text-sm font-medium text-gray-900">{{type.name}}</span>
+                      <span class="block text-sm font-medium text-gray-900">{{type.id}}</span>
                     </span>
                   </span>
                   <svg v-if="providerType == type.id" class="h-5 w-5 text-indigo-600" viewBox="0 0 20 20" fill="currentColor" aria-hidden="true">