From eb88e7f9dc35d51ad932526f03ea82469882cb20 Mon Sep 17 00:00:00 2001
From: parth aranke <arankeparth1912@gmail.com>
Date: Thu, 6 Jun 2024 22:32:36 +0530
Subject: [PATCH] Uploadig each separate part to s3

---
 Utils.py  |  5 +----
 server.py | 15 ++++++++++++---
 2 files changed, 13 insertions(+), 7 deletions(-)

diff --git a/Utils.py b/Utils.py
index 6a8d665..47fe1aa 100644
--- a/Utils.py
+++ b/Utils.py
@@ -104,7 +104,6 @@ class ProceesAudio():
             modified_seg = add_stereo(final_audios, len(final_audios[0]), sample_rate=sr)
             modified_segs.append(modified_seg)
         
-        modified_segs = np.concatenate(modified_segs)
         return modified_segs, part_amps
 
     def perform_modulation(self, data, sr, index):
@@ -118,9 +117,7 @@ class ProceesAudio():
         index = int((int(index)/100)*len(x))
 
         y = self._get_output_amps(x, index)
-        
-        y[0][index+1:] *= max_amp
-
+    
         return self.change_amps(new_amps=y[0], sr=sr, data=data)
                 
     def get_training_data(self,  file_path, data_dir):
diff --git a/server.py b/server.py
index 5207ba0..545dbb9 100644
--- a/server.py
+++ b/server.py
@@ -21,7 +21,7 @@ def health_check():
     return jsonify({"status": "ok"}), 200
 
 @app.route('/process_and_upload', methods=['POST'])
-def upload_to_s3(processed_data, userId):
+def upload_to_s3(processed_data, userId, type):
     # Create an in-memory bytes buffer
     buffer = io.BytesIO()
 
@@ -31,7 +31,7 @@ def upload_to_s3(processed_data, userId):
 
     # Generate a unique file name
     timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-    s3_key = f'uploads/processed_audio_{userId}_{timestamp}.wav'
+    s3_key = f'uploads/processed_audio_{userId}_{timestamp}_{type}.wav'
 
     # Upload the buffer to AWS S3
     s3.upload_fileobj(buffer, BUCKET_NAME, s3_key, ExtraArgs={'ContentType': 'audio/wav'})
@@ -69,6 +69,15 @@ def process_audio_bytes(audio_bytes, index):
     pa = ProceesAudio()
 
     processed_data, part_amps = pa.perform_modulation(data=data, sr=sample_rate, index=index)
+    
+    part_urls= []
+
+    for i in range(10):
+        url = upload_to_s3(processed_data=processed_data[i], userId="parth", type="part1")
+        part_urls.append(url)
+
+    processed_data = np.concatenate(processed_data)
+
     file_url = upload_to_s3(processed_data=processed_data, userId="parth")
     arr_to_show = []
 
@@ -89,7 +98,7 @@ def process_audio_bytes(audio_bytes, index):
             temp.append(float(part[i]))
         ret.append(temp)
     
-    return jsonify({"file_url": file_url, "array": arr_to_show, "part_amps": ret})
+    return jsonify({"file_url": file_url, "array": arr_to_show, "part_amps": ret, "part_urls": part_urls})
 
 @app.route('/modify', methods=['POST'])
 def modify():