Update app.py

app.py

@@ -28,22 +28,18 @@ def set_model(current_model):
     current_model = models[current_model]
     return gr.update(label=(f"{current_model}"))
 
-
 def list_saved_prompts_and_images():
     saved_prompts = os.listdir('saved_prompts')
     saved_images = os.listdir('saved_images')
 
-    prompt_image_pairs = []
+    html_str = "<h2>Saved Prompts and Images:</h2><ul>"
     for prompt_file in saved_prompts:
-        prompt_path = os.path.join('saved_prompts', prompt_file)
-        image_path = os.path.join('saved_images', f"{prompt_file[:-4]}.png")  # Remove .txt and add .png
-        if os.path.exists(image_path):
-            with open(prompt_path, 'r') as f:
-                prompt = f.read()
-            image = Image.open(image_path)
-            prompt_image_pairs.append((prompt, image))
+        image_file = f"{prompt_file[:-4]}.png"
+        if image_file in saved_images:
+            html_str += f'<li>Prompt: {prompt_file[:-4]} | <a href="saved_images/{image_file}" download>Download Image</a></li>'
+    html_str += "</ul>"
     
-    return prompt_image_pairs
+    return html_str
 
 def send_it1(inputs, model_choice):
     proc1 = models2[model_choice]
@@ -52,8 +48,8 @@ def send_it1(inputs, model_choice):
     
     # List all saved prompts and images after generating a new one
     prompt_image_pairs = list_saved_prompts_and_images()
-    
-    return output1, prompt_image_pairs
+    saved_output = list_saved_prompts_and_images()
+    return output1, saved_output
     
 css=""""""
 
@@ -76,6 +72,9 @@ with gr.Blocks(css=css) as myface:
   </head>
 </html>
 """)
+    with gr.Row():
+        with gr.Column(scale=100):
+            saved_output = gr.HTML(label="Saved Prompts and Images")
     with gr.Row():
         with gr.Tab("Title"):
                 gr.HTML("""<title>Prompt to Generate Image</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;">
@@ -134,7 +133,7 @@ with gr.Blocks(css=css) as myface:
         return(inputs)
     
     model_name1.change(set_model,inputs=model_name1,outputs=[output1])
-    run.click(send_it1, inputs=[magic1, model_name1], outputs=[output1])
+    run.click(send_it1, inputs=[magic1, model_name1], outputs=[output1, saved_output])
     use_short.click(short_prompt,inputs=[input_text],outputs=magic1)
     see_prompts.click(text_it1,inputs=[input_text],outputs=magic1)