bugifx on ui about model selection

app.py

@@ -144,12 +144,40 @@ def main():
             model_dd  = gr.Dropdown(mf, value=mf[0], label='Decoder Weights')
             clip_dd   = gr.Dropdown(cf, value=cf[0], label='CLIP Weights')
 
-        # On any selection change, preload the llm
-        size_dd.change(fn=lambda s, m, c: update_llm(s, m, c), inputs=[size_dd, model_dd, clip_dd], outputs=[])
-        model_dd.change(fn=lambda s, m, c: update_llm(s, m, c), inputs=[size_dd, model_dd, clip_dd], outputs=[])
-        clip_dd.change(fn=lambda s, m, c: update_llm(s, m, c), inputs=[size_dd, model_dd, clip_dd], outputs=[])
+                # When size changes: update dropdowns AND preload llm with the new first weights
+        def on_size_change(sz):
+            mlist, clist = get_weight_files(sz)
+            # update dropdown choices and default values
+            update_ui = (
+                gr.update(choices=mlist, value=mlist[0]),
+                gr.update(choices=clist, value=clist[0])
+            )
+            # preload with first weights
+            update_llm(sz, mlist[0], clist[0])
+            return update_ui
+        size_dd.change(
+            fn=on_size_change,
+            inputs=[size_dd],
+            outputs=[model_dd, clip_dd]
+        )
+
+        # When model weight changes: preload llm
+        model_dd.change(
+            fn=lambda sz, mf, cf: update_llm(sz, mf, cf),
+            inputs=[size_dd, model_dd, clip_dd],
+            outputs=[]
+        )
+        # When clip weight changes: preload llm
+        clip_dd.change(
+            fn=lambda sz, mf, cf: update_llm(sz, mf, cf),
+            inputs=[size_dd, model_dd, clip_dd],
+            outputs=[]
+        )
 
-        # Initial load
+        # Initial preload with defaults
+        update_llm(default, mf[0], cf[0])
+
+        interval = gr.Slider(100, 20000, step=100, value=3000, label='Interval (ms)')
         update_llm(default, mf[0], cf[0])
 
         interval = gr.Slider(100, 20000, step=100, value=3000, label='Interval (ms)')