Update app.py

app.py

@@ -5,7 +5,6 @@ import struct
 import logging
 from flask import Flask, render_template, request, send_file, jsonify
 from werkzeug.utils import secure_filename
-from pydub import AudioSegment
 
 app = Flask(__name__)
 
@@ -41,7 +40,7 @@ def process_file():
         logger.info(f"File saved: {filepath}")
 
         factor = int(request.form['factor'])
-        is_decrypting = filename.lower().endswith('.mp3')
+        is_decrypting = filename.lower().endswith('.wav')
         logger.info(f"Processing file: {'decrypting' if is_decrypting else 'encrypting'} with factor {factor}")
 
         try:
@@ -84,23 +83,12 @@ def encrypt_file(filepath, factor):
         wav_output.setframerate(44100)
         wav_output.writeframes(struct.pack(f'{len(samples)}h', *samples))
     
-    # Convert WAV to MP3
-    output.seek(0)
-    audio = AudioSegment.from_wav(output)
-    mp3_output = io.BytesIO()
-    audio.export(mp3_output, format="mp3")
-    
-    logger.info(f"Encryption complete. Output size: {mp3_output.getbuffer().nbytes} bytes")
-    return mp3_output.getvalue(), 'encrypted.mp3', 'audio/mpeg'
+    logger.info(f"Encryption complete. Output size: {output.getbuffer().nbytes} bytes")
+    return output.getvalue(), 'encrypted.wav', 'audio/wav'
 
 def decrypt_file(filepath, factor):
     logger.info(f"Decrypting file: {filepath}")
-    audio = AudioSegment.from_mp3(filepath)
-    wav_data = io.BytesIO()
-    audio.export(wav_data, format="wav")
-    wav_data.seek(0)
-    
-    with wave.open(wav_data, 'rb') as wav_file:
+    with wave.open(filepath, 'rb') as wav_file:
         params = wav_file.getparams()
         frames = wav_file.readframes(wav_file.getnframes())