Update app.py

app.py

@@ -5,6 +5,7 @@ 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__)
 
@@ -12,7 +13,7 @@ logging.basicConfig(level=logging.DEBUG)
 logger = app.logger
 
 UPLOAD_FOLDER = '/tmp'
-ALLOWED_EXTENSIONS = {'mp3', 'wav'}
+ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'mp3', 'wav', 'txt', 'pdf'}
 
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 
@@ -41,17 +42,13 @@ def process_file():
 
         factor = int(request.form['factor'])
         is_decrypting = filename.lower().endswith('.mp3')
-        logger.info(f"Processing file: {'deconverting' if is_decrypting else 'converting'} with factor {factor}")
+        logger.info(f"Processing file: {'decrypting' if is_decrypting else 'encrypting'} with factor {factor}")
 
         try:
             if is_decrypting:
-                output = deconvert_file(filepath, factor)
-                output_filename = 'deconverted_file.wav'
-                mimetype = 'audio/wav'
+                output, output_filename, mimetype = decrypt_file(filepath, factor)
             else:
-                output = convert_file(filepath, factor)
-                output_filename = 'converted_file.mp3'
-                mimetype = 'audio/mpeg'
+                output, output_filename, mimetype = encrypt_file(filepath, factor)
 
             logger.info(f"File processed successfully. Output size: {len(output)} bytes")
             os.remove(filepath)
@@ -71,39 +68,47 @@ def process_file():
     logger.error("Invalid file type")
     return jsonify({'error': 'Invalid file type'}), 400
 
-def convert_file(filepath, factor):
-    logger.info(f"Converting file: {filepath}")
-    with wave.open(filepath, 'rb') as wav_file:
-        params = wav_file.getparams()
-        frames = wav_file.readframes(wav_file.getnframes())
+def encrypt_file(filepath, factor):
+    logger.info(f"Encrypting file: {filepath}")
+    with open(filepath, 'rb') as file:
+        data = file.read()
     
-    audio_data = struct.unpack(f'{len(frames)//params.sampwidth}h', frames)
-    modified_data = [int(sample * factor / 100) for sample in audio_data]
+    # Convert data to audio samples
+    samples = [ord(byte) * factor for byte in data.decode('latin-1')]
     
+    # Create WAV file
     output = io.BytesIO()
     with wave.open(output, 'wb') as wav_output:
-        wav_output.setparams(params)
-        wav_output.writeframes(struct.pack(f'{len(modified_data)}h', *modified_data))
+        wav_output.setnchannels(1)
+        wav_output.setsampwidth(2)
+        wav_output.setframerate(44100)
+        wav_output.writeframes(struct.pack(f'{len(samples)}h', *samples))
     
-    logger.info(f"Conversion complete. Output size: {output.getbuffer().nbytes} bytes")
-    return output.getvalue()
+    # 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'
 
-def deconvert_file(filepath, factor):
-    logger.info(f"Deconverting file: {filepath}")
-    with wave.open(filepath, 'rb') as wav_file:
+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:
         params = wav_file.getparams()
         frames = wav_file.readframes(wav_file.getnframes())
     
-    audio_data = struct.unpack(f'{len(frames)//params.sampwidth}h', frames)
-    modified_data = [int(sample * 100 / factor) for sample in audio_data]
-    
-    output = io.BytesIO()
-    with wave.open(output, 'wb') as wav_output:
-        wav_output.setparams(params)
-        wav_output.writeframes(struct.pack(f'{len(modified_data)}h', *modified_data))
+    samples = struct.unpack(f'{len(frames)//2}h', frames)
+    decrypted_data = ''.join(chr(int(sample / factor)) for sample in samples).encode('latin-1')
     
-    logger.info(f"Deconversion complete. Output size: {output.getbuffer().nbytes} bytes")
-    return output.getvalue()
+    logger.info(f"Decryption complete. Output size: {len(decrypted_data)} bytes")
+    return decrypted_data, 'decrypted_file', 'application/octet-stream'
 
 if __name__ == '__main__':
     app.run(debug=True)