Update app.py

app.py

@@ -2245,8 +2245,13 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
             final_image = image
             progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
             yield image, seed, gr.update(value=progress_bar, visible=True)
-            
-        yield final_image, seed, gr.update(value=progress_bar, visible=False)
+        # Save and encode final image
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
+        final_image.save(temp_file.name, "PNG")
+        with open(temp_file.name, "rb") as f:
+            base64_str = base64.b64encode(f.read()).decode("utf-8")
+
+        yield final_image, temp_file.name, base64_str, seed, gr.update(value=progress_bar, visible=False)
         
 def get_huggingface_safetensors(link):
   split_link = link.split("/")
@@ -2379,6 +2384,7 @@ with gr.Blocks(theme="YTheme/Minecraft", css=css, delete_cache=(60, 60)) as app:
         with gr.Column():
             progress_bar = gr.Markdown(elem_id="progress",visible=False)
             result = gr.Image(label="Generated Image")
+            base64_output = gr.Textbox(label="Base64 Encoded Image")
 
     with gr.Row():
         with gr.Accordion("Advanced Settings", open=False):