Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import tempfile
 import shutil
 import os
 from modules.split import split_file
@@ -7,118 +8,64 @@ from modules.video import create_video_from_images, extract_frames_from_video
 from modules.frame_decoder import recover_binary_files
 from modules.merge import merge_cunks
 
-def encode(file_path):
-    """
-    Encodes a file by splitting it into chunks, rendering frames, and creating a video from the frames.
-    Streams progress updates as text.
-    """
-    total_steps = 5
-    current_step = 0
-    
-    # Step 1: Splitting file into chunks
-    yield f"Progress: {int((current_step/total_steps)*100)}% - Splitting file into chunks...", None
-    split_file(input_file=file_path)
-    current_step += 1
-    
-    # Step 2: Rendering frames from chunks
-    yield f"Progress: {int((current_step/total_steps)*100)}% - Rendering frames from chunks...", None
+# 🌝 Encoding Function
+def encode(file):
+    if not file:
+        return "No file uploaded!"
+
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_vid:
+        video_path = temp_vid.name
+
+    split_file(input_file=file)
     render_frame()
-    current_step += 1
-    
-    # Step 3: Creating video from frames
-    yield f"Progress: {int((current_step/total_steps)*100)}% - Creating video from frames...", None
-    create_video_from_images()
-    current_step += 1
-    
-    # Step 4: Cleaning up temporary files
-    yield f"Progress: {int((current_step/total_steps)*100)}% - Cleaning up temporary files...", None
+    create_video_from_images(output_video=video_path)
+
+    # Cleanup
     shutil.rmtree("./chunks", ignore_errors=True)
     shutil.rmtree("./images", ignore_errors=True)
     shutil.rmtree("./extracted_frames", ignore_errors=True)
     shutil.rmtree("./recovered", ignore_errors=True)
-    current_step += 1
-    
-    # Step 5: Completed
-    yield f"Progress: {int((current_step/total_steps)*100)}% - Encoding completed! Encoded video is ready.", "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.
-    Streams progress updates as text.
-    """
-    total_steps = 5
-    current_step = 0
-    
-    # Step 1: Extracting frames from video
-    yield f"Progress: {int((current_step/total_steps)*100)}% - Extracting frames from video...", None
-    extract_frames_from_video()
-    current_step += 1
-    
-    # Step 2: Recovering binary files from frames
-    yield f"Progress: {int((current_step/total_steps)*100)}% - Recovering binary files from frames...", None
+    return video_path
+
+# 🔄 Decoding Function
+def decode(video):
+    if not video:
+        return "No video uploaded!"
+
+    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+        output_file = temp_file.name
+
+    extract_frames_from_video(video)
     recover_binary_files()
-    current_step += 1
-    
-    # Step 3: Merging chunks to reconstruct the original file
-    yield f"Progress: {int((current_step/total_steps)*100)}% - Merging chunks to reconstruct the original file...", None
     merge_cunks("./recovered", output_file)
-    current_step += 1
-    
-    # Step 4: Cleaning up temporary files
-    yield f"Progress: {int((current_step/total_steps)*100)}% - Cleaning up temporary files...", None
+
+    # Cleanup
     shutil.rmtree("./extracted_frames", ignore_errors=True)
     shutil.rmtree("./chunks", ignore_errors=True)
     shutil.rmtree("./recovered", ignore_errors=True)
     shutil.rmtree("./images", ignore_errors=True)
-    current_step += 1
-    
-    # Step 5: Completed
-    yield f"Progress: {int((current_step/total_steps)*100)}% - Decoding completed! File successfully reconstructed.", f"Decoding completed! Output file: {output_file}"
 
-def encode_interface(file):
-    # Start encoding and stream progress updates.
-    return encode(file.name)
+    return output_file
 
-def decode_interface(video, output_file):
-    # Start decoding and stream progress updates.
-    return decode(output_file)
+# 🎨 Gradio UI
+with gr.Blocks() as app:
+    gr.Markdown("# 🏞 File to Video Encoder & Decoder")
 
-# 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")
-        progress_output = gr.Textbox(label="Encoding Progress", interactive=False)
-        video_output = gr.Video(label="Encoded Video")
-        download_button = gr.Button("Download Encoded Video")
-        
-        # Use stream=True to update progress in real time.
-        encode_button.click(
-            encode_interface, 
-            inputs=file_input, 
-            outputs=[progress_output, video_output], 
-            stream=True
-        )
-        download_button.click(
-            lambda: "encoded_video.mp4", 
-            inputs=None, 
-            outputs=gr.File(label="Download Encoded Video")
-        )
-    
+        with gr.Row():
+            file_input = gr.File(label="Upload a file to encode")
+            encode_button = gr.Button("Encode")
+        video_output = 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_progress = gr.Textbox(label="Decoding Progress", interactive=False)
-        
-        decode_button.click(
-            decode_interface, 
-            inputs=[video_input, output_file_input], 
-            outputs=decode_progress, 
-            stream=True
-        )
+        with gr.Row():
+            video_input = gr.File(label="Upload encoded video")
+            decode_button = gr.Button("Decode")
+        file_output = gr.File(label="Download decoded file")
+
+    encode_button.click(encode, inputs=[file_input], outputs=[video_output])
+    decode_button.click(decode, inputs=[video_input], outputs=[file_output])
 
-# Launch the Gradio app
-demo.launch()
+# 🚀 Run the app
+app.launch()