Update steganography.py

steganography.py

@@ -119,26 +119,19 @@ def gradio_image_to_audio_fn(upload_image):
 def gradio_decode_fn(upload_audio):
     return display_audio_spectrogram(upload_audio)
 
-
-def display_progress(percent, message, progress=gr.Progress()):
-    progress(percent, desc=message)
-
-def extract_audio(video_path, progress):
-    display_progress(0.1, "Extracting audio from video", progress)
+def extract_audio(video_path):
     try:
         video = VideoFileClip(video_path)
         if video.audio is None:
             raise ValueError("No audio found in the video")
         audio_path = "extracted_audio.wav"
         video.audio.write_audiofile(audio_path)
-        display_progress(0.2, "Audio extracted", progress)
         return audio_path
     except Exception as e:
-        display_progress(0.2, f"Failed to extract audio: {e}", progress)
+        logging.error(f"Failed to extract audio: {e}")
         return None
 
-def extract_frames(video_path, progress):
-    display_progress(0.3, "Extracting frames from video", progress)
+def extract_frames(video_path):
     try:
         video = cv2.VideoCapture(video_path)
         frames = []
@@ -147,10 +140,9 @@ def extract_frames(video_path, progress):
             frames.append(frame)
             success, frame = video.read()
         video.release()
-        display_progress(0.4, "Frames extracted", progress)
         return frames
     except Exception as e:
-        display_progress(0.4, f"Failed to extract frames: {e}", progress)
+        logging.error(f"Failed to extract frames: {e}")
         return None
 
 def frame_to_spectrogram(frame, sr=DEFAULT_SAMPLE_RATE):
@@ -173,7 +165,7 @@ def save_spectrogram_image(S, frame_number, temp_dir):
     plt.close()
     return image_path
 
-def process_video_frames(frames, sr=DEFAULT_SAMPLE_RATE, temp_dir=None, progress=gr.Progress()):
+def process_video_frames(frames, sr=DEFAULT_SAMPLE_RATE, temp_dir=None):
     processed_frames = []
     total_frames = len(frames)
     for i, frame in enumerate(frames):
@@ -182,8 +174,6 @@ def process_video_frames(frames, sr=DEFAULT_SAMPLE_RATE, temp_dir=None, progress
         image_path = save_spectrogram_image(S, i, temp_dir)
         processed_frame = cv2.imread(image_path)
         processed_frames.append(processed_frame)
-        display_progress(0.5 + int((i + 1) / total_frames * 0.7), f"Frame processing {i + 1}/{total_frames}", progress)
-    display_progress(0.8, "All frames processed", progress)
     return processed_frames
 
 def save_video_from_frames(frames, output_path, fps=30):
@@ -193,18 +183,16 @@ def save_video_from_frames(frames, output_path, fps=30):
         video.write(frame)
     video.release()
 
-def add_audio_to_video(video_path, audio_path, output_path, progress):
-    display_progress(0.9, "Adding audio back to video", progress)
+def add_audio_to_video(video_path, audio_path, output_path):
     try:
         video = VideoFileClip(video_path)
         audio = AudioFileClip(audio_path)
         final_video = video.set_audio(audio)
         final_video.write_videofile(output_path, codec='libx264', audio_codec='aac')
-        display_progress(1, "Video's ready", progress)
     except Exception as e:
-        display_progress(1, f"Failed to add audio to video: {e}", progress)
+        logging.error(f"Failed to add audio to video: {e}")
 
-def process_video(video_path, progress=gr.Progress()):
+def process_video(video_path):
     try:
         video = VideoFileClip(video_path)
         if video.duration > 10:
@@ -215,19 +203,19 @@ def process_video(video_path, progress=gr.Progress()):
     except Exception as e:
         return f"Failed to load video: {e}"
 
-    audio_path = extract_audio(video_path, progress)
+    audio_path = extract_audio(video_path)
     if audio_path is None:
         return "Failed to extract audio from video."
-    frames = extract_frames(video_path, progress)
+    frames = extract_frames(video_path)
     if frames is None:
         return "Failed to extract frames from video."
 
     with tempfile.TemporaryDirectory() as temp_dir:
-        processed_frames = process_video_frames(frames, temp_dir=temp_dir, progress=progress)
+        processed_frames = process_video_frames(frames, temp_dir=temp_dir)
         temp_video_path = os.path.join(temp_dir, 'processed_video.mp4')
         save_video_from_frames(processed_frames, temp_video_path)
         output_video_path = 'output_video_with_audio.mp4'
-        add_audio_to_video(temp_video_path, audio_path, output_video_path, progress)
+        add_audio_to_video(temp_video_path, audio_path, output_video_path)
         return output_video_path
 
 def create_gradio_interface():
@@ -286,4 +274,4 @@ def create_gradio_interface():
 
 if __name__ == "__main__":
     txt2spec = create_gradio_interface()
-    txt2spec.launch(share=True)
+    txt2spec.launch(share=True)