Change flaim image viewer to viu (#285)

flaim/.flox/env/manifest.toml

@@ -10,6 +10,7 @@ gum.pkg-path = "gum"
 curl.pkg-path = "curl"
 coreutils.pkg-path = "coreutils"
 figlet.pkg-path = "toilet"
+viu.pkg-path = "viu"
 
 # Required for stdc linking
 gcc-unwrapped.pkg-path = "gcc-unwrapped"
@@ -98,7 +99,6 @@ transformers
 sentencepiece
 bitsandbytes
 protobuf
-sixel
 jupyterlab
 ipywidgets
 EOF

flaim/blart/blart-flux.py

@@ -25,8 +25,6 @@
 
 # grab some image tools
 from PIL import Image
-import sixel
-from io import BytesIO
 import sys
 import gc
 
@@ -42,10 +40,15 @@
 
 # draw images in the terminal
 def draw(image):
-    buffer = BytesIO()
-    writer = sixel.SixelWriter()
-    image.save(buffer, format='png')
-    writer.draw(buffer)
+    import tempfile
+    import subprocess
+    import os
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
+        output_file = temp_file.name
+        image.save(output_file)
+
+    subprocess.run(['viu', output_file], check=True)
+    os.remove(output_file)
 
 # set our device and nope out if we don't have either CUDA or Metal
 if torch.cuda.is_available():
@@ -119,7 +122,6 @@ def draw(image):
             protocomp.paste(im, (x_offset, -212))
             x_offset += im.size[0]
 
-        # this requires a sixel terminal to do anything, sadly
         draw(protocomp)
         print("\n")
 

flaim/blart/blart-proteus.py

@@ -20,8 +20,6 @@
 
 # grab some image tools
 from PIL import Image
-import sixel
-from io import BytesIO
 import sys
 import gc
 
@@ -37,10 +35,15 @@
 
 # draw images in the terminal
 def draw(image):
-    buffer = BytesIO()
-    writer = sixel.SixelWriter()
-    image.save(buffer, format='png')
-    writer.draw(buffer)
+    import tempfile
+    import subprocess
+    import os
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
+        output_file = temp_file.name
+        image.save(output_file)
+
+    subprocess.run(['viu', output_file], check=True)
+    os.remove(output_file)
 
 # set our device and nope out if we don't have either CUDA or Metal
 if torch.cuda.is_available():
@@ -123,7 +126,6 @@ def draw(image):
             protocomp.paste(im, (x_offset, -212))
             x_offset += im.size[0]
 
-        # this requires a sixel terminal to do anything, sadly
         draw(protocomp)
         print("\n")
 

flaim/blart/blart-sd3.py

@@ -24,8 +24,6 @@
 
 # grab some image tools
 from PIL import Image
-import sixel
-from io import BytesIO
 import sys
 import gc
 
@@ -41,10 +39,15 @@
 
 # draw images in the terminal
 def draw(image):
-    buffer = BytesIO()
-    writer = sixel.SixelWriter()
-    image.save(buffer, format='png')
-    writer.draw(buffer)
+    import tempfile
+    import subprocess
+    import os
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
+        output_file = temp_file.name
+        image.save(output_file)
+
+    subprocess.run(['viu', output_file], check=True)
+    os.remove(output_file)
 
 # set our device and nope out if we don't have either CUDA or Metal
 if torch.cuda.is_available():
@@ -117,7 +120,6 @@ def draw(image):
             protocomp.paste(im, (x_offset, -212))
             x_offset += im.size[0]
 
-        # this requires a sixel terminal to do anything, sadly
         draw(protocomp)
         print("\n")
 

flaim/blart/blart-sd35.py

@@ -27,8 +27,6 @@
 
 # grab some image tools
 from PIL import Image
-from io import BytesIO
-import sixel
 import sys
 import gc
 
@@ -44,10 +42,15 @@
 
 # draw images in the terminal
 def draw(image):
-    buffer = BytesIO()
-    writer = sixel.SixelWriter()
-    image.save(buffer, format='png')
-    writer.draw(buffer)
+    import tempfile
+    import subprocess
+    import os
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
+        output_file = temp_file.name
+        image.save(output_file)
+
+    subprocess.run(['viu', output_file], check=True)
+    os.remove(output_file)
 
 # set our device and nope out if we don't have either CUDA or Metal
 if torch.cuda.is_available():
@@ -153,7 +156,6 @@ def draw(image):
             protocomp.paste(im, (x_offset, -212))
             x_offset += im.size[0]
 
-        # this requires a sixel terminal to do anything, sadly
         draw(protocomp)
         print("\n")
 

flaim/helpers/genimg.py

@@ -5,16 +5,19 @@
 
 import sys
 import torch
-import sixel
-from io import BytesIO
 from diffusers import StableDiffusionPipeline
 from diffusers import logging
 
 def draw(image):
-    buffer = BytesIO()
-    writer = sixel.SixelWriter()
-    image.save(buffer, format='png')
-    writer.draw(buffer)
+    import tempfile
+    import subprocess
+    import os
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
+        output_file = temp_file.name
+        image.save(output_file)
+
+    subprocess.run(['viu', output_file], check=True)
+    os.remove(output_file)
 
 logging.set_verbosity(50)
 logging.disable_progress_bar()