added support to add more fonts

app.py

@@ -25,6 +25,8 @@ sys.path.append("/home/user/.local/lib/python3.10/site-packages/diffvg-0.0.1-py3
 
 os.chdir('/home/user/app')
 
+# os.system('bash code/data/fonts/arabic/download_fonts.sh')
+
 import torch
 from diffusers import StableDiffusionPipeline
 
@@ -72,6 +74,26 @@ warnings.filterwarnings("ignore")
 pydiffvg.set_print_timing(False)
 gamma = 1.0
 
+def read_font_names(all_scripts):
+
+    font_names = []
+    font_dict = {}
+    for script in all_scripts:
+        script = script.lower()
+        font_dict[script] = []
+        if script == "simplified chinese":
+            script = "chinese"
+
+        path = f"code/data/fonts/{script.lower()}/font_names.txt"
+        if not os.path.exists(path):
+            font_dict[script] = [x[:-4] for x in os.listdir(os.path.dirname(path)) if "ttf" in x]
+        else:
+            with open(path, 'r', encoding="utf-8") as fin:
+                font_dict[script] = [line.strip() for line in fin.readlines()]
+
+        font_names.extend([f"{script.capitalize()}: {f}" for f in font_dict[script]])
+
+    return ["Default"] + sorted(font_names), font_dict
 
 def set_config(semantic_concept, word, script, prompt_suffix, font_name, num_steps, seed, is_seed_rand, dist_loss_weight, pixel_dist_kernel_blur, pixel_dist_sigma, angeles_w):
     
@@ -99,18 +121,13 @@ def set_config(semantic_concept, word, script, prompt_suffix, font_name, num_ste
     cfg.optimized_letter = word
     cfg.script = script.lower()
 
-    if cfg.script == "simplified chinese":
-        cfg.script = "chinese"
-
-    script_path = f"code/data/fonts/{cfg.script}"
-    cfg.font = [x for x in os.listdir(script_path) if "ttf" in x][0][:-4]
+    cfg.font = font_name
 
     if is_seed_rand == "Random Seed":
         cfg.seed = np.random.randint(10000)
     else:
         cfg.seed = int(seed)
 
-
     cfg.num_iter = num_steps
     cfg.batch_size = 1
     cfg.loss.tone.dist_loss_weight = int(dist_loss_weight)
@@ -118,7 +135,6 @@ def set_config(semantic_concept, word, script, prompt_suffix, font_name, num_ste
     cfg.loss.tone.pixel_dist_sigma = int(pixel_dist_sigma)
     cfg.loss.conformal.angeles_w = angeles_w    
 
-
     cfg.caption = f"a {cfg.semantic_concept}. {cfg.prompt_suffix}"
     cfg.log_dir = f"{cfg.script}"
     if cfg.optimized_letter in cfg.word:
@@ -171,12 +187,37 @@ def init_shapes(svg_path, trainable: Mapping[str, bool]):
     return shapes_init, shape_groups_init, parameters
 
 
-def run_main_ex(word, semantic_concept, script, num_steps, seed):
+def run_main_ex(word, semantic_concept, script, font_selector, num_steps, seed):
     prompt_suffix = "minimal flat 2d vector. lineal color. trending on artstation"
     is_seed_rand = "Use Set Value"
-    return list(next(run_main_app(semantic_concept, word, script, prompt_suffix, font_name, num_steps, seed, is_seed_rand, 100, 201, 30, 0.5, 1)))
+    return list(next(run_main_app(semantic_concept, word, script, font_selector, prompt_suffix, num_steps, seed, is_seed_rand, 100, 201, 30, 0.5, 1)))
                         
-def run_main_app(semantic_concept, word, script, prompt_suffix, font_name, num_steps, seed, is_seed_rand, dist_loss_weight, pixel_dist_kernel_blur, pixel_dist_sigma, angeles_w, example=0):
+def run_main_app(semantic_concept, word, script, font_selected, prompt_suffix, num_steps, seed, is_seed_rand, dist_loss_weight, pixel_dist_kernel_blur, pixel_dist_sigma, angeles_w, example=0):
+    
+    if font_selected.lower() != "default":
+        font_key, font_val = font_selected.lower().split(":")
+        font_key = font_key.strip()
+        font_val = font_val.strip()
+    else:
+        font_key = "default"
+        font_val = "default"
+
+    if script.lower() == "simplified chinese":
+        script = "chinese"
+
+    if font_key != script.lower():
+        print(f"Setting font to {script} default font")
+        font_key = script.lower()
+
+    if len(font_dict[font_key]) == 1:
+        font_name = font_dict[font_key][0]
+    else:
+        if font_val == "default":
+            font_name = "00"
+        else:
+            font_name = str(font_dict[font_key].index(font_val)).zfill(2)
+
+    print(font_name)
     
     cfg = set_config(semantic_concept, word, script, prompt_suffix, font_name, num_steps, seed, is_seed_rand, dist_loss_weight, pixel_dist_kernel_blur, pixel_dist_sigma, angeles_w)
 
@@ -247,8 +288,8 @@ def run_main_app(semantic_concept, word, script, prompt_suffix, font_name, num_s
 
         if step % skip == 0:
             img_tensor = x.detach().cpu()
-            img_tensor = torch.nn.functional.interpolate(img_tensor, size=(200, 200), mode='bicubic', align_corners=False)
-            img_tensor = img_tensor.permute(0, 3, 2, 1).squeeze(0)
+            img_tensor = torch.nn.functional.interpolate(img_tensor, size=(300, 300), mode='bilinear', align_corners=False)
+            img_tensor = img_tensor.permute(0, 2, 3, 1).squeeze(0)
             gif_frames += [img_tensor.numpy()]
         
         x = x.repeat(cfg.batch_size, 1, 1, 1)
@@ -278,13 +319,13 @@ def run_main_app(semantic_concept, word, script, prompt_suffix, font_name, num_s
     # for frame in gif_frames: writer.append_data(frame)
     # writer.close()
     gif_frames = np.array(gif_frames) * 255
-    print(gif_frames[0])
     imageio.mimsave(filename, gif_frames.astype(np.uint8))
     # imageio.mimsave(filename, np.array(gif_frames))
 
     yield gr.update(value=filename_init,visible=True),gr.update(visible=False),gr.update(value=filename,visible=True),gr.update(value=cfg.caption,visible=True),gr.update(value=cfg.seed,visible=True)
  
 
+all_scripts = ["Arabic", "Simplified Chinese", "Cyrillic", "Greek", "Latin", "Tamil"]
 with gr.Blocks() as demo:
 
     gr.HTML(TITLE)
@@ -307,12 +348,21 @@ with gr.Blocks() as demo:
                 'Enter a semantic concept that you want your text to morph into (in English). For example: cat'
             )
 
-            script = gr.Dropdown(
-                ["Arabic", "Simplified Chinese", "Cyrillic", "Greek", "Latin", "Tamil"], 
-                value="Arabic",
-                label="Font Script"
-            )
+            with gr.Row():
 
+                script_selector = gr.Dropdown(
+                    all_scripts, 
+                    value="Arabic",
+                    label="Font Script"
+                )
+
+                font_names, font_dict = read_font_names(all_scripts)
+                font_selector = gr.Dropdown(
+                    font_names,
+                    value=font_names[0],
+                    label="Font Name",
+                    visible=True,
+                )
 
             prompt_suffix = gr.Text(
                 label='Prompt Suffix',
@@ -358,15 +408,7 @@ with gr.Blocks() as demo:
                       maximum=500,
                       step=10,
                       value=250)
-            
-            font_name = gr.Text(value=None,visible=False,label="Font Name")
-        
-            
-            def on_select(evt: gr.SelectData):
-                return evt.value
-                
-            font_name.value = "ArefRuqaa"
-          
+                          
             run = gr.Button('Generate')
 
         with gr.Column():
@@ -395,7 +437,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         # examples
         examples = [
-            ["قطة", "Cat", "Arabic", 50, 42],
+            ["قطة", "Cat", "Arabic", "Arabic: ArefRuqaa-Regular", 100, 42],
             # ["猫", "Cat", "Simplified Chinese", 250, 42],
             # ["γάτα", "Cat", "Greek", 250, 42],
             # ["кошка", "Cat", "Cyrillic", 250, 42],
@@ -407,7 +449,8 @@ with gr.Blocks() as demo:
                 inputs=[
                     word,
                     semantic_concept,
-                    script,
+                    script_selector,
+                    font_selector,
                     num_steps,
                     seed
                 ],
@@ -426,9 +469,9 @@ with gr.Blocks() as demo:
     inputs = [
         semantic_concept,
         word, 
-        script,
+        script_selector,
+        font_selector,
         prompt_suffix,
-        font_name,
         num_steps,
         seed,
         is_seed_rand,

code/data/fonts/arabic/download_fonts.sh

@@ -1,20 +1,20 @@
+#!/bin/sh
+
 wget https://arbfonts.com//wp-content/fonts/diwany-arabic-fonts//mcs-diwany-jaly-s-u.ttf -O 01.ttf
 wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//46-Diwani-Bent.ttf -O 02.ttf
 wget https://arbfonts.com//wp-content/fonts/diwany-arabic-fonts//diwany-edited.ttf -O 03.ttf
 wget https://arbfonts.com/wp-content/fonts/diwany-arabic-fonts//arbfonts-samt-7017.ttf -O 04.ttf
 wget https://arbfonts.com//wp-content/fonts/kufi-arabic-fonts//QadasiRegular.ttf -O 05.ttf
 wget https://arbfonts.com//wp-content/fonts/kufi-arabic-fonts//Spirit-Of-Doha-Black.otf -O 06.ttf
-wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//AlQalam-alavi.ttf -O 07.ttf
-wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//22-andlso.ttf -O 08.ttf
-wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//ArefRuqaa-Bold-1.ttf -O 09.ttf 
-wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//ArefRuqaa-Regular-1.ttf -O 10.ttf
-wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//117-Barada-Reqa.ttf -O 11.ttf
-wget https://arbfonts.com/wp-content/fonts/diwany-arabic-fonts//arbfonts-diwany-thuluth.ttf -O 12.ttf
-wget https://arbfonts.com//wp-content/fonts/unlimited-free-arabic-fonts//UthmanicHafs1-Ver09_2.otf -O 13.ttf
-wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//ALAMERAH-Thin.ttf -O 14.ttf
-wget https://arbfonts.com/font_files/horr/unicode/Boahmed%20Alhour.ttf -O 15.ttf
-wget https://arbfonts.com//wp-content/fonts/misc//K-Kamran.ttf -O 16.ttf
-wget https://arbfonts.com//wp-content/fonts/farsi-free-fonts//Jamil-nory.ttf -O 17.ttf
-wget https://arbfonts.com//wp-content/fonts/brands-arasbic-fonts//Mobily.ttf -O 18.ttf
-wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//QTSManga-Regular-1.ttf -O 19.ttf
-wget https://arbfonts.com//wp-content/fonts/arabic-fonts/new//Al-Jazeera-Arabic-Regular.ttf -O 20.ttf
+wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//22-andlso.ttf -O 07.ttf
+wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//ArefRuqaa-Bold-1.ttf -O 08.ttf 
+wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//ArefRuqaa-Regular-1.ttf -O 09.ttf
+wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//117-Barada-Reqa.ttf -O 10.ttf
+wget https://arbfonts.com/wp-content/fonts/diwany-arabic-fonts//arbfonts-diwany-thuluth.ttf -O 11.ttf
+wget https://arbfonts.com//wp-content/fonts/unlimited-free-arabic-fonts//UthmanicHafs1-Ver09_2.otf -O 12.ttf
+wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//ALAMERAH-Thin.ttf -O 13.ttf
+wget https://arbfonts.com/font_files/horr/unicode/Boahmed%20Alhour.ttf -O 14.ttf
+wget https://arbfonts.com//wp-content/fonts/misc//K-Kamran.ttf -O 15.ttf
+wget https://arbfonts.com//wp-content/fonts/brands-arasbic-fonts//Mobily.ttf -O 16.ttf
+wget https://arbfonts.com//wp-content/fonts/new-arabic-fonts//QTSManga-Regular-1.ttf -O 17.ttf
+wget https://arbfonts.com//wp-content/fonts/arabic-fonts/new//Al-Jazeera-Arabic-Regular.ttf -O 18.ttf

code/data/fonts/arabic/font_names.txt

@@ -1,10 +1,10 @@
+ArefRuqaa-Regular
 رقعة عارف
 ديواني جلي
 ديواني مشكل
 ديواني طويل
 ديواني بسيط
 كوفي بسيط
-كوفي منحني
 فارسي بسيط
 مغربي اندلس
 رقعة مدبب
@@ -14,7 +14,6 @@
 ثلث بسيط
 مربع بسيط
 حر مدبب
-حر بسيط
 حر طويل
 موبايلي
 منجا