diff --git a/main.py b/main.py new file mode 100644 index 0000000..d56b1a4 --- /dev/null +++ b/main.py @@ -0,0 +1,23 @@ +import tracery +from tracery.modifiers import base_english + +rules = { + 'prompt': '#setting#, #modifiers#, #modifiers#, #modifiers#', + 'setting': ['#cosy#', '#place#', '#countryside#', '#train#', '#cityscene#'], + 'cosy': ['cosy scene #time#', 'a hot cup of tea, #weather# outside', 'looking out the window at #city# #time#'], + 'countryside': ['#country# #time#', '#weather# #country# #time#', '#country#'], + 'train': ['#transit# #time#', '#weather# #transit# #time#'], + 'transit': ['train station', 'an old bus stop', 'rail station', 'subway station'], + 'time': ['at night', 'at sunrise', 'at dusk', 'at midnight', 'midday'], + 'weather': ['rainy', 'cloudy', 'foggy', 'clear', 'overcast'], + 'city': ['Tokyo', 'Sydney', 'Paris', 'Kyoto', 'Chicago', 'New York', 'San Francisco', 'Reykjavik', 'Hanoi', 'Hong Kong', 'London', 'Madrid', 'Barcelona', 'Prague', 'Santa Fe', 'Portland', 'New Orleans'], + 'cityscene': ['#place# in #city# #time#', '#weather# #place# in #city#', '#place# #time#'], + 'place': ['art museum', 'old diner', 'fancy restaurant', 'abandoned storefront', 'alleyway', 'dive bar', 'post office', 'cosy cabin', 'hotel', 'covered bridge', 'waterfront', 'pier', 'street scene'], + 'country': ['lakeside dock', 'deep in the woods', 'hillside', 'deer trail', 'hiking trail', 'mountain overlook', 'mountain pass', 'riverbank', 'forest glen', 'meadow'], + 'modifiers': ['detailed', 'trending on artstation', '4k', 'highly realistic', 'fantasy vivid colors', 'ghibli inspired', 'studio ghibli', 'anime', 'anime art style', 'anime concept art', 'vivid colors', 'bokeh', 'aesthetic', 'trending on cgsociety', 'cinematic',] +} + +grammar = tracery.Grammar(rules) +grammar.add_modifiers(base_english) + +print(grammar.flatten("#prompt#")) diff --git a/tweet.py b/tweet.py new file mode 100755 index 0000000..79629bf --- /dev/null +++ b/tweet.py @@ -0,0 +1,22 @@ +#!/usr/bin/env python3 + +import tweepy +from dotenv import load_dotenv +import os +import sys + +load_dotenv() + +# Authenticate to Twitter +auth = tweepy.OAuthHandler(os.getenv("APIKEY"), os.getenv("APISECRET")) +auth.set_access_token(os.getenv("TOKEN"), os.getenv("TOKENSECRET")) + +# Create API object +api = tweepy.API(auth) + +# Create a tweet +media = api.media_upload(sys.argv[1]) + +tweet = "" +api.update_status(status=tweet, media_ids=[media.media_id]) + diff --git a/tweet.sh b/tweet.sh new file mode 100755 index 0000000..b223227 --- /dev/null +++ b/tweet.sh @@ -0,0 +1,7 @@ +#!/bin/bash + +CONTAINER=$(/snap/bin/docker ps | grep miniconda | awk '{print $1}') +/snap/bin/docker exec -i $CONTAINER /sd/run.sh +IMAGE="$(ls /home/gabe/stable-diffusion/outputs/*.png -t | head -n1)" +/home/gabe/realesrgan/realesrgan-ncnn-vulkan -i "$IMAGE" -o /tmp/upscaled.png +python3 /home/gabe/lofi-imgen/tweet.py /tmp/upscaled.png