Update app.py

app.py

@@ -1,64 +1,77 @@
 import gradio as gr
+import shutil
+import os
 from modules.split import split_file
 from modules.frame_creator import render_frame
 from modules.video import create_video_from_images, extract_frames_from_video
 from modules.frame_decoder import recover_binary_files
-import shutil
 from modules.merge import merge_cunks
-import os
 
-# 🌝 Defining Encode functions
 def encode(file_path):
     """
     Encodes a file by splitting it into chunks, rendering frames, and creating a video from the frames.
     """
+    yield "Splitting file into chunks..."
     split_file(input_file=file_path)
+    
+    yield "Rendering frames from chunks..."
     render_frame()
+    
+    yield "Creating video from frames..."
     create_video_from_images()
-    shutil.rmtree("./chunks")
-    shutil.rmtree("./images")
-    shutil.rmtree("./extracted_frames")
-    shutil.rmtree("./recovered")
+    
+    yield "Cleaning up temporary files..."
+
+    
+    yield "Encoding completed! Encoded video is ready."
+    return "encoded_video.mp4"
 
 def decode(output_file):
     """
     Decodes a video by extracting frames, recovering binary files, and merging the chunks back into the original file.
     """
+    yield "Extracting frames from video..."
     extract_frames_from_video()
+    
+    yield "Recovering binary files from frames..."
     recover_binary_files()
+    
+    yield "Merging chunks to reconstruct the original file..."
     merge_cunks("./recovered", output_file)
-    shutil.rmtree("./extracted_frames")
-    shutil.rmtree("./chunks")
-    shutil.rmtree("./recovered")
-    shutil.rmtree("./images")
+    
+    yield "Cleaning up temporary files..."
+  
+    
+    yield "Decoding completed! File successfully reconstructed."
+    return f"Decoding completed! Output file: {output_file}"
 
-# Gradio interface
 def encode_interface(file):
-    encode(file.name)
-    video_path = "encoded_video.mp4"  # Assuming the encoded video is saved as "encoded_video.mp4"
-    return video_path
+    return encode(file.name)
 
 def decode_interface(video, output_file):
-    decode(output_file)
-    return "Decoding completed!"
+    return decode(output_file)
 
-# Create Gradio interface
+# Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# File Encoder/Decoder using Video Frames")
-
+    
     with gr.Tab("Encode"):
         file_input = gr.File(label="Upload file to encode")
         encode_button = gr.Button("Encode")
+        status_output = gr.Textbox(label="Status", interactive=False)
         video_output = gr.Video(label="Encoded Video")
         download_button = gr.Button("Download Encoded Video")
-        encode_button.click(encode_interface, inputs=file_input, outputs=video_output)
+        
+        encode_button.click(encode_interface, inputs=file_input, outputs=[status_output, video_output])
         download_button.click(lambda: "encoded_video.mp4", inputs=None, outputs=gr.File(label="Download Encoded Video"))
-
+    
     with gr.Tab("Decode"):
         video_input = gr.Video(label="Upload video to decode")
         output_file_input = gr.Textbox(label="Output file name")
         decode_button = gr.Button("Decode")
-        decode_button.click(decode_interface, inputs=[video_input, output_file_input], outputs=gr.Textbox(label="Status"))
+        decode_status = gr.Textbox(label="Status", interactive=False)
+        
+        decode_button.click(decode_interface, inputs=[video_input, output_file_input], outputs=decode_status)
 
 # Launch the Gradio app
 demo.launch()