Update app.py

app.py

@@ -34,21 +34,28 @@ def process_file():
         factor = int(request.form['factor'])
         is_decrypting = filename.lower().endswith('.mp3')
 
-        if is_decrypting:
-            output = deconvert_file(filepath, factor)
-            output_filename = 'deconverted_file.wav'
-        else:
-            output = convert_file(filepath, factor)
-            output_filename = 'converted_file.mp3'
+        try:
+            if is_decrypting:
+                output = deconvert_file(filepath, factor)
+                output_filename = 'deconverted_file.wav'
+                mimetype = 'audio/wav'
+            else:
+                output = convert_file(filepath, factor)
+                output_filename = 'converted_file.mp3'
+                mimetype = 'audio/mpeg'
 
-        os.remove(filepath)
+            os.remove(filepath)
+
+            return send_file(
+                io.BytesIO(output),
+                mimetype=mimetype,
+                as_attachment=True,
+                download_name=output_filename
+            )
+        except Exception as e:
+            os.remove(filepath)
+            return jsonify({'error': str(e)}), 500
 
-        return send_file(
-            io.BytesIO(output),
-            mimetype='audio/mpeg' if not is_decrypting else 'audio/wav',
-            as_attachment=True,
-            download_name=output_filename
-        )
     return jsonify({'error': 'Invalid file type'}), 400
 
 def convert_file(filepath, factor):
@@ -56,7 +63,7 @@ def convert_file(filepath, factor):
         params = wav_file.getparams()
         frames = wav_file.readframes(wav_file.getnframes())
     
-    audio_data = struct.unpack(f'{len(frames)//2}h', frames)
+    audio_data = struct.unpack(f'{len(frames)//params.sampwidth}h', frames)
     modified_data = [int(sample * factor / 100) for sample in audio_data]
     
     output = io.BytesIO()
@@ -71,7 +78,7 @@ def deconvert_file(filepath, factor):
         params = wav_file.getparams()
         frames = wav_file.readframes(wav_file.getnframes())
     
-    audio_data = struct.unpack(f'{len(frames)//2}h', frames)
+    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()