fixes

app.py

@@ -12,10 +12,10 @@ import sqlite3
 import subprocess
 from jsonschema import ValidationError
 
-MODE = os.environ.get('FLASK_ENV', 'production')
-IS_DEV = MODE == 'development'
-app = Flask(__name__, static_url_path='/static')
-app.config['JSONIFY_PRETTYPRINT_REGULAR'] = False
+MODE = os.environ.get("FLASK_ENV", "production")
+IS_DEV = MODE == "development"
+app = Flask(__name__, static_url_path="/static")
+app.config["JSONIFY_PRETTYPRINT_REGULAR"] = False
 
 schema = {
     "type": "object",
@@ -30,29 +30,28 @@ schema = {
                 "properties": {
                     "colors": {
                         "type": "array",
-                        "items": {
-                            "type": "string"
-                        },
+                        "items": {"type": "string"},
                         "maxItems": 5,
-                        "minItems": 5
+                        "minItems": 5,
                     },
-                    "imgURL": {"type": "string"}}
-            }
-        }
+                    "imgURL": {"type": "string"},
+                },
+            },
+        },
     },
     "minProperties": 2,
-    "maxProperties": 2
+    "maxProperties": 2,
 }
 
 CORS(app)
 
 DB_FILE = Path("./data.db")
-TOKEN = os.environ.get('HUGGING_FACE_HUB_TOKEN')
+TOKEN = os.environ.get("HUGGING_FACE_HUB_TOKEN")
 repo = Repository(
     local_dir="data",
     repo_type="dataset",
     clone_from="huggingface-projects/color-palettes-sd",
-    use_auth_token=TOKEN
+    use_auth_token=TOKEN,
 )
 repo.git_pull()
 # copy db on db to local path
@@ -66,12 +65,13 @@ try:
     db.close()
 except sqlite3.OperationalError:
     db.execute(
-        'CREATE TABLE palettes (id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, data json, created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL)')
+        "CREATE TABLE palettes (id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, data json, created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL)"
+    )
     db.commit()
 
 
 def get_db():
-    db = getattr(g, '_database', None)
+    db = getattr(g, "_database", None)
     if db is None:
         db = g._database = sqlite3.connect(DB_FILE)
         db.row_factory = sqlite3.Row
@@ -80,7 +80,7 @@ def get_db():
 
 @app.teardown_appcontext
 def close_connection(exception):
-    db = getattr(g, '_database', None)
+    db = getattr(g, "_database", None)
     if db is not None:
         db.close()
 
@@ -93,45 +93,60 @@ def update_repository():
     with sqlite3.connect("./data/data.db") as db:
         db.row_factory = sqlite3.Row
         palettes = db.execute("SELECT * FROM palettes").fetchall()
-        data = [{'id': row['id'], 'data': json.loads(
-            row['data']), 'created_at': row['created_at']} for row in palettes]
+        data = [
+            {
+                "id": row["id"],
+                "data": json.loads(row["data"]),
+                "created_at": row["created_at"],
+            }
+            for row in palettes
+        ]
 
-    with open('./data/data.json', 'w') as f:
-        json.dump(data, f, separators=(',', ':'))
+    with open("./data/data.json", "w") as f:
+        json.dump(data, f, separators=(",", ":"))
 
     print("Updating repository")
     subprocess.Popen(
-        "git add . && git commit --amend -m 'update' && git push --force", cwd="./data", shell=True)
+        "git add . && git commit --amend -m 'update' && git push --force",
+        cwd="./data",
+        shell=True,
+    )
     repo.push_to_hub(blocking=False)
 
 
-@app.route('/')
+@app.route("/")
 def index():
-    return app.send_static_file('index.html')
+    return app.send_static_file("index.html")
 
 
-@app.route('/force_push')
+@app.route("/force_push")
 def push():
-    if (request.headers['token'] == TOKEN):
+    if request.headers["token"] == TOKEN:
         update_repository()
-        return jsonify({'success': True})
+        return jsonify({"success": True})
     else:
         return "Error", 401
 
 
 def getAllData():
     palettes = get_db().execute("SELECT * FROM palettes").fetchall()
-    data = [{'id': row['id'], 'data': json.loads(
-            row['data']), 'created_at': row['created_at']} for row in palettes]
+    data = [
+        {
+            "id": row["id"],
+            "data": json.loads(row["data"]),
+            "created_at": row["created_at"],
+        }
+        for row in palettes
+    ]
     return data
 
 
-@app.route('/data')
+@app.route("/data")
 def getdata():
     return jsonify(getAllData())
 
 
-@app.route('/new_palette', methods=['POST'])
+@app.route("/new_palette", methods=["POST"])
 @expects_json(schema)
 def create():
     data = g.data
@@ -146,18 +161,26 @@ def create():
 def bad_request(error):
     if isinstance(error.description, ValidationError):
         original_error = error.description
-        return jsonify({'error': original_error.message}), 400
+        return jsonify({"error": original_error.message}), 400
     return error
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     if not IS_DEV:
         print("Starting scheduler -- Running Production")
         scheduler = APScheduler()
-        scheduler.add_job(id='Update Dataset Repository',
-                          func=update_repository, trigger='interval', hours=1)
+        scheduler.add_job(
+            id="Update Dataset Repository",
+            func=update_repository,
+            trigger="interval",
+            hours=1,
+        )
         scheduler.start()
     else:
         print("Not Starting scheduler -- Running Development")
-    app.run(host='0.0.0.0',  port=int(
-        os.environ.get('PORT', 7860)), debug=True, use_reloader=IS_DEV)
+    app.run(
+        host="0.0.0.0",
+        port=int(os.environ.get("PORT", 7860)),
+        debug=True,
+        use_reloader=IS_DEV,
+    )

frontend/src/lib/Palette.svelte

@@ -12,7 +12,17 @@
 
 	$: prompt = promptData?.prompt;
 	$: colors = promptData?.images[seletecdImage]?.colors.map((e) => d3.rgb(e)) || [];
-	$: imageSrc = promptData?.images[seletecdImage]?.imgURL;
+	$: imageSrc = fixLink(promptData?.images[seletecdImage]?.imgURL);
+
+	function fixLink(link: string) {
+		if (link.includes('s3.amazonaws.com')) {
+			return link.replace(
+				's3.amazonaws.com/moonup/production/uploads/noauth',
+				'cdn-uploads.huggingface.co/production/uploads/noauth'
+			);
+		}
+		return link;
+	}
 	let isCopying = false;
 
 	async function copyStringToClipboard(text: string) {

frontend/src/routes/+page.svelte

@@ -3,7 +3,7 @@
 	import type { ColorsPrompt, ColorsImage } from '$lib/types';
 	import { randomSeed, extractPalette, uploadImage } from '$lib/utils';
 	import { isLoading, loadingState } from '$lib/store';
-	import { PUBLIC_WS_ENDPOINT, PUBLIC_API} from '$env/static/public';
+	import { PUBLIC_WS_ENDPOINT, PUBLIC_API } from '$env/static/public';
 	import Pallette from '$lib/Palette.svelte';
 	import ArrowRight from '$lib/ArrowRight.svelte';
 	import ArrowLeft from '$lib/ArrowLeft.svelte';
@@ -81,12 +81,12 @@
 		const sessionHash = crypto.randomUUID();
 
 		const hashpayload = {
-			fn_index: 2,
+			fn_index: 3,
 			session_hash: sessionHash
 		};
 
 		const datapayload = {
-			data: [_prompt]
+			data: [_prompt, '', 9]
 		};
 
 		const websocket = new WebSocket(PUBLIC_WS_ENDPOINT);