Create app.py

app.py

@@ -0,0 +1,155 @@
+import requests
+from PIL import Image
+import oss2
+import os
+import gradio as gr
+import uuid
+import base64
+from oss2.credentials import EnvironmentVariableCredentialsProvider,StaticCredentialsProvider
+
+# 下载网络图片
+def download_image(url):
+    # 通过 URL 下载图片
+    response = requests.get(url)
+    if response.status_code == 200:
+        return Image.open(BytesIO(response.content))
+    else:
+        raise Exception(f"Failed to download image, status code: {response.status_code}")
+
+
+def remove_metadata(image, output_path=None):
+    # 获取图片的格式
+    img_format = image.format.lower()
+
+    # 如果没有指定输出路径，自动生成路径
+    if output_path is None:
+        output_path = f"downloaded_image_no_metadata.{img_format}"
+
+    # 根据图片的格式保存图片，不包含元数据
+    image.save(output_path, img_format)
+
+    return output_path
+
+# 上传到阿里云 OSS
+def upload_to_oss(file_path, object_name, bucket_name, access_key_id, access_key_secret, security_token, endpoint):
+    try:
+        auth = oss2.ProviderAuthV4(StaticCredentialsProvider(access_key_id,access_key_secret,security_token))
+        # auth = oss2.ProviderAuthV4(access_key_id, access_key_secret, security_token)
+        # 填写Bucket名称。
+        bucket = oss2.Bucket(auth, f'https://{endpoint}', bucket_name, region=endpoint.split('.')[0].split('-', 1)[1])
+        result = bucket.put_object_from_file(object_name, file_path)
+        # 返回上传的文件 URL
+        return f"https://{bucket_name}.{endpoint}/{object_name}"
+    except oss2.exceptions.OssError as e:
+        print(f"OSS上传失败: {e}")
+        return None
+
+# 删除本地文件
+def delete_local_file(file_path):
+    if os.path.exists(file_path):
+        os.remove(file_path)
+        print(f"{file_path} 已删除。")
+    else:
+        print(f"{file_path} 文件不存在。")
+
+
+def run(url,quality,output_format,storage,access_key_id,access_key_secret,securityToken,endpoint,bucket_name,upload_filename=uuid.uuid4()):
+    
+    # 网络图片 URL 和本地保存路径
+    tmp_file_name = uuid.uuid4()
+    temp_original = f'{tmp_file_name}_original_image.jpg'
+    temp_compressed = f'{tmp_file_name}_compressed_image.jpg'
+    
+    # 步骤 1：下载图片到本地
+    downloaded_image = download_image(url)
+    
+    if downloaded_image:
+        if access_key_id and endpoint and bucket_name:
+            # 步骤 3：上传到阿里云 OSS
+            oss_url = upload_to_oss(compressed_image, upload_filename, bucket_name, access_key_id, access_key_secret,securityToken, endpoint)
+        
+            # 步骤 4：删除本地的临时文件
+            delete_local_file(temp_original)  # 删除原始图片
+            delete_local_file(temp_compressed)  # 删除压缩后的图片
+            if oss_url:
+                return f"图片上传成功，URL：{oss_url}"
+            else:
+                return "上传到 OSS 失败"
+        else:
+            with open(compressed_image,'rb') as f:
+                image_base64 = base64.b64encode(f.read())
+            delete_local_file(temp_original)  # 删除原始图片
+            delete_local_file(temp_compressed)  # 删除压缩后的图片
+            return image_base64
+    else:
+        return '图片下载失败'
+
+with gr.Blocks() as demo:
+    
+    with gr.Column(elem_id="col-container"):
+        with gr.Row():
+            url = gr.Text(
+                label="图片地址",
+                show_label=False,
+                max_lines=1,
+                placeholder="input image url",
+                container=False,
+            ) 
+        with gr.Row():
+            access_key_id = gr.Textbox(
+                label="AccessKeyId",
+                placeholder="云存储 AccessKeyId",
+            ) 
+        with gr.Row():
+            access_key_secret = gr.Textbox(
+                label="AccessKeySecret",
+                placeholder="云存储 AccessKeySecret",
+            ) 
+        with gr.Row():
+            securityToken = gr.Textbox(
+                label="SecurityToken",
+                placeholder="云存储 securityToken",
+            ) 
+        with gr.Row():
+            endpoint = gr.Textbox(
+                label="Endpoint",
+                placeholder="云存储 endpoint",
+            ) 
+        with gr.Row():
+            bucket_name = gr.Textbox(
+                label="BucketName",
+                placeholder="云存储 bucket_name",
+            ) 
+        with gr.Row():
+            upload_filename = gr.Textbox(
+                label="uploadFielName",
+                placeholder="云存储 文件名称",
+            ) 
+
+        with gr.Row():
+            output = gr.Textbox(
+                label="Output",
+                placeholder="Result will be displayed here",
+                lines=10,
+                interactive=False
+            )
+        
+        with gr.Row():
+            run_button = gr.Button("Run", scale=0)
+
+    gr.on(
+        triggers=[run_button.click],
+        fn = run,
+        inputs = [
+            url,
+            access_key_id,
+            access_key_secret,
+            securityToken,
+            endpoint,
+            bucket_name,
+            upload_filename
+        ],
+        outputs = [output]
+    )
+
+demo.queue().launch()