Update app.py

app.py

@@ -106,11 +106,6 @@ def predict(prompt, guidance, steps, seed=1231231):
     if nsfw_content_detected:
         nsfw=gr.Button("🕹️NSFW🎨", scale=1)
 
-    # Generate file name
-    #date_str = datetime.datetime.now().strftime("%Y%m%d")
-    #safe_prompt = prompt.replace(" ", "_")[:50]  # Truncate long prompts
-    #filename = f"{date_str}_{safe_prompt}.png"
-
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.datetime.now().strftime("%Y%m%d")
     replaced_prompt = prompt.replace(" ", "_").replace("\n", "_")
@@ -123,8 +118,6 @@ def predict(prompt, guidance, steps, seed=1231231):
         image_path = os.path.join("", filename)  # Specify your directory
         results.images[0].save(image_path)
         print(f"#Image saved as {image_path}")
-        #filename = safe_filename(prompt)
-        #image.save(filename)
         encoded_image = encode_image(image)
         html_link = f'<a href="data:image/png;base64,{encoded_image}" download="{filename}">Download Image</a>'
         gr.Markdown(html_link)
@@ -161,7 +154,7 @@ with gr.Blocks(css=css) as demo:
         # Image Result from last prompt
         image = gr.Image(type="filepath")
 
-        # Gallery
+        # Gallery of Generated Images with Image Names in Random Set to Download
         with gr.Row(variant="compact"):
             text = gr.Textbox(
                 label="Image Sets",
@@ -169,12 +162,12 @@ with gr.Blocks(css=css) as demo:
                 max_lines=1,
                 placeholder="Enter your prompt",
             )
-            btn = gr.Button("Generate image")
-
+            btn = gr.Button("Generate Gallery of Saved Images")
         gallery = gr.Gallery(
-            label="Generated images", show_label=False, elem_id="gallery"
+            label="Generated Images", show_label=False, elem_id="gallery"
         )
-        
+
+        # Advanced Generate Options
         with gr.Accordion("Advanced options", open=False):
             guidance = gr.Slider(
                 label="Guidance", minimum=0.0, maximum=5, value=0.3, step=0.001
@@ -183,6 +176,8 @@ with gr.Blocks(css=css) as demo:
             seed = gr.Slider(
                 randomize=True, minimum=0, maximum=12013012031030, label="Seed", step=1
             )
+
+        # Diffusers
         with gr.Accordion("Run with diffusers"):
             gr.Markdown(
                 """## Running LCM-LoRAs it with `diffusers`
@@ -207,6 +202,7 @@ with gr.Blocks(css=css) as demo:
             """
             )
 
+        # Function IO Eventing and Controls
         inputs = [prompt, guidance, steps, seed]
         generate_bt.click(fn=predict, inputs=inputs, outputs=image, show_progress=False)
         btn.click(fake_gan, None, gallery)