Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os 
 import sys
 from pathlib import Path
+from PIL import Image
 
 def load_models_from_file(filename):
     with open(filename, 'r') as f:
@@ -28,10 +29,32 @@ def set_model(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 = []
+    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))
+    
+    return prompt_image_pairs
+
 def send_it1(inputs, model_choice):
-        proc1=models2[model_choice]
-        output1=proc1(inputs)
-        return(output1)
+    proc1 = models2[model_choice]
+    output1 = proc1(inputs)
+    # ... Existing code for directory creation and file saving ...
+    
+    # List all saved prompts and images after generating a new one
+    prompt_image_pairs = list_saved_prompts_and_images()
+    
+    return output1, prompt_image_pairs
+    
 css=""""""
 
 
@@ -102,6 +125,10 @@ with gr.Blocks(css=css) as myface:
             input_text=gr.Textbox(label="Prompt Idea",lines=2)
             use_short=gr.Button("Use Short Prompt")
             see_prompts=gr.Button("Extend Idea")
+
+    with gr.Row():
+        with gr.Column(scale=100):
+            saved_output = gr.Custom(type="custom", label="Saved Prompts and Images")
                 
     def short_prompt(inputs):
         return(inputs)