add silly code

Signed-off-by: Balazs Horvath <acsipont@gmail.com>

fixcringe.py

@@ -0,0 +1,56 @@
+import os
+import re
+
+def fix_blurhash_block(content):
+    # Pattern to match the blurhash block
+    pattern = r'{{<\s*blurhash\s*\n((?:[^>}]|\n)*?)>}}'
+    
+    def replace_block(match):
+        lines = match.group(1).strip().split('\n')
+        # Process each line to add proper indentation and clean up
+        processed_lines = []
+        for line in lines:
+            # Strip existing whitespace
+            clean_line = line.strip()
+            # Add 2 spaces indentation
+            processed_lines.append(f"  {clean_line}")
+            
+            # If this is the alt line, add grid="true" after it
+            if 'alt=' in clean_line:
+                grid_line = '  grid="true"'
+                processed_lines.append(grid_line)
+                
+        # Reconstruct the block
+        return '{{< blurhash\n' + '\n'.join(processed_lines) + '\n>}}'
+    
+    # Replace all occurrences in the content
+    return re.sub(pattern, replace_block, content, flags=re.MULTILINE)
+
+def process_file(file_path):
+    try:
+        # Read the file content
+        with open(file_path, 'r', encoding='utf-8') as f:
+            content = f.read()
+        
+        # Fix the content
+        modified_content = fix_blurhash_block(content)
+        
+        # Write back only if changes were made
+        if content != modified_content:
+            with open(file_path, 'w', encoding='utf-8') as f:
+                f.write(modified_content)
+            print(f"Processed: {file_path}")
+            
+    except Exception as e:
+        print(f"Error processing {file_path}: {str(e)}")
+
+def main():
+    # Walk through all directories recursively
+    for root, dirs, files in os.walk('.'):
+        for file in files:
+            if file.endswith('.cringe'):
+                file_path = os.path.join(root, file)
+                process_file(file_path)
+
+if __name__ == '__main__':
+    main()

shortcode.py

@@ -0,0 +1,40 @@
+import os
+import json
+
+def process_json_file(json_path, base_dir):
+    with open(json_path, 'r') as file:
+        data = json.load(file)
+        filename = data['filename']
+        width = data['width']
+        height = data['height']
+        blurhash = data['blurhash']
+        caption = data['caption']
+        
+        image_filename, image_fileextension = os.path.splitext(filename)
+        image_fileextension = image_fileextension.lstrip('.')
+        
+        directory = os.path.relpath(os.path.dirname(json_path), base_dir)
+        directory = directory.replace('static/', '', 1)  # Remove the first occurrence of 'static/'
+        
+        cringe_filename = f"{image_filename}.{image_fileextension}.cringe"
+        cringe_content = f"""{{{{< blurhash
+src="https://huggingface.co/k4d3/yiff_toolkit/resolve/main/static/{directory}/{filename}"
+    blurhash="{blurhash}"
+    width="{width}"
+    height="{height}"
+    alt="{caption}"
+>}}}}"""
+        
+        with open(os.path.join(os.path.dirname(json_path), cringe_filename), 'w') as cringe_file:
+            cringe_file.write(cringe_content)
+
+def recurse_directories(base_dir):
+    for root, _, files in os.walk(base_dir):
+        for file in files:
+            if file.endswith('.json'):
+                process_json_file(os.path.join(root, file), base_dir)
+
+if __name__ == "__main__":
+    base_directory = os.path.join(os.getcwd(), 'yiff_toolkit')
+    recurse_directories(base_directory)
+

yiffdata.py

@@ -0,0 +1,46 @@
+import os
+import json
+from PIL import Image
+
+def get_image_info(image_path):
+    with Image.open(image_path) as img:
+        width, height = img.size
+    return width, height
+
+def process_directory(directory):
+    for root, _, files in os.walk(directory):
+        for file in files:
+            if file.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')):
+                image_path = os.path.join(root, file)
+                width, height = get_image_info(image_path)
+                
+                base_name, ext = os.path.splitext(file)
+                blurhash_file = os.path.join(root, f"{base_name}{ext}.bh")
+                caption_file = os.path.join(root, f"{base_name}.caption")
+                
+                blurhash = None
+                caption = None
+                
+                if os.path.exists(blurhash_file):
+                    with open(blurhash_file, 'r') as bh_file:
+                        blurhash = bh_file.read().strip()
+                
+                if os.path.exists(caption_file):
+                    with open(caption_file, 'r') as cap_file:
+                        caption = cap_file.read().strip()
+                
+                json_data = {
+                    "filename": file,
+                    "width": width,
+                    "height": height,
+                    "blurhash": blurhash,
+                    "caption": caption
+                }
+                
+                json_file = os.path.join(root, f"{base_name}{ext}.json")
+                with open(json_file, 'w') as jf:
+                    json.dump(json_data, jf, indent=4)
+
+if __name__ == "__main__":
+    process_directory(os.getcwd())
+