Update app.py

app.py

@@ -4,6 +4,17 @@ import sys
 from pathlib import Path
 from PIL import Image
 
+# Coder: Create directories if they don't exist
+if not os.path.exists('saved_prompts'):
+    os.makedirs('saved_prompts')
+
+if not os.path.exists('saved_images'):
+    os.makedirs('saved_images')
+    
+# Humanities: Elegant function to generate a safe filename 📝
+def generate_safe_filename(text):
+    return re.sub('[^a-zA-Z0-9]', '_', text)
+    
 def load_models_from_file(filename):
     with open(filename, 'r') as f:
         return [line.strip() for line in f]
@@ -28,6 +39,7 @@ def set_model(current_model):
     current_model = models[current_model]
     return gr.update(label=(f"{current_model}"))
 
+# Analysis: Function to list saved prompts and images 📊
 def list_saved_prompts_and_images():
     saved_prompts = os.listdir('saved_prompts')
     saved_images = os.listdir('saved_images')
@@ -41,15 +53,23 @@ def list_saved_prompts_and_images():
     
     return html_str
 
+# Coder: Modified function to save the prompt and image 🖼️
 def send_it1(inputs, model_choice):
     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()
-    saved_output = list_saved_prompts_and_images()
-    return output1, saved_output
+    safe_filename = generate_safe_filename(inputs[0])
+    image_path = f"saved_images/{safe_filename}.png"
+    prompt_path = f"saved_prompts/{safe_filename}.txt"
+
+    with open(prompt_path, 'w') as f:
+        f.write(inputs[0])
+
+    Image.fromarray(output1).save(image_path)
+
+    return output1
+
+
     
 css=""""""
 
@@ -72,9 +92,11 @@ 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;">
@@ -132,10 +154,13 @@ with gr.Blocks(css=css) as myface:
     def short_prompt(inputs):
         return(inputs)
     
-    model_name1.change(set_model,inputs=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)
-    
+use_short.click(short_prompt,inputs=[input_text],outputs=magic1)
+see_prompts.click(text_it1,inputs=[input_text],outputs=magic1)
+
+# Reasoning: Link functions to Gradio components 🎛️
+model_name1.change(set_model, inputs=model_name1, outputs=[output1])
+run.click(send_it1, inputs=[magic1, model_name1], outputs=[output1])
+saved_output.set(list_saved_prompts_and_images)
+
 myface.queue(concurrency_count=200)
 myface.launch(inline=True, show_api=False, max_threads=400)