Delete app.py

app.py

@@ -1,400 +0,0 @@
-# Copyright 2025 lishiyan
-# 
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-# 
-#     https://www.apache.org/licenses/LICENSE-2.0
-# 
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import gradio as gr
-import requests
-import json
-import os
-import base64
-import tempfile
-from urllib.parse import urlparse
-from datetime import datetime
-from PIL import Image
-import io
-
-# API配置
-API_KEY = "ns19uq2DpPZ9fuaZunfFmodmmdGrh4Erqz7G9ykeosOrPrzoZ0sVwQdeX4Mx0VWG"
-API_URL = "https://api-staging.zhipuai-infra.cn/v1/images/edits"
-
-def get_image_base64(image_path):
-    """将本地图片文件转换为 base64 字符串"""
-    try:
-        with open(image_path, 'rb') as image_file:
-            image_data = image_file.read()
-            base64_str = base64.b64encode(image_data).decode('utf-8')
-            return base64_str
-    except Exception as e:
-        print(f"转换图片为 base64 时出错: {e}")
-        return None
-
-def pil_to_base64(pil_image):
-    """将PIL图像转换为base64字符串"""
-    try:
-        # 转换为RGB模式（如果需要）
-        if pil_image.mode != 'RGB':
-            pil_image = pil_image.convert('RGB')
-        
-        # 保存到内存中的字节流
-        buffer = io.BytesIO()
-        pil_image.save(buffer, format='PNG')
-        buffer.seek(0)
-        
-        # 转换为base64
-        image_data = buffer.getvalue()
-        base64_str = base64.b64encode(image_data).decode('utf-8')
-        return base64_str
-    except Exception as e:
-        print(f"转换PIL图像为 base64 时出错: {e}")
-        return None
-
-def resize_image_to_ratio(pil_image, target_ratio):
-    """调整PIL图片到指定比例"""
-    try:
-        # 解析目标比例
-        ratio_parts = target_ratio.split(':')
-        if len(ratio_parts) != 2:
-            return None
-        
-        target_width_ratio = float(ratio_parts[0])
-        target_height_ratio = float(ratio_parts[1])
-        target_aspect_ratio = target_width_ratio / target_height_ratio
-        
-        original_width, original_height = pil_image.size
-        original_aspect_ratio = original_width / original_height
-        
-        # 如果比例已经匹配，直接返回原图
-        if abs(original_aspect_ratio - target_aspect_ratio) < 0.01:
-            return pil_image
-        
-        # 计算新的尺寸
-        if target_aspect_ratio > original_aspect_ratio:
-            # 目标比例更宽，以宽度为准
-            new_width = original_width
-            new_height = int(original_width / target_aspect_ratio)
-        else:
-            # 目标比例更高，以高度为准
-            new_height = original_height
-            new_width = int(original_height * target_aspect_ratio)
-        
-        # 调整图片尺寸
-        resized_img = pil_image.resize((new_width, new_height), Image.Resampling.LANCZOS)
-        return resized_img
-        
-    except Exception as e:
-        print(f"调整图片比例时出错: {e}")
-        return None
-
-def get_template_image(target_ratio):
-    """根据比例获取模板图片"""
-    # 本地input目录路径
-    input_dir = "image_editor/input"
-    
-    # 预定义的比例映射到不同的图片文件
-    ratio_mapping = {
-        "1:1": "1-1.png",
-        "16:9": "16-9.png", 
-        "9:16": "9-16.png",
-        "3:4": "3-4.png",
-        "4:3": "4-3.png"
-    }
-    
-    if target_ratio in ratio_mapping:
-        image_path = os.path.join(input_dir, ratio_mapping[target_ratio])
-        if os.path.exists(image_path):
-            try:
-                return Image.open(image_path)
-            except Exception as e:
-                print(f"打开模板图片失败: {e}")
-                return None
-    
-    # 如果没有找到对应的模板，返回默认的1:1图片
-    default_path = os.path.join(input_dir, ratio_mapping["1:1"])
-    if os.path.exists(default_path):
-        try:
-            return Image.open(default_path)
-        except Exception as e:
-            print(f"打开默认模板图片失败: {e}")
-            return None
-    
-    return None
-
-def edit_image_with_zhipu_api(prompt, image_base64):
-    """使用智谱AI图像编辑API编辑图片"""
-    headers = {
-        'Authorization': f'Bearer {API_KEY}',
-        'Accept': 'application/json',
-        'Content-Type': 'application/json'
-    }
-    
-    data = {
-        "model": "toc-qwen-image-edit-stage",
-        "prompt": prompt,
-        "image_url": image_base64
-    }
-    
-    try:
-        response = requests.post(API_URL, headers=headers, json=data)
-        response.raise_for_status()
-        return response.json()
-    except requests.exceptions.RequestException as e:
-        print(f"请求错误: {e}")
-        return None
-    except json.JSONDecodeError as e:
-        print(f"JSON解析错误: {e}")
-        return None
-
-def download_image_from_url(image_url):
-    """从URL下载图片并返回PIL图像"""
-    try:
-        response = requests.get(image_url, stream=True)
-        response.raise_for_status()
-        
-        # 将响应内容转换为PIL图像
-        image_data = response.content
-        pil_image = Image.open(io.BytesIO(image_data))
-        return pil_image
-        
-    except Exception as e:
-        print(f"下载图片时出错: {e}")
-        return None
-
-def save_image_to_output(pil_image, prefix="edited"):
-    """保存PIL图像到output目录"""
-    try:
-        # 创建输出目录
-        output_dir = "output"
-        if not os.path.exists(output_dir):
-            os.makedirs(output_dir)
-        
-        # 生成文件名
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename = f"{prefix}_{timestamp}.png"
-        save_path = os.path.join(output_dir, filename)
-        
-        # 保存图片
-        pil_image.save(save_path, 'PNG')
-        print(f"图片已保存到: {save_path}")
-        return save_path
-        
-    except Exception as e:
-        print(f"保存图片时出错: {e}")
-        return None
-
-def img2img_edit(input_image, prompt, target_ratio, resize_option):
-    """图生图功能"""
-    if input_image is None:
-        return None, "请上传一张图片"
-    
-    if not prompt.strip():
-        return None, "请输入编辑提示词"
-    
-    try:
-        # 处理图片
-        processed_image = input_image
-        
-        # 根据用户选择调整图片比例
-        if resize_option == "调整到指定比例" and target_ratio != "保持原比例":
-            processed_image = resize_image_to_ratio(input_image, target_ratio)
-            if processed_image is None:
-                return None, "图片比例调整失败"
-        
-        # 转换为base64
-        image_base64 = pil_to_base64(processed_image)
-        if not image_base64:
-            return None, "图片转换失败"
-        
-        # 调用API
-        result = edit_image_with_zhipu_api(prompt, image_base64)
-        
-        if result and 'data' in result and len(result['data']) > 0:
-            # 获取第一张编辑后的图片
-            if 'url' in result['data'][0]:
-                edited_image_url = result['data'][0]['url']
-                edited_image = download_image_from_url(edited_image_url)
-                
-                if edited_image:
-                    # 保存到本地
-                    save_image_to_output(edited_image, "img2img")
-                    return edited_image, "图片编辑成功！"
-                else:
-                    return None, "下载编辑后的图片失败"
-            else:
-                return None, "API响应中没有找到图片URL"
-        else:
-            return None, "API调用失败或返回结果为空"
-            
-    except Exception as e:
-        return None, f"处理过程中出错: {str(e)}"
-
-def text2img_edit(prompt, target_ratio):
-    """文本到图像编辑功能（基于模板）"""
-    if not prompt.strip():
-        return None, "请输入编辑提示词"
-    
-    try:
-        # 获取模板图片
-        template_image = get_template_image(target_ratio)
-        if template_image is None:
-            return None, f"无法获取比例为 {target_ratio} 的模板图片"
-        
-        # 转换为base64
-        image_base64 = pil_to_base64(template_image)
-        if not image_base64:
-            return None, "模板图片转换失败"
-        
-        # 调用API
-        result = edit_image_with_zhipu_api(prompt, image_base64)
-        
-        if result and 'data' in result and len(result['data']) > 0:
-            # 获取第一张编辑后的图片
-            if 'url' in result['data'][0]:
-                edited_image_url = result['data'][0]['url']
-                edited_image = download_image_from_url(edited_image_url)
-                
-                if edited_image:
-                    # 保存到本地
-                    save_image_to_output(edited_image, f"text2img_{target_ratio.replace(':', '-')}")
-                    return edited_image, f"基于 {target_ratio} 模板的图片编辑成功！"
-                else:
-                    return None, "下载编辑后的图片失败"
-            else:
-                return None, "API响应中没有找到图片URL"
-        else:
-            return None, "API调用失败或返回结果为空"
-            
-    except Exception as e:
-        return None, f"处理过程中出错: {str(e)}"
-
-# 创建Gradio界面
-def create_interface():
-    with gr.Blocks(title="智谱AI图像编辑工具", theme=gr.themes.Soft()) as demo:
-        gr.Markdown("# 🎨 智谱AI图像编辑工具")
-        gr.Markdown("基于智谱AI的图像编辑功能，支持图生图��基于模板的文本到图像编辑")
-        
-        with gr.Tabs():
-            # 图生图选项卡
-            with gr.TabItem("📸 图生图"):
-                gr.Markdown("### 上传图片并根据提示词进行编辑")
-                
-                with gr.Row():
-                    with gr.Column():
-                        img2img_input = gr.Image(
-                            label="上传图片", 
-                            type="pil",
-                            height=400
-                        )
-                        img2img_prompt = gr.Textbox(
-                            label="编辑提示词",
-                            placeholder="描述你想要的编辑效果，例如：将背景改为海滩",
-                            lines=3
-                        )
-                        
-                        with gr.Row():
-                            img2img_resize = gr.Radio(
-                                choices=["保持原比例", "调整到指定比例"],
-                                value="保持原比例",
-                                label="比例选项"
-                            )
-                            img2img_ratio = gr.Dropdown(
-                                choices=["1:1", "16:9", "9:16", "4:3", "3:4"],
-                                value="1:1",
-                                label="目标比例",
-                                visible=False
-                            )
-                        
-                        img2img_btn = gr.Button("🎨 开始编辑", variant="primary", size="lg")
-                    
-                    with gr.Column():
-                        img2img_output = gr.Image(label="编辑结果", height=400)
-                        img2img_status = gr.Textbox(label="状态信息", interactive=False)
-                
-                # 显示/隐藏比例选择
-                def toggle_ratio_visibility(resize_option):
-                    return gr.update(visible=(resize_option == "调整到指定比例"))
-                
-                img2img_resize.change(
-                    toggle_ratio_visibility,
-                    inputs=[img2img_resize],
-                    outputs=[img2img_ratio]
-                )
-                
-                # 绑定编辑功能
-                img2img_btn.click(
-                    img2img_edit,
-                    inputs=[img2img_input, img2img_prompt, img2img_ratio, img2img_resize],
-                    outputs=[img2img_output, img2img_status]
-                )
-            
-            # 文本到图像编辑选项卡
-            with gr.TabItem("📝 文生图"):
-                gr.Markdown("### 基于预设模板图片进行编辑")
-                
-                with gr.Row():
-                    with gr.Column():
-                        text2img_prompt = gr.Textbox(
-                            label="编辑提示词",
-                            placeholder="描述你想要生成的图像，例如：一只可爱的小猫在花园里",
-                            lines=4
-                        )
-                        text2img_ratio = gr.Dropdown(
-                            choices=["1:1", "16:9", "9:16", "4:3", "3:4"],
-                            value="1:1",
-                            label="选择模板比例"
-                        )
-                        text2img_btn = gr.Button("🎨 开始编辑", variant="primary", size="lg")
-                    
-                    with gr.Column():
-                        text2img_output = gr.Image(label="编辑结果", height=400)
-                        text2img_status = gr.Textbox(label="状态信息", interactive=False)
-                
-                # 绑定编辑功能
-                text2img_btn.click(
-                    text2img_edit,
-                    inputs=[text2img_prompt, text2img_ratio],
-                    outputs=[text2img_output, text2img_status]
-                )
-        
-        # 添加说明信息
-        with gr.Accordion("📋 使用说明", open=False):
-            gr.Markdown("""
-            ### 功能说明：
-            
-            **图生图：**
-            - 上传任意图片
-            - 输入编辑提示词描述想要的效果
-            - 可选择保持原比例或调整到指定比例
-            - 支持的比例：1:1（正方形）、16:9（宽屏）、9:16（竖屏）、4:3（传统）、3:4（竖版传统）
-            
-            **文生图：**
-            - 基于预设的不同比例模板图片进行编辑
-            - 输入提示词描述想要生成的内容
-            - 选择合适的输出比例
-            
-            ### 注意事项：
-            - 编辑后的图片会自动保存到 `output` 目录
-            - 请确保提示词描述清晰具体
-            - API调用可能需要一些时间，请耐心等待
-            """)
-    
-    return demo
-
-if __name__ == "__main__":
-    # 创建并启动界面
-    demo = create_interface()
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7861,
-        share=False,
-        show_error=True
-    )