Refactoring and bug fixes

Author: sd109

charts/flux-image-gen/Chart.yaml

@@ -21,4 +21,4 @@ version: 0.1.0
 # incremented each time you make changes to the application. Versions are not expected to
 # follow Semantic Versioning. They should reflect the version the application is using.
 # It is recommended to use it with quotes.
-appVersion: "02aeef3"
+appVersion: "fb1397e"

charts/flux-image-gen/templates/ui/configmap.yml renamed to charts/flux-image-gen/templates/ui/configmap.yaml

@@ -11,4 +11,6 @@ data:
     {{- range $model := .Values.models }}
       - name: {{ . }}
         address: {{ printf "http://%s.%s.svc:%v" ( printf "%s-%s-api" (include "flux-image-gen.fullname" $) . ) $.Release.Namespace $.Values.api.service.port }}
-    {{- end -}}
+    {{- end }}
+    example_prompt: |
+      {{- .Values.examplePrompt | nindent 6 -}}

charts/flux-image-gen/templates/ui/deployment.yaml

@@ -14,6 +14,8 @@ spec:
     metadata:
       {{- with .Values.ui.podAnnotations }}
       annotations:
+        # Recreate pods if settings config map changes
+        checksum/config: {{ include (print $.Template.BasePath "/ui/configmap.yaml") . | sha256sum }}
         {{- toYaml . | nindent 8 }}
       {{- end }}
       labels:

charts/flux-image-gen/templates/ui/ingress.yaml

@@ -35,7 +35,7 @@ spec:
             {{- end }}
             backend:
               service:
-                name: {{ include "flux-image-gen.fullname" $ }}
+                name: {{ include "flux-image-gen.fullname" $ }}-ui
                 port:
                   number: {{ $.Values.ui.service.port }}
           {{- end }}

charts/flux-image-gen/values.yaml

@@ -6,6 +6,9 @@ models:
   - flux-schnell
   # - flux-dev
 
+examplePrompt: |
+  Yoda riding a tiny unicorn through space.
+
 # This sets the container image more information can be found here: https://kubernetes.io/docs/concepts/containers/images/
 image:
   repository: ghcr.io/stackhpc/flux-image-model-inference