Rename packages.txt to upload.py

packages.txt

@@ -1,3 +0,0 @@
-ffmpeg
-libsm6
-libxext6

upload.py

@@ -0,0 +1,93 @@
+import os
+import uuid
+import requests
+import oss2
+from requests.cookies import RequestsCookieJar
+
+def upload_image_to_cdn(base_url, auth_token, image_path):
+    """
+    Uploads an image to the CDN using STS credentials.
+    
+    Args:
+        base_url (str): The base URL of the API endpoint (e.g., "https://chat.qwen.ai").
+        auth_token (str): The Bearer token for authentication.
+        image_path (str): The local path to the image file to upload.
+    
+    Returns:
+        str: The CDN URL where the image is accessible.
+    
+    Raises:
+        Exception: If the upload fails at any step.
+    """
+    class ImageUploader:
+        def __init__(self, base_url, auth_token):
+            self.session = requests.Session()
+            self.base_url = base_url
+            self.auth_token = auth_token
+            self.headers = self._create_headers()
+            self.cookies = RequestsCookieJar()
+
+        def _create_headers(self):
+            return {
+                'x-request-id': str(uuid.uuid4()),
+                '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'
+            }
+
+        def get_sts_token(self, filename, filesize, filetype):
+            endpoint = '/api/v1/files/getstsToken'
+            payload = {
+                'filename': filename,
+                'filesize': filesize,
+                'filetype': filetype
+            }
+            
+            response = self.session.post(
+                f'{self.base_url}{endpoint}',
+                json=payload,
+                headers=self.headers,
+                cookies=self.cookies
+            )
+            
+            response.raise_for_status()
+            return response.json()
+
+        def upload_to_oss(self, sts_data, image_path):
+            auth = oss2.StsAuth(
+                sts_data['access_key_id'],
+                sts_data['access_key_secret'],
+                sts_data['security_token']
+            )
+            
+            endpoint = f"https://{sts_data['region']}.aliyuncs.com"
+            bucket = oss2.Bucket(auth, endpoint, sts_data['bucketname'])
+            
+            with open(image_path, 'rb') as file:
+                result = bucket.put_object(sts_data['file_path'], file)
+            
+            return result.status == 200
+
+        def upload_image(self, image_path):
+            filename = os.path.basename(image_path)
+            filesize = os.path.getsize(image_path)
+            filetype = 'image'  # Modify if you need specific MIME type detection
+            
+            # 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, image_path)
+            
+            if upload_success:
+                return sts_data['file_url']
+            else:
+                raise Exception("Upload failed")
+
+    # Initialize the uploader and perform the upload
+    uploader = ImageUploader(base_url, auth_token)
+    try:
+        file_url = uploader.upload_image(image_path)
+        return file_url
+    except Exception as e:
+        raise Exception(f"Image upload failed: {str(e)}")