Karl/ai-frame-image-server
1
0
Fork 0
mirror of https://github.com/karl0ss/ai_image_frame_server.git synced 2025-05-30 00:25:06 +01:00
Code Issues Packages Projects Releases Wiki Activity

show version on the index page

Browse Source
This commit is contained in:
Karl 2025-05-17 11:34:13 +01:00
parent ec6ab8123d
commit 9d60135ab5
4 changed files with 46 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -7,4 +7,5 @@ dist/
user_config.cfg user_config.cfg
output/ output/
prompts_log.jsonl prompts_log.jsonl
publish.sh publish.sh
test.py

5
ai_frame_image_server.py
View File

@ -9,7 +9,7 @@ import os
import time import time
import threading import threading
from apscheduler.schedulers.background import BackgroundScheduler from apscheduler.schedulers.background import BackgroundScheduler
from libs.generic import load_config, load_recent_prompts, get_details_from_png from libs.generic import load_config, load_recent_prompts, get_details_from_png, get_current_version
from libs.comfyui import cancel_current_job, create_image from libs.comfyui import cancel_current_job, create_image
from libs.ollama import create_prompt_on_openwebui from libs.ollama import create_prompt_on_openwebui
@ -30,11 +30,14 @@ def index() -> str:
prompt = get_details_from_png(image_path)["p"] prompt = get_details_from_png(image_path)["p"]
version = get_current_version()
return render_template( return render_template(
"index.html", "index.html",
image=image_filename, image=image_filename,
prompt=prompt, prompt=prompt,
reload_interval=user_config["frame"]["reload_interval"], reload_interval=user_config["frame"]["reload_interval"],
version=version
) )
@app.route("/images", methods=["GET"]) @app.route("/images", methods=["GET"])

17
libs/generic.py
View File

@ -1,3 +1,4 @@
import subprocess
import configparser import configparser
import logging import logging
import sys import sys
@ -81,6 +82,22 @@ def get_details_from_png(path):
print(f"Error reading metadata from {path}: {e}") print(f"Error reading metadata from {path}: {e}")
return "" return ""
def get_current_version():
try:
# Run the command and capture the output
result = subprocess.run(
['bump-my-version', 'show', 'current_version'],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
text=True, # to get string output instead of bytes
check=True # raises exception if command fails
)
version = result.stdout.strip()
return version
except subprocess.CalledProcessError as e:
print("Error running bump-my-version:", e)
return None
user_config = load_config() user_config = load_config()
output_folder = user_config["comfyui"]["output_dir"] output_folder = user_config["comfyui"]["output_dir"]

27
templates/index.html
View File

@ -2,8 +2,8 @@
<html lang="en"> <html lang="en">
<head> <head>
<meta charset="UTF-8"> <meta charset="UTF-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0"> <meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>AI Image of the Day</title> <title>AI Image of the Day</title>
<style> <style>
* { * {
@ -21,6 +21,7 @@
background: black; background: black;
color: white; color: white;
font-family: Arial, sans-serif; font-family: Arial, sans-serif;
position: relative; /* So fixed elements inside work well */
} }
.image-container { .image-container {
@ -74,6 +75,19 @@
.button-link:hover { .button-link:hover {
background: #555; background: #555;
} }
/* New style for version number */
.version {
position: fixed;
bottom: 8px;
right: 12px;
color: #666;
font-size: 12px;
font-family: monospace;
user-select: none;
pointer-events: none;
opacity: 0.6;
}
</style> </style>
<script> <script>
setInterval(() => { setInterval(() => {
@ -85,7 +99,7 @@
<body> <body>
{% if image %} {% if image %}
<div class="image-container"> <div class="image-container">
<img src="{{ url_for('images', filename=image) }}" alt="Latest Image"> <img src="{{ url_for('images', filename=image) }}" alt="Latest Image" />
</div> </div>
{% if prompt %} {% if prompt %}
<div class="prompt">{{ prompt }}</div> <div class="prompt">{{ prompt }}</div>
@ -97,4 +111,9 @@
{% else %} {% else %}
<p>No images found</p> <p>No images found</p>
{% endif %} {% endif %}
</body>
<!-- Version number at bottom right -->
<div class="version">v{{ version }}</div>
</body>
</html>