feat: img-to-erc721 metadata complete

Author: netpoe

erc721/.gitignore

@@ -0,0 +1,2 @@
+.env
+/files/

erc721/img-to-erc721.py

@@ -0,0 +1,185 @@
+import base64
+import os
+import json
+from openai import OpenAI
+import requests
+from dotenv import load_dotenv
+from PIL import Image
+from PIL import ImageFont
+from PIL import ImageDraw
+
+# Load the environment variables from the .env file
+load_dotenv()
+
+open_ai_client = OpenAI(
+    api_key=os.getenv("OPENAI_API_KEY"),
+)
+
+src_dir_path = "./files/hellheads"
+watermarks_dir_path = "./files/hellheads/watermarks"
+thumbnail_extension = "_thumbnail"
+
+ipfs_gateway_url = "https://blockchainassetregistry.infura-ipfs.io/ipfs/"
+
+# Get the values of projectId and projectSecret from the environment variables
+infura_project_id = os.getenv("INFURA_PROJECT_ID")
+infura_project_secret = os.getenv("INFURA_PROJECT_SECRET")
+
+
+def encode_image(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode("utf-8")
+
+
+def create_img_thumbnails():
+    # Loop through each file in the directory
+    for filename in os.listdir(src_dir_path):
+        # Check if the file is an image file
+        if filename.lower().endswith(".png") or filename.lower().endswith(".jpg"):
+            # Open an image file
+            file_path = os.path.join(src_dir_path, filename)
+            with Image.open(file_path) as img:
+                # Specify the size of the thumbnail
+                size = (187, 187)
+                img.thumbnail(size)
+                # get the file extension
+                file_extension = filename.split(".")[-1]
+                # change the file name to include _thumbnail
+                filename = (
+                    filename.split(".")[0] + thumbnail_extension + "." + file_extension
+                )
+                # Save the thumbnail
+                img.save(os.path.join(src_dir_path, filename))
+
+
+def add_image_watermark():
+    # Loop through each file in the directory
+    for filename in os.listdir(src_dir_path):
+        # Check if the file is an image file
+        if filename.lower().endswith(".png") or filename.lower().endswith(".jpg"):
+            print(f"add_image_watermark: {filename}")
+            # Open an image file
+            file_path = os.path.join(src_dir_path, filename)
+            with Image.open(file_path) as img:
+                draw = ImageDraw.Draw(img)
+                w, h = img.size
+                font_size = int(w * 0.02)
+                x, y = w - font_size * 4, h - font_size * 4
+                font = ImageFont.truetype("PPNeueMachina-InktrapRegular.otf", font_size)
+                draw.text(
+                    (x, y), "@larskristo", fill=(64, 64, 74), font=font, anchor="ms"
+                )
+                img.save(os.path.join(watermarks_dir_path, filename))
+
+
+def upload_file_to_ipfs(file_path):
+    print(f"upload_to_ipfs: {file_path}")
+
+    # Read the file as binary data
+    with open(file_path, "rb") as file:
+        file_data = file.read()
+
+    # Create the IPFS upload API endpoint URL
+    endpoint_url = "https://ipfs.infura.io:5001/api/v0/add"
+
+    # Set the request parameters
+    params = {"pin": "true"}  # Pin the uploaded file to ensure it is not removed
+
+    # Send the POST request to upload the file
+    response = requests.post(
+        endpoint_url,
+        params=params,
+        files={"file": file_data},
+        auth=(infura_project_id, infura_project_secret),
+    )
+
+    # Check if the request was successful
+    if response.status_code == 200:
+        # Extract the IPFS hash from the response
+        ipfs_hash = response.json()["Hash"]
+        return ipfs_hash
+    else:
+        # Handle the error case
+        print(f"Failed to upload file to IPFS. Error: {response.text}")
+        return None
+
+
+def get_image_description(image_url):
+    print(f"get_image_description: {image_url}")
+
+    response = open_ai_client.chat.completions.create(
+        model="gpt-4-vision-preview",
+        messages=[
+            {
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": "What's in this image?"},
+                    {
+                        "type": "image_url",
+                        "image_url": {"url": f"data:image/jpeg;base64,{image_url}"},
+                        "detail": "auto",
+                    },
+                ],
+            }
+        ],
+        max_tokens=140,
+    )
+    description = response.choices[0].message[1].content[0].text
+    print(f"get_image_description: {description}")
+
+    return response.choices[0].message[1].content[0].text
+
+
+def create_metadata():
+    metadata_list = []
+    counter = 0
+    total_items = len([filename for filename in os.listdir(watermarks_dir_path)])
+
+    for filename in os.listdir(watermarks_dir_path):
+        if thumbnail_extension not in filename and (
+            filename.lower().endswith(".png") or filename.lower().endswith(".jpg")
+        ):
+            # Create a variable for the actual file
+            file_path = os.path.join(watermarks_dir_path, filename)
+            # base64_image = encode_image(file_path)
+
+            # Upload the file to IPFS and get the IPFS hash
+            image_ipfs_hash = upload_file_to_ipfs(file_path)
+            print(f"create_metadata.image_ipfs_hash: {image_ipfs_hash}")
+
+            # change the file name to include _thumbnail
+            file_extension = filename.split(".")[-1]
+            name = filename.split(".")[0]
+            thumbnail_filename = name + thumbnail_extension + "." + file_extension
+            thumbnail_path = os.path.join(watermarks_dir_path, thumbnail_filename)
+            thumbnail_ipfs_hash = upload_file_to_ipfs(thumbnail_path)
+            print(f"create_metadata.thumbnail_ipfs_hash: {thumbnail_ipfs_hash}")
+
+            # description = get_image_description(f"{ipfs_gateway_url}{image_ipfs_hash}")
+
+            # Create a metadata object
+            metadata = {
+                "name": name,
+                "description": "Experience the captivating world of dark art through the mesmerizing creations of Lars Kristo. Delve into the depths of the human psyche as each piece unveils a haunting narrative, blending beauty and darkness in a unique and thought-provoking way. With intricate details and a mastery of technique, this artist pushes the boundaries of conventional art, inviting you to explore the shadows and embrace the enigmatic allure of the unknown.",
+                "image": f"{ipfs_gateway_url}{image_ipfs_hash}",
+                "thumbnail": f"{ipfs_gateway_url}{thumbnail_ipfs_hash}",
+            }
+
+            # Append the metadata object to the list
+            metadata_list.append(metadata)
+            counter += 1
+            print(f"create_metadata: {counter} of {total_items} items")
+
+    # Write the list of metadata objects to a JSON file
+    with open("metadata.json", "w") as f:
+        json.dump(metadata_list, f)
+
+
+def main():
+    create_metadata()
+    # create_img_thumbnails()
+    # add_image_watermark()
+
+
+if __name__ == "__main__":
+    main()