Update app.py

app.py

@@ -1,10 +1,10 @@
-import os, sys, json, re, base64, random, requests, shutil
+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
+from PIL import Image as PILImage, ExifTags as PILExifTags
 import colorsys
 
 MAX_SEED = 9999
@@ -23,16 +23,15 @@ 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 = "Pollinations Image Generator"
-description = "Pollinations API + Randomizer"
-
+title = "Bilder Builder"
 
 theme = gr.themes.Soft(
-    primary_hue="amber",
+    primary_hue="cyan",
     radius_size="sm",
 )
 
@@ -55,9 +54,9 @@ def read_image_exfi_data(image_path):
     return None, image_path, img_exif_comment
 
 
-def groq_enhance_process(Prompt):
+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)
@@ -74,30 +73,22 @@ def groq_enhance_process(Prompt):
 
     if completion.choices[0].message.content != "":
         enhanced_prompt = completion.choices[0].message.content
+        enhanced_prompt = re.sub(r"[\.\"]+", "", enhanced_prompt)
 
     return enhanced_prompt
 
 
-def enhance_process(Prompt):
-    encode_prompt = urllib.parse.quote(Prompt)
-    request_url = f"https://image.pollinations.ai/prompt/{encode_prompt}?model=turbo&width=512&height=512&nologo=true&enhance=true&nofeed=true"
-    print(request_url)
-    response = requests.get(request_url)
-    if response.status_code == 200:
-        filename = datetime.now().strftime("%Y-%m-%d_%H-%M-%S") + ".png"
-        file_path = os.path.join(CACHE_DIR, filename)
-        with open(file_path, "wb") as f:
-            f.write(response.content)
-        img_exif_comment = read_exif(file_path)
-        prompt_value = img_exif_comment.get("prompt", "")
-        prompt_value = prompt_value.replace(Prompt, "").strip()
-        print(prompt_value)
-        return prompt_value
+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_colors(image_path):
+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
@@ -115,82 +106,118 @@ def image_get_colors(image_path):
     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)
 
-    timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
-    filename = timestamp + ".png"
-    # file_path = os.path.join(filename_dir, filename)
-    file_path = os.path.join(IMAGE_DIR, 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")
 
-    encode_prompt = urllib.parse.quote(Prompt)
-    request_url = f"https://image.pollinations.ai/prompt/{encode_prompt}?model=flux&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)
+        time.sleep(1)
 
-        img_exif_comment = read_exif(file_path)
-        img_dominant_color = image_get_colors(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)
+    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
 
-        return ({"value": file_path, "__type__": "update"}, {"value": request_url, "visible": False, "__type__": "update"}, used_seed, img_exif_comment, {"visible": True, "__type__": "update"}, {"value": Prompt, "visible": True, "__type__": "update"}, {"value": file_path, "visible": True, "__type__": "update"})
+        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)
 
-with gr.Blocks(theme=theme, css=custom_css) as demo:
+            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(scale=1):
                 with gr.Row():
-                    text_prompt = gr.Textbox(label="Prompt", show_label=False, lines=12, max_lines=18, placeholder="Enter your Image prompt here...", elem_classes="prompt-input", autofocus=True)
-                with gr.Row():
-                    with gr.Column(scale=2, elem_classes="image_size_selctor_wrapper"):
-                        image_width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, value=1024, step=128, elem_classes="image-width-selector")
-                        image_height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, value=683, step=128, elem_classes="image-height-selector")
-                        # with gr.Row():
-                        #     image_ratio_buttons = gr.Radio(["16:9", "4:3", "3:2", "1:1"], value="3:2", label="Bild Größe", interactive=True, elem_classes="image-ratio-buttons", container=True, show_label=False)
-                        #     switch_width_height = gr.Button("🔁", size="sm", elem_id="switchRatioBtn", elem_classes="switch-ratio-btn", variant="primary")
+                    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():
-                    with gr.Column():
-                        image_seed = gr.Slider(label="Seed", minimum=0, step=1, value=42, maximum=MAX_SEED)
-                        randomize_seed = gr.Checkbox(label="Randomize seed", value=False)
-                with gr.Row():
-                    run_button = gr.Button("Erstellen", variant="primary", elem_classes="run_button")
-                    enhance_button = gr.Button("Prompt verbessern", variant="secondary", elem_classes="enhance-button")
+                    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="", 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(scale=3):
                 with gr.Row():
-                    with gr.Column(scale=3):
-                        output_image = gr.Image(show_label=False, height=720)
                     with gr.Column(scale=1):
-                        image_informations = gr.Markdown("""## Bildinformationen""", visible=False)
-                        # textbox_your_prompt = gr.Textbox(label="Dein Prmopt", lines=2, max_lines=4, interactive=False, show_copy_button=True)
-                        textbox_prompt = gr.Textbox("Bild Prompt", lines=4, max_lines=8, interactive=False, show_copy_button=True, visible=False)
-                        image_download_button = gr.DownloadButton("Bild herunterladen", value=None, elem_classes="download-button", variant="Primary", visible=False)
-
-                # output_image = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery", columns=[1], rows=[1], object_fit="contain", height="768")
-                output_url = gr.Textbox(label="Output URL", show_label=True, interactive=False, show_copy_button=True, visible=False)
+                        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_image = gr.Image(label="Generated image", show_label=True, height=720, show_share_button=False)
-                # gallery = gr.Gallery(label="Generated Gallery", show_label=True, elem_id="gallery", columns=[3], rows=[1], object_fit="contain", height="auto", type="filepath")
-                # testimage = gr.Image(value="images/2024-08-22/2024-08-22_00-45-31.png", type="pil")
-                # gr.HTML("<div>")
-                # for image_path in os.listdir(images_dir):
-                #     gr.HTML(f'<img src="images/2024-08-21/{image_path}" />')
-                # gr.HTML("</div>")
+                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, image_seed, outpu_image_comment, image_informations, textbox_prompt, image_download_button])
-        enhance_button.click(fn=groq_enhance_process, inputs=[text_prompt], outputs=[text_prompt])
+        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()
+demo.launch()