Update upload.py

upload.py

@@ -2,8 +2,24 @@ import os
 import uuid
 import requests
 import oss2
+from pydub import AudioSegment  # For audio conversion
 from requests.cookies import RequestsCookieJar
 
+def convert_to_mp3(input_path, output_path):
+    """
+    Converts an audio file to MP3 format using pydub.
+    
+    Args:
+        input_path (str): Path to the input audio file.
+        output_path (str): Path where the converted MP3 file will be saved.
+    
+    Returns:
+        None
+    """
+    sound = AudioSegment.from_file(input_path)
+    sound.export(output_path, format="mp3")
+
+
 def upload_image_to_cdn(base_url, auth_token, image_path):
     """
     Uploads an image to the CDN using STS credentials.
@@ -118,7 +134,7 @@ def upload_audio_to_cdn(base_url, auth_token, audio_path):
         def _create_headers(self):
             return {
                 'x-request-id': str(uuid.uuid4()),
-                'Authorization': f'Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6ImRjOGQyYzY4LWZjNmEtNDEwYy05NWZjLWQ5MDBmNTM4ZTMwMiIsImV4cCI6MTc0NjUyNzI2OH0.QpXyM9AgGCZwkhjrlD_KhvW0timS5y1kfukGr4v2nck',
+                'Authorization': f'Bearer {self.auth_token}',
                 'Content-Type': 'application/json',
                 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/134.0.0.0 Safari/537.36'
             }
@@ -152,19 +168,28 @@ def upload_audio_to_cdn(base_url, auth_token, audio_path):
             bucket = oss2.Bucket(auth, endpoint, sts_data['bucketname'])
             
             with open(file_path, 'rb') as file:
-                result = bucket.put_object(sts_data['file_path'], file)
+                result = bucket.put_object(
+                    sts_data['file_path'], 
+                    file, 
+                    headers={'Content-Type': 'audio/mpeg'}  # Set correct MIME type
+                )
             
             return result.status == 200
 
         def upload_file(self, file_path, filetype):
-            filename = os.path.basename(file_path)
-            filesize = os.path.getsize(file_path)
+            # Convert the audio file to MP3 format
+            mp3_path = f"{os.path.splitext(file_path)[0]}.mp3"
+            convert_to_mp3(file_path, mp3_path)
+
+            # Use the converted MP3 file for upload
+            filename = os.path.basename(mp3_path)
+            filesize = os.path.getsize(mp3_path)
             
             # Step 1: Get STS token
             sts_data = self.get_sts_token(filename, filesize, filetype)
             
             # Step 2: Upload to OSS
-            upload_success = self.upload_to_oss(sts_data, file_path)
+            upload_success = self.upload_to_oss(sts_data, mp3_path)
             
             if upload_success:
                 return sts_data['file_url']