Merge branch 'dev'

rtuszik · Sep 3, 2024 · 6c8b907 · 6c8b907
2 parents 0266d2d + 1b2b013
commit 6c8b907
Show file tree

Hide file tree

Showing 6 changed files with 243 additions and 159 deletions.
diff --git a/.gitignore b/.gitignore
@@ -161,3 +161,5 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
 settings.json
+settings.local.toml
+.secrets.toml
diff --git a/settings.toml b/settings.toml
@@ -13,6 +13,7 @@ height = 1024
 seed = -1
 output_folder = ""
 replicate_model = ""
-prompt = ""
+prompt = "Enter Prompt here..."
 
-[user_models]
+[models]
+user_added = []
diff --git a/src/config.py b/src/config.py
@@ -7,3 +7,7 @@
     load_dotenv=True,
     lowercase_read=False,
 )
+
+
+def get_api_key():
+    return settings.get("REPLICATE_API_KEY") or settings.get("replicate_api_key")