Spaces:
Running
Running
| import os, sys, json, re, time, base64, random, requests, shutil | |
| import gradio as gr | |
| from datetime import datetime | |
| import urllib.parse | |
| from groq import Groq | |
| from exif import Image | |
| from PIL import Image as PILImage, ExifTags as PILExifTags | |
| import colorsys | |
| MAX_SEED = 9999 | |
| MAX_IMAGE_SIZE = 2048 | |
| GROQ_APIKEY_PROMPTENHANCE = os.environ["GROQ_APIKEY_PROMPTENHANCE"] | |
| # delete cache folder if exist and create new | |
| CACHE_DIR = os.path.join(os.path.dirname(__file__), "cache") | |
| IMAGE_DIR = os.path.join(CACHE_DIR, "images") | |
| if os.path.exists(CACHE_DIR): | |
| shutil.rmtree(CACHE_DIR) | |
| os.makedirs(CACHE_DIR) | |
| os.makedirs(IMAGE_DIR) | |
| RES = os.path.join(os.path.dirname(__file__), "_res") | |
| gr.set_static_paths(paths=["_res/assets/"]) | |
| custom_css = RES + "/_custom.css" | |
| custom_js = RES + "/_custom.js" | |
| title = "Bilder Builder" | |
| theme = gr.themes.Soft( | |
| primary_hue="cyan", | |
| radius_size="sm", | |
| ) | |
| def read_exif(image_path): | |
| with open(image_path, "rb") as src: | |
| img = Image(src) | |
| img_comment = json.loads(img.user_comment) | |
| # checking if the key exists before removing | |
| if "concept" in img_comment: | |
| img_comment.pop("concept") | |
| return img_comment | |
| def read_image_exfi_data(image_path): | |
| print("Imagepath:", image_path) | |
| img_exif_make, img_exif_comment = read_exif(image_path) | |
| return None, image_path, img_exif_comment | |
| def groq_enhance_process(Prompt=""): | |
| client = Groq(api_key=GROQ_APIKEY_PROMPTENHANCE) | |
| Prompt = "random prompt" if Prompt == "" else Prompt | |
| SYSTEMPROMPT = os.path.join(RES, "groq_systemmessage_prompt_enhance.json") | |
| with open(SYSTEMPROMPT, "r") as f: | |
| SYSTEMPROMPT = json.load(f) | |
| completion = client.chat.completions.create( | |
| model="llama-3.1-70b-versatile", | |
| messages=[SYSTEMPROMPT, {"role": "user", "content": Prompt}], | |
| temperature=0.8, | |
| max_tokens=8000, | |
| top_p=0.9, | |
| stream=False, | |
| stop=None, | |
| ) | |
| if completion.choices[0].message.content != "": | |
| enhanced_prompt = completion.choices[0].message.content | |
| enhanced_prompt = re.sub(r"[\.\"]+", "", enhanced_prompt) | |
| return enhanced_prompt | |
| def image_get_size(image_path): | |
| img = PILImage.open(image_path) | |
| print("Image size:", img.size) | |
| width, height = img.size | |
| return width, height | |
| def image_get_dominant_color(image_path): | |
| img = PILImage.open(image_path) | |
| img = img.convert("RGB") | |
| img = img.resize((100, 100), resample=0) | |
| pixels = list(img.getdata()) | |
| # Erzeuge eine Liste mit den Häufigkeiten der Farben | |
| colors = [] | |
| for pixel in pixels: | |
| r, g, b = pixel | |
| h, s, v = colorsys.rgb_to_hsv(r / 255, g / 255, b / 255) | |
| if v > 0.5: # Filteriere hellere Farben aus | |
| continue | |
| if v > 0.99: # Filteriere Weiß aus | |
| continue | |
| colors.append((h, s, v)) | |
| # Ermittle die dominante Farbe | |
| dominant_color = max(colors, key=lambda x: x[2]) | |
| dominant_color_rgb = colorsys.hsv_to_rgb(dominant_color[0], dominant_color[1], dominant_color[2]) | |
| dominant_color_rgb = [int(c * 255) for c in dominant_color_rgb] | |
| dominant_color_rgb = f"rgb({dominant_color_rgb[0]}, {dominant_color_rgb[1]}, {dominant_color_rgb[2]})" | |
| print(dominant_color_rgb) | |
| return dominant_color_rgb | |
| def process(Prompt, image_width, image_height, image_seed, randomize_seed): | |
| used_seed = random.randint(0, MAX_SEED) if image_seed == 0 or randomize_seed else image_seed | |
| if "test image" in Prompt.lower() or Prompt == "": | |
| print("Prompt:", Prompt) | |
| TESTIMAGES = os.path.join(RES, "testimages") | |
| random_num = random.randint(1, 10) | |
| match = re.match(r"test image (\d+)", Prompt) | |
| if match and 1 <= int(match.group(1)) <= 10: | |
| random_num = match.group(1) | |
| print("\n\nrandom: ", random_num) | |
| Prompt = "test image " + str(random_num) if Prompt == "" else Prompt | |
| filename = str(random_num) + ".png" | |
| print("Filename:", filename) | |
| file_path = os.path.join(TESTIMAGES, filename) | |
| img_exif_comment = read_exif(file_path) | |
| img_dominant_color = image_get_dominant_color(file_path) | |
| img_width, img_height = image_get_size(file_path) | |
| print("ImageSize: " + str(img_width) + "x" + str(img_height) + "px") | |
| time.sleep(1) | |
| return ({"value": file_path, "__type__": "update"}, {"value": None, "visible": False, "__type__": "update"}, img_exif_comment, {"visible": True, "__type__": "update"}, {"value": Prompt, "visible": True, "__type__": "update"}, img_width, img_height, used_seed, {"value": file_path, "visible": True, "__type__": "update"}, img_dominant_color) | |
| else: | |
| used_seed = random.randint(0, MAX_SEED) if image_seed == 0 or randomize_seed else image_seed | |
| used_model = "turbo" if int(image_width) > 1024 or int(image_height) > 1024 else "flux" # turbo, flux | |
| timestamp = datetime.now().strftime("%Y-%m-%d-%H-%M-%S") | |
| filename_prompt = re.sub(r"[^\w\s-]", "", Prompt).strip().replace(" ", "_") | |
| filename = timestamp + "_" + filename_prompt[:100] + ".png" | |
| # file_path = os.path.join(filename_dir, filename) | |
| file_path = os.path.join(IMAGE_DIR, filename) | |
| encode_prompt = urllib.parse.quote(Prompt) | |
| request_url = f"https://image.pollinations.ai/prompt/{encode_prompt}?model={used_model}&width={image_width}&height={image_height}&nologo=true&enhance=false&nofeed=true&seed={used_seed}" | |
| print(request_url) | |
| response = requests.get(request_url) | |
| if response.status_code == 200: | |
| with open(file_path, "wb") as f: | |
| f.write(response.content) | |
| img_exif_comment = read_exif(file_path) | |
| img_dominant_color = image_get_dominant_color(file_path) | |
| img_width, img_height = image_get_size(file_path) | |
| return ({"value": file_path, "__type__": "update"}, {"value": None, "visible": False, "__type__": "update"}, img_exif_comment, {"visible": True, "__type__": "update"}, {"value": Prompt, "visible": True, "__type__": "update"}, img_width, img_height, used_seed, {"value": file_path, "visible": True, "__type__": "update"}, img_dominant_color) | |
| with gr.Blocks(theme=theme, css=custom_css, js=custom_js, title=title) as demo: | |
| with gr.Tab("Image Generator"): | |
| with gr.Row(): | |
| with gr.Column(min_width=420, scale=1): | |
| with gr.Row(): | |
| placeholder_text = "[???] Generiert dir einen zufälligen Prompt.\n[STERN] optimiert deinen eignen Prompt.\n[RUN] generiert dein Bild." | |
| text_prompt = gr.Textbox(label="Prompt", show_label=False, lines=12, max_lines=18, placeholder=placeholder_text, elem_classes="prompt-input", autofocus=True) | |
| with gr.Row(): | |
| random_prompt_button = gr.Button("", variant="secondary", elem_id="random_prompt_btn", elem_classes="random-prompt-btn", icon="_res/assets/star_light_48.png") | |
| enhance_prompt_button = gr.Button("", variant="secondary", elem_id="enhance_prompt_btn", elem_classes="enhance-prompt-btn", icon="_res/assets/star_light_48.png") | |
| run_button = gr.Button("Erstellen", variant="primary", elem_id="run_btn", elem_classes="run-btn") | |
| with gr.Row(elem_classes="image_size_selctor_wrapper"): | |
| with gr.Column(scale=1): | |
| with gr.Row(): | |
| # image_width = gr.Textbox(label="Breite", minimum=256, maximum=MAX_IMAGE_SIZE, value=1024, step=128, elem_id="image_width_selector", elem_classes="image-width-selector", scale=1) | |
| image_width = gr.Textbox(label="Breite", value=1024, lines=1, max_lines=1, elem_id="image_width_selector", elem_classes="image-width-selector", scale=1) | |
| # image_height = gr.Number(label="Höhe", minimum=256, maximum=MAX_IMAGE_SIZE, value=683, step=128, elem_id="image_height_selector", elem_classes="image-height-selector", scale=1) | |
| image_height = gr.Textbox(label="Höhe", value=683, lines=1, max_lines=1, elem_id="image_height_selector", elem_classes="image-height-selector", scale=1) | |
| with gr.Row(): | |
| image_ratio_buttons = gr.Radio(["16:9", "4:3", "3:2", "1:1"], value="3:2", label="Querformat", show_label=True, info="Die gängigsten Seitenverhältnissen.", interactive=True, elem_id="image_ratio_buttons", elem_classes="image-ratio-buttons", container=True, scale=2) | |
| switch_width_height = gr.Button("", size="sm", elem_id="switch_width_height", elem_classes="switch-ratio-btn", variant="primary", scale=1) | |
| with gr.Column(): | |
| image_seed = gr.Slider(label="Seed", info="Jeder Seed generiert ein anderes Bild mit dem selben Prompt", minimum=0, step=1, value=42, maximum=MAX_SEED) | |
| randomize_seed = gr.Checkbox(label="Randomize seed", value=False) | |
| with gr.Column(min_width=600, scale=3): | |
| with gr.Row(): | |
| with gr.Column(scale=1): | |
| with gr.Row(): | |
| output_image = gr.Image(show_label=False, height=720, min_width=320, scale=3, elem_id="output_image", type="filepath") | |
| with gr.Column(scale=1, visible=False, elem_classes="image-info-wrapper") as image_info_wrapper: | |
| # with gr.Group(visible=False) as image_info_wrapper: | |
| with gr.Group(): | |
| image_informations = gr.Markdown("""## Bildinformationen""", visible=True) | |
| # textbox_your_prompt = gr.Textbox(label="Dein Prmopt", lines=2, max_lines=4, interactive=False, show_copy_button=True) | |
| with gr.Row(elem_classes="img-size-wrapper"): | |
| image_info_tb_width = gr.Textbox(label="Breite", lines=1, max_lines=1, interactive=False, show_copy_button=True, elem_classes="image-info-tb-width") | |
| image_info_tb_height = gr.Textbox(label="Höhe", lines=1, max_lines=1, interactive=False, show_copy_button=True, elem_classes="image-info-tb-height") | |
| with gr.Row(elem_classes="img-seed-wrapper"): | |
| image_info_tb_seed = gr.Textbox(label="Seed", lines=1, max_lines=1, interactive=False, show_copy_button=True, elem_classes="image-info-tb-seed") | |
| image_info_tb_prompt = gr.Textbox("Bild Prompt", lines=4, max_lines=8, interactive=False, show_copy_button=True, visible=True) | |
| image_download_button = gr.DownloadButton("Bild herunterladen", value=None, elem_classes="download-button", variant="primary", visible=False) | |
| output_url = gr.Textbox(label="Output URL", show_label=True, interactive=False, visible=False) | |
| outpu_image_comment = gr.Json(visible=False) | |
| output_dominant_image_color = gr.Textbox(show_label=False, elem_id="dominant_image_color", visible=True, elem_classes="output-dominant-image-color") | |
| def switch_image_size_values(image_width, image_height): | |
| return image_height, image_width | |
| # switch_width_height.click(fn=switch_image_size_values, inputs=[image_width, image_height], outputs=[image_width, image_height]) | |
| run_button.click(fn=process, inputs=[text_prompt, image_width, image_height, image_seed, randomize_seed], outputs=[output_image, output_url, outpu_image_comment, image_informations, image_info_tb_prompt, image_info_tb_width, image_info_tb_height, image_info_tb_seed, image_download_button, output_dominant_image_color]) | |
| enhance_prompt_button.click(fn=groq_enhance_process, inputs=[text_prompt], outputs=[text_prompt]) | |
| random_prompt_button.click(fn=groq_enhance_process, inputs=None, outputs=[text_prompt]) | |
| demo.launch() | |