Update app.py

app.py

@@ -280,9 +280,51 @@ def fonc_load_info(nb_of_models_to_gen,index_tag,index_first_model):
     str_temp+=f"{index_first_model+1}/{len(tags_plus_models[index_tag][2])} to {index_first_model+nb_of_models_to_gen}/{len(tags_plus_models[index_tag][2])}) :\n\n"
     for i in range(nb_of_models_to_gen):
         list_models_temp.append(tags_plus_models[index_tag][2][i+index_first_model])
-        str_temp+=f"{tags_plus_models[index_tag][2][i+index_first_model]}\n"
-    
+        str_temp+=f"\"{tags_plus_models[index_tag][2][i+index_first_model]}\",\n"
     return nb_of_models_to_gen,gr.Textbox(str_temp),gr.Dropdown(choices=[["",list_models_temp]], value=list_models_temp )
+def fonc_load_info_custom(nb_of_models_to_gen,list_model_custom,index_first_model):
+    str_temp=""
+    list_models_temp=[]
+    if index_first_model+nb_of_models_to_gen>len(list_model_custom):
+        if nb_of_models_to_gen>len(list_model_custom):
+            str_temp+="warning : to many model chosen"
+        else:
+            str_temp+="warning : first model to close to the last model"
+        nb_of_models_to_gen= len(list_model_custom)-index_first_model
+        str_temp+=f" - only {nb_of_models_to_gen} will be use\n\n"
+    str_temp+="list of models use (from "
+    str_temp+=f"{index_first_model+1}/{len(list_model_custom)} to {index_first_model+nb_of_models_to_gen}/{len(list_model_custom)}) :\n\n"
+    for i in range(nb_of_models_to_gen):
+        list_models_temp.append(list_model_custom[i+index_first_model])
+        str_temp+=f"\"{list_model_custom[i+index_first_model]}\",\n"
+    return nb_of_models_to_gen,gr.Textbox(str_temp),gr.Dropdown(choices=[["",list_models_temp]], value=list_models_temp )
+
+
+
+    
+def cutStrg(longStrg,start,end):
+    shortStrg=''
+    for i in range(end-start):
+        shortStrg+=longStrg[start+i]
+    return shortStrg
+def aff_models_perso(txt_list_perso,models=models):
+    list_perso=[]
+    t1=True
+    start=txt_list_perso.find('\"')
+    if start!=-1:
+        while t1:
+            start+=1
+            end=txt_list_perso.find('\"',start)
+            if end != -1:
+                txtTemp=cutStrg(txt_list_perso,start,end)
+                if txtTemp in models:
+                    list_perso.append(cutStrg(txt_list_perso,start,end))
+            else :
+                t1=False
+            start=txt_list_perso.find('\"',end+1)
+            if start==-1:
+                t1=False
+    return gr.Dropdown(choices=[["",list_perso]], value=list_perso )
             
 def make_me():
         with gr.Column():
@@ -326,15 +368,30 @@ def make_me():
                 nb_images_by_prompt=gr.Number(2,label="Number of images by prompt:",interactive=True)
                 nb_of_models_to_gen=gr.Number(10,label="Number of Models:",interactive=True)
                 index_tag=gr.Dropdown(label="Tag",choices=list(list_tags),type="index")
-                index_first_model=gr.Dropdown(label="First model",choices=list(tags_plus_models[1][2]), type="index")
-                #index_tag.change(lambda i:gr.Dropdown(choices=list(tags_plus_models[i][2])),index_tag,index_first_model)
+                index_first_model=gr.Dropdown(label="First model",choices=list([]), type="index")
                 index_tag.change(lambda i:gr.Dropdown(choices=list([f"({j+1}/{len(tags_plus_models[i][2])}) {tags_plus_models[i][2][j]}" for j in range(len(tags_plus_models[i][2]))])),
                                  index_tag,index_first_model)
             with gr.Row():
                 list_models_to_gen=gr.Dropdown(choices=[["",[]]], value=[], visible=False)
                 disp_info=gr.Textbox(label="Info")
-                load_info=gr.Button("Load info")
+                load_info=gr.Button("Load Models")
                 load_info.click(fonc_load_info,[nb_of_models_to_gen,index_tag,index_first_model],[nb_of_models_to_gen,disp_info,list_models_to_gen])
+                
+             with gr.Accordion("Display/Load Models",show=False) :
+                with gr.Row():
+                    text_list_model_custom=gr.Textbox(label="List Models Custom")
+                    with gr.Column():
+                        list_model_custom=gr.Dropdown(choices=[["",[]]], value=[], visible=False)
+                        #use_models_custom=gr.Radio("Use Models Custom",value=False)
+                        cut_model_custom=gr.Button("Cut Text Models Custom")
+                        cut_model_custom.click(aff_models_perso,[text_list_model_custom],[list_model_custom])
+                        
+                        index_first_model_custom=gr.Dropdown(label="First model",choices=list([]), type="index")
+                        list_model_custom.change(lambda li:gr.Dropdown(choices=list([f"({j+1}/{len(li)}) {li[j]}" for j in range(len(li))])),
+                                         [list_model_custom],index_first_model_custom)
+                        
+                        load_model_custom=gr.Button("Load Text Models Custom")
+                        load_model_custom.click(fonc_load_info_custom,[nb_of_models_to_gen,list_model_custom,index_first_model_custom],[nb_of_models_to_gen,disp_info,list_models_to_gen])
 
 js_code = """