mirror of
https://github.com/karl0ss/ai_image_frame_server.git
synced 2025-09-08 07:23:16 +01:00
Compare commits
4 Commits
57bb0fed5b
...
fa59f3cfeb
Author | SHA1 | Date | |
---|---|---|---|
fa59f3cfeb | |||
fdd2893255 | |||
d40f6a95b0 | |||
f381fbc9c7 |
@ -1,5 +1,5 @@
|
||||
[tool.bumpversion]
|
||||
current_version = "0.3.1"
|
||||
current_version = "0.3.3"
|
||||
parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
|
||||
serialize = ["{major}.{minor}.{patch}"]
|
||||
replace = "{new_version}"
|
||||
|
@ -4,7 +4,7 @@ FROM python:3.11-slim
|
||||
# Set the working directory in the container
|
||||
WORKDIR /app
|
||||
# Set version label
|
||||
ARG VERSION="0.3.1"
|
||||
ARG VERSION="0.3.3"
|
||||
LABEL version=$VERSION
|
||||
|
||||
# Copy project files into the container
|
||||
|
@ -18,6 +18,13 @@ user_config = load_config()
|
||||
app = Flask(__name__)
|
||||
app.secret_key = os.environ.get("SECRET_KEY")
|
||||
|
||||
# Make version available to all templates
|
||||
from libs.generic import get_current_version
|
||||
@app.context_processor
|
||||
def inject_version():
|
||||
version = get_current_version()
|
||||
return dict(version=version)
|
||||
|
||||
# Inject config into routes that need it
|
||||
create_routes.init_app(user_config)
|
||||
auth_routes.init_app(user_config)
|
||||
@ -39,7 +46,16 @@ from libs.comfyui import create_image
|
||||
|
||||
def scheduled_task():
|
||||
print(f"Executing scheduled task at {time.strftime('%Y-%m-%d %H:%M:%S')}")
|
||||
create_image(None)
|
||||
# Generate a random prompt using either OpenWebUI or OpenRouter
|
||||
from libs.generic import create_prompt_with_random_model
|
||||
prompt = create_prompt_with_random_model("Generate a random detailed prompt for stable diffusion.")
|
||||
if prompt:
|
||||
# Select a random model
|
||||
import random
|
||||
model = "Random Image Model"
|
||||
create_image(prompt, model)
|
||||
else:
|
||||
print("Failed to generate a prompt for the scheduled task.")
|
||||
|
||||
if user_config["frame"]["auto_regen"] == "True":
|
||||
if os.environ.get("WERKZEUG_RUN_MAIN") == "true":
|
||||
|
@ -145,8 +145,12 @@ def create_image(prompt: str | None = None, model: str = "Random Image Model") -
|
||||
"""Generate an image with a chosen workflow (Random, FLUX*, or SDXL*)."""
|
||||
|
||||
if prompt is None:
|
||||
logging.error("No prompt provided.")
|
||||
return
|
||||
# Generate a random prompt using either OpenWebUI or OpenRouter
|
||||
from libs.generic import create_prompt_with_random_model
|
||||
prompt = create_prompt_with_random_model("Generate a random detailed prompt for stable diffusion.")
|
||||
if not prompt:
|
||||
logging.error("Failed to generate a prompt.")
|
||||
return
|
||||
|
||||
if not prompt:
|
||||
logging.error("No prompt generated.")
|
||||
|
@ -110,7 +110,7 @@ def get_current_version():
|
||||
return version
|
||||
except subprocess.CalledProcessError as e:
|
||||
print("Error running bump-my-version:", e)
|
||||
return None
|
||||
return "unknown"
|
||||
|
||||
def load_models_from_config():
|
||||
flux_models = load_config()["comfyui:flux"]["models"].split(",")
|
||||
|
@ -11,12 +11,10 @@ def index():
|
||||
image_filename = "./image.png"
|
||||
image_path = os.path.join(image_folder, image_filename)
|
||||
prompt = get_details_from_png(image_path)["p"]
|
||||
version = get_current_version()
|
||||
|
||||
return render_template(
|
||||
"index.html",
|
||||
image=image_filename,
|
||||
prompt=prompt,
|
||||
reload_interval=user_config["frame"]["reload_interval"],
|
||||
version=version,
|
||||
)
|
||||
|
@ -12,7 +12,7 @@
|
||||
|
||||
<!-- Version number at bottom right -->
|
||||
<div class="version">
|
||||
<a href="{{ url_for('settings_route.config_editor') }}">v{{ version }}</a>
|
||||
<a href="{{ url_for('settings_route.config_editor') }}">{% if version and version != 'unknown' %}v{{ version }}{% else %}v?.?.?{% endif %}</a>
|
||||
</div>
|
||||
|
||||
{% block scripts %}{% endblock %}
|
||||
|
Loading…
x
Reference in New Issue
Block a user