Update app.py

app.py

@@ -1,886 +1,698 @@
+import gradio as gr
 import requests
 import re
 import json
 from datetime import datetime
 import os
-import gradio as gr
 from dotenv import load_dotenv
-
-# ========== LOAD ENVIRONMENT VARIABLES ==========
-def load_environment_variables():
-    """Load environment variables for both local and Hugging Face"""
-    # For Hugging Face Spaces (use environment variables)
-    huggingface_url = os.environ.get('WORDPRESS_URL')
-    huggingface_username = os.environ.get('WORDPRESS_USERNAME')
-    huggingface_password = os.environ.get('WORDPRESS_APP_PASSWORD')
-    huggingface_status = os.environ.get('DEFAULT_STATUS')
-    
-    # If Hugging Face env vars exist, use them
-    if huggingface_url and huggingface_username:
-        return {
-            'url': huggingface_url,
-            'username': huggingface_username,
-            'password': huggingface_password or '',
-            'status': huggingface_status or 'draft'
-        }
-    
-    # Fallback to local .env file for development
-    try:
-        load_dotenv(dotenv_path='rank_cd.env')
-        return {
-            'url': os.getenv('WORDPRESS_URL', 'https://cdgarment.com'),
-            'username': os.getenv('WORDPRESS_USERNAME', ''),
-            'password': os.getenv('WORDPRESS_APP_PASSWORD', ''),
-            'status': os.getenv('DEFAULT_STATUS', 'draft')
-        }
-    except Exception as e:
-        print(f"Could not load rank_cd.env: {str(e)}")
-        return {
-            'url': 'https://cdgarment.com',
-            'username': '',
-            'password': '',
-            'status': 'draft'
-        }
-
-# ========== FUNCTIONS ==========
-def parse_gemini_content(content, post_status='draft'):
-    """Parse Gemini content for cdgarment.com with Rank Math"""
-    data = {
-        'seo_title': '',
-        'primary_keyword': '',
-        'meta_description': '',
-        'tags': [],
-        'article_title': '',
-        'content': '',
-        'url_slug': '',
-        'rank_math_meta': {}
-    }
-    
-    try:
-        # Clean content
-        content = re.sub(r'\r\n', '\n', content)
-        content = re.sub(r'\n\s*\n', '\n\n', content)
-        
-        # Extract SEO Title
-        seo_title_match = re.search(r'SEO\s*Title\s*:?\s*(.+?)(?=\n|$)', content, re.IGNORECASE | re.DOTALL)
-        if seo_title_match:
-            data['seo_title'] = seo_title_match.group(1).strip()
-        
-        # Extract Primary Keyword
-        keyword_match = re.search(r'Primary\s*Keyword\s*:?\s*(.+?)(?=\n|$)', content, re.IGNORECASE | re.DOTALL)
-        if keyword_match:
-            data['primary_keyword'] = keyword_match.group(1).strip()
-        
-        # Extract Meta Description
-        meta_match = re.search(r'Meta\s*Description\s*:?\s*(.+?)(?=\n\s*\n|\nCharacter|\nTags|\n---|$)', content, re.IGNORECASE | re.DOTALL)
-        if meta_match:
-            data['meta_description'] = meta_match.group(1).strip()
-        
-        # Extract Tags
-        tags_patterns = [
-            r'Tags\s*:?\s*(.+?)(?=\n\s*\n|\n---|\n##|\n📝|$)',  # Tags: on same line
-            r'Tags\s*\n(.+?)(?=\n\s*\n|\n---|\n##|$)'  # Tags on one line, actual tags on next
+import markdown
+from PIL import Image
+from io import BytesIO
+import time
+
+# ========== 加载环境变量 ==========
+load_dotenv(dotenv_path='rank_cd.env')
+
+# ========== 初始化全局状态 ==========
+parsed_data = None
+uploaded_image = None
+image_description = ""
+wp_config = {
+    'url': os.getenv('WORDPRESS_URL', 'https://cdgarment.com'),
+    'username': os.getenv('WORDPRESS_USERNAME', ''),
+    'password': os.getenv('WORDPRESS_APP_PASSWORD', ''),
+    'status': os.getenv('DEFAULT_STATUS', 'draft')
+}
+wp_config_locked = False
+post_status = "draft"
+parse_method = "auto"
+
+# ========== 增强解析器类 ==========
+class GeminiContentParser:
+    def __init__(self):
+        self.parsers = [
+            self.parse_json_format,
+            self.parse_markdown_table_format,
+            self.parse_simple_format
         ]
-        
-        tags_str = ""
-        for pattern in tags_patterns:
-            tags_match = re.search(pattern, content, re.IGNORECASE | re.DOTALL)
+   
+    def parse(self, content: str):
+        """尝试多种解析策略"""
+        for parser in self.parsers:
+            result = parser(content)
+            if result and (result.get('article_content') or result.get('content')):
+                return result
+       
+        return None
+   
+    def parse_json_format(self, content: str):
+        """解析JSON格式"""
+        json_patterns = [
+            r'```json\s*(.*?)\s*```',
+            r'{\s*"post_id".*?}',
+            r'.*# Machine-Readable Data.*?({.*})',
+            r'.*Machine-Readable Data.*?JSON.*?({.*})',
+        ]
+       
+        for pattern in json_patterns:
+            match = re.search(pattern, content, re.DOTALL | re.IGNORECASE)
+            if match:
+                try:
+                    json_str = match.group(1)
+                except IndexError:
+                    json_str = match.group(0)
+               
+                try:
+                    data = json.loads(json_str)
+                   
+                    seo_toolkit = data.get('seo_toolkit', {})
+                   
+                    article_content = data.get('article_content', '')
+                    if not article_content:
+                        before_json = content[:match.start()].strip()
+                        after_json = content[match.end():].strip()
+                        article_content = after_json if len(after_json) > len(before_json) else before_json
+                   
+                    lines = article_content.split('\n')
+                    article_title = lines[0].strip('# ').strip() if lines else ""
+                   
+                    result = {
+                        'seo_title': seo_toolkit.get('seo_title', article_title),
+                        'primary_keyword': seo_toolkit.get('primary_keyword', ''),
+                        'secondary_keywords': seo_toolkit.get('secondary_keywords', []),
+                        'meta_description': seo_toolkit.get('meta_description', seo_toolkit.get('description', '')),
+                        'tags': seo_toolkit.get('tags', []),
+                        'article_title': article_title,
+                        'content': article_content,
+                        'post_id': data.get('post_id', ''),
+                        'character_count': seo_toolkit.get('character_count', 0),
+                        'parse_method': 'json'
+                    }
+                   
+                    if result['seo_title']:
+                        slug = result['seo_title'].lower()
+                        slug = re.sub(r'[^\w\s-]', '', slug)
+                        slug = re.sub(r'[-\s]+', '-', slug)
+                        result['url_slug'] = slug[:100]
+                   
+                    return result
+                except json.JSONDecodeError:
+                    continue
+       
+        return None
+   
+    def parse_markdown_table_format(self, content):
+        """解析Markdown表格格式"""
+        data = {
+            'seo_title': '',
+            'primary_keyword': '',
+            'meta_description': '',
+            'tags': [],
+            'article_title': '',
+            'content': '',
+            'url_slug': '',
+            'parse_method': 'markdown_table'
+        }
+       
+        try:
+            seo_title_match = re.search(r'SEO Title:\s*(.+?)(?=\n)', content)
+            if seo_title_match:
+                data['seo_title'] = seo_title_match.group(1).strip()
+           
+            keyword_match = re.search(r'Primary Keyword:\s*(.+?)(?=\n)', content)
+            if keyword_match:
+                data['primary_keyword'] = keyword_match.group(1).strip()
+           
+            meta_match = re.search(r'Meta Description:\s*(.+?)(?=\n)', content)
+            if meta_match:
+                data['meta_description'] = meta_match.group(1).strip()
+           
+            tags_match = re.search(r'Tags:\s*(.+?)(?=\n|📝|\$)', content)
             if tags_match:
                 tags_str = tags_match.group(1).strip()
-                break
-        
-        if tags_str:
-            # Clean and split tags
-            tags_list = []
-            for tag in re.split(r'[,;\n]', tags_str):
-                cleaned_tag = tag.strip()
-                if cleaned_tag and cleaned_tag.lower() not in ['', 'tags', 'tags:']:
-                    tags_list.append(cleaned_tag)
-            data['tags'] = tags_list
-        
-        # Extract Article Content
-        content_patterns = [
-            r'---\s*\n(.+?)(?=\n##\s|\n###\s|\nConclusion:|$)',  # After ---
-            r'📝\s*Article[^\n]*\n(.+?)(?=\n##\s|\n###\s|\nConclusion:|$)',  # After emoji
-            r'##\s+(.+?)(?=\n##\s|\n###\s|\nConclusion:|$)'  # After heading
-        ]
-        
-        for pattern in content_patterns:
-            article_match = re.search(pattern, content, re.DOTALL)
+                data['tags'] = [tag.strip() for tag in re.split(r'[,;]', tags_str) if tag.strip()]
+           
+            article_match = re.search(r'(?:📝|▶|●|◆).*?(?:Article|Content)[:-]?\s*(.+)', content, re.DOTALL)
             if article_match:
                 full_content = article_match.group(1).strip()
                 lines = full_content.split('\n')
                 if lines:
-                    # Remove markdown heading markers if present
-                    title = lines[0].strip()
-                    title = re.sub(r'^#+\s*', '', title)
-                    data['article_title'] = title
+                    data['article_title'] = lines[0].strip()
                     data['content'] = '\n'.join(lines[1:])
-                break
-        
-        # Generate URL slug
-        if data['seo_title']:
-            slug = data['seo_title'].lower()
-            slug = re.sub(r'[^\w\s-]', '', slug)
-            slug = re.sub(r'[-\s]+', '-', slug)
-            data['url_slug'] = slug[:100]
-        
-        # Fallback title
-        if not data['seo_title'] and data['article_title']:
-            data['seo_title'] = data['article_title']
-        
-        # Prepare Rank Math meta fields
-        data['rank_math_meta'] = {
-            'rank_math_title': data['seo_title'],
-            'rank_math_description': data['meta_description'],
-            'rank_math_robots': ['index'] if post_status == 'publish' else ['noindex'],
-            'rank_math_news_sitemap_robots': 'index',
-            'rank_math_facebook_title': data['seo_title'],
-            'rank_math_facebook_description': data['meta_description'],
-            'rank_math_twitter_title': data['seo_title'],
-            'rank_math_twitter_description': data['meta_description'],
-            'rank_math_canonical_url': '',
+           
+            if data['seo_title']:
+                slug = data['seo_title'].lower()
+                slug = re.sub(r'[^\w\s-]', '', slug)
+                slug = re.sub(r'[-\s]+', '-', slug)
+                data['url_slug'] = slug[:100]
+           
+            if not data['seo_title'] and data['article_title']:
+                data['seo_title'] = data['article_title']
+           
+            return data
+       
+        except Exception:
+            return None
+   
+    def parse_simple_format(self, content):
+        """解析简单冒号分隔格式"""
+        data = {
+            'seo_title': '',
+            'primary_keyword': '',
+            'meta_description': '',
+            'tags': [],
+            'article_title': '',
+            'content': '',
+            'url_slug': '',
+            'parse_method': 'simple'
         }
-        
-        # Add primary keyword if available
-        if data['primary_keyword']:
-            data['rank_math_meta']['rank_math_focus_keyword'] = data['primary_keyword']
-        
-        return data
-        
+       
+        try:
+            seo_title_match = re.search(r'SEO Title[:\s]+(.+)', content, re.IGNORECASE)
+            if not seo_title_match:
+                seo_title_match = re.search(r'Title[:\s]+(.+)', content, re.IGNORECASE)
+            if seo_title_match:
+                data['seo_title'] = seo_title_match.group(1).strip()
+           
+            keyword_match = re.search(r'Primary Keyword[:\s]+(.+)', content, re.IGNORECASE)
+            if not keyword_match:
+                keyword_match = re.search(r'Keyword[:\s]+(.+)', content, re.IGNORECASE)
+            if keyword_match:
+                data['primary_keyword'] = keyword_match.group(1).strip()
+           
+            meta_match = re.search(r'Meta Description[:\s]+(.+)', content, re.IGNORECASE)
+            if not meta_match:
+                meta_match = re.search(r'Description[:\s]+(.+)', content, re.IGNORECASE)
+            if meta_match:
+                data['meta_description'] = meta_match.group(1).strip()
+           
+            tags_match = re.search(r'Tags[:\s]+(.+)', content, re.IGNORECASE)
+            if tags_match:
+                tags_str = tags_match.group(1).strip()
+                data['tags'] = [tag.strip() for tag in re.split(r'[,;]', tags_str) if tag.strip()]
+           
+            article_match = re.search(r'(?:Article|Content)[:\s]+(.+)', content, re.DOTALL | re.IGNORECASE)
+            if not article_match:
+                metadata_end = content.find('\n\n')
+                if metadata_end != -1:
+                    article_content = content[metadata_end:].strip()
+                    lines = article_content.split('\n')
+                    if lines:
+                        data['article_title'] = lines[0].strip()
+                        data['content'] = '\n'.join(lines[1:])
+            else:
+                full_content = article_match.group(1).strip()
+                lines = full_content.split('\n')
+                if lines:
+                    data['article_title'] = lines[0].strip()
+                    data['content'] = '\n'.join(lines[1:])
+           
+            if data['seo_title']:
+                slug = data['seo_title'].lower()
+                slug = re.sub(r'[^\w\s-]', '', slug)
+                slug = re.sub(r'[-\s]+', '-', slug)
+                data['url_slug'] = slug[:100]
+           
+            if not data['seo_title'] and data['article_title']:
+                data['seo_title'] = data['article_title']
+           
+            return data
+       
+        except Exception:
+            return None
+
+# ========== 功能函数 ==========
+def parse_gemini_content(content):
+    global parsed_data, parse_method
+    if not content or not content.strip():
+        return None, "❌ 内容为空"
+   
+    parser = GeminiContentParser()
+    result = parser.parse(content)
+   
+    if result:
+        parse_method = result.get('parse_method', 'unknown')
+        parsed_data = result
+       
+        method_msg = {
+            'json': '✅ 使用 JSON 格式解析',
+            'markdown_table': 'ℹ️ 使用 Markdown 表格格式解析',
+            'simple': 'ℹ️ 使用简单格式解析'
+        }.get(parse_method, f'ℹ️ 使用 {parse_method} 格式解析')
+       
+        return result, method_msg
+   
+    return None, "❌ 无法用任何已知格式解析内容"
+
+def get_or_create_tag(tag_name, wp_config):
+    try:
+        auth = (wp_config['username'], wp_config['password'])
+       
+        response = requests.get(
+            f"{wp_config['url']}/wp-json/wp/v2/tags",
+            auth=auth,
+            params={'search': tag_name, 'per_page': 10}
+        )
+       
+        if response.status_code == 200:
+            tags = response.json()
+            for tag in tags:
+                if tag['name'].lower() == tag_name.lower():
+                    return tag['id']
+       
+        create_response = requests.post(
+            f"{wp_config['url']}/wp-json/wp/v2/tags",
+            auth=auth,
+            json={'name': tag_name}
+        )
+       
+        if create_response.status_code == 201:
+            return create_response.json()['id']
+       
+        return None
+       
     except Exception as e:
-        print(f"Error parsing: {str(e)}")
-        return {'error': f"Error parsing: {str(e)}"}
+        return f"❌ 处理标签 '{tag_name}' 时出错: {str(e)}"
 
 def upload_image_to_wordpress(image_file, wp_config, filename_slug):
-    """Upload image to WordPress with auto-naming"""
     try:
-        # Handle file path from Gradio
-        if isinstance(image_file, str):
-            with open(image_file, 'rb') as f:
-                image_data = f.read()
-            file_extension = image_file.split('.')[-1].lower()
-        else:
-            # Handle file object
-            image_data = image_file.read()
-            file_extension = image_file.name.split('.')[-1].lower()
-        
-        # Generate filename from slug
-        filename = f"{filename_slug}.{file_extension}"
-        
-        # Prepare image data
+        img = Image.open(image_file)
+       
+        if img.mode == 'RGBA':
+            background = Image.new('RGB', img.size, (255, 255, 255))
+            if 'A' in img.getbands():
+                background.paste(img, mask=img.split()[-1])
+            else:
+                background.paste(img)
+            img = background
+        elif img.mode != 'RGB':
+            img = img.convert('RGB')
+       
+        max_size = 2000
+        if max(img.size) > max_size:
+            ratio = max_size / max(img.size)
+            new_size = tuple([int(dim * ratio) for dim in img.size])
+            img = img.resize(new_size, Image.Resampling.LANCZOS)
+       
+        buffer = BytesIO()
+        img.save(buffer, format='JPEG', quality=90, optimize=True)
+        image_data = buffer.getvalue()
+       
+        filename = f"{filename_slug}.jpg"
+       
         files = {
-            'file': (filename, image_data, f'image/{file_extension}')
+            'file': (filename, image_data, 'image/jpeg')
         }
-        
+       
         auth = (wp_config['username'], wp_config['password'])
-        
-        # Upload to WordPress
-        response = requests.post(
-            f"{wp_config['url']}/wp-json/wp/v2/media",
-            auth=auth,
-            files=files
-        )
-        
-        if response.status_code == 201:
-            media_data = response.json()
-            
-            # Update alt text with SEO title
-            update_response = requests.post(
-                f"{wp_config['url']}/wp-json/wp/v2/media/{media_data['id']}",
-                auth=auth,
-                json={
-                    'alt_text': filename_slug.replace('-', ' ').title(),
-                    'caption': filename_slug.replace('-', ' ').title(),
-                    'description': f"Featured image for: {filename_slug.replace('-', ' ').title()}"
-                }
-            )
-            
-            return media_data['id']
-        else:
-            print(f"Image upload failed: {response.text}")
-            return None
-            
+       
+        max_retries = 3
+        retry_delay = 2
+       
+        for attempt in range(max_retries):
+            try:
+                response = requests.post(
+                    f"{wp_config['url']}/wp-json/wp/v2/media",
+                    auth=auth,
+                    files=files,
+                    timeout=60,
+                    verify=False
+                )
+               
+                if response.status_code == 201:
+                    media_data = response.json()
+                    try:
+                        update_response = requests.post(
+                            f"{wp_config['url']}/wp-json/wp/v2/media/{media_data['id']}",
+                            auth=auth,
+                            json={
+                                'alt_text': image_description,
+                                'caption': filename_slug.replace('-', ' ').title(),
+                                'description': f"Featured image for: {filename_slug.replace('-', ' ').title()}"
+                            },
+                            timeout=10
+                        )
+                    except Exception:
+                        pass
+                   
+                    return media_data['id'], f"✅ 图片上传成功！(ID: {media_data['id']})"
+               
+                elif response.status_code == 413:
+                    return None, "❌ 图片太大，请压缩后重试"
+               
+                elif response.status_code == 401:
+                    return None, "❌ 认证失败，请检查 WordPress 凭据"
+               
+                else:
+                    if attempt < max_retries - 1:
+                        time.sleep(retry_delay * (attempt + 1))
+                        continue
+                    else:
+                        return None, f"❌ 图片上传失败，状态码: {response.status_code}"
+                       
+            except requests.exceptions.Timeout:
+                if attempt < max_retries - 1:
+                    time.sleep(retry_delay * (attempt + 1))
+                    continue
+                else:
+                    return None, "❌ 图片上传超时"
+                   
+            except requests.exceptions.ConnectionError:
+                if attempt < max_retries - 1:
+                    time.sleep(retry_delay * (attempt + 1))
+                    continue
+                else:
+                    return None, "❌ 连接失败，请检查网络和服务器"
+                   
+            except Exception as e:
+                return None, f"❌ 上传时出错: {str(e)}"
+               
+        return None, "❌ 图片上传失败"
+           
     except Exception as e:
-        print(f"Error uploading image: {str(e)}")
-        return None
+        return None, f"❌ 处理图片时出错: {str(e)}"
 
-def create_wordpress_post(parsed_data, wp_config, media_id=None, post_status='draft'):
-    """Create WordPress post with Rank Math meta"""
+def create_wordpress_post(parsed_data, wp_config, media_id=None):
     try:
-        # Base post data
+        rank_math_meta = {
+            'rank_math_title': parsed_data.get('seo_title', ''),
+            'rank_math_description': parsed_data.get('meta_description', ''),
+            'rank_math_robots': ['index'] if post_status == 'publish' else ['noindex'],
+            'rank_math_news_sitemap_robots': 'index',
+            'rank_math_facebook_title': parsed_data.get('seo_title', ''),
+            'rank_math_facebook_description': parsed_data.get('meta_description', ''),
+            'rank_math_twitter_title': parsed_data.get('seo_title', ''),
+            'rank_math_twitter_description': parsed_data.get('meta_description', ''),
+            'rank_math_canonical_url': '',
+        }
+       
+        if parsed_data.get('primary_keyword'):
+            rank_math_meta['rank_math_focus_keyword'] = parsed_data['primary_keyword']
+       
+        html_content = markdown.markdown(parsed_data.get('content', ''))
+       
         post_data = {
-            'title': parsed_data['seo_title'],
-            'content': parsed_data['content'],
-            'slug': parsed_data['url_slug'],
+            'title': parsed_data.get('seo_title', 'Untitled'),
+            'content': html_content,
+            'slug': parsed_data.get('url_slug', ''),
             'status': post_status,
-            'meta': parsed_data['rank_math_meta']
+            'meta': rank_math_meta
         }
-        
-        # Add tags if available
-        if parsed_data['tags']:
-            post_data['tags'] = parsed_data['tags']
-        
-        # Add category (default to uncategorized)
+       
+        if parsed_data.get('tags'):
+            tag_ids = []
+            for tag_name in parsed_data['tags']:
+                tag_id = get_or_create_tag(tag_name, wp_config)
+                if tag_id and isinstance(tag_id, int):
+                    tag_ids.append(tag_id)
+           
+            if tag_ids:
+                post_data['tags'] = tag_ids
+       
         post_data['categories'] = [1]
-        
-        # Add featured image
+       
         if media_id:
             post_data['featured_media'] = media_id
-            
-            # Try to set social images
+           
             try:
-                # Get media URL
                 media_response = requests.get(
                     f"{wp_config['url']}/wp-json/wp/v2/media/{media_id}",
                     auth=(wp_config['username'], wp_config['password'])
                 )
                 if media_response.status_code == 200:
                     media_url = media_response.json().get('source_url')
-                    post_data['meta']['rank_math_facebook_image'] = media_url
-                    post_data['meta']['rank_math_twitter_image'] = media_url
+                    rank_math_meta['rank_math_facebook_image'] = media_url
+                    rank_math_meta['rank_math_twitter_image'] = media_url
             except:
                 pass
-        
-        # Send to WordPress
+       
         response = requests.post(
             f"{wp_config['url']}/wp-json/wp/v2/posts",
             auth=(wp_config['username'], wp_config['password']),
             json=post_data,
             headers={'Content-Type': 'application/json'}
         )
-        
+       
         if response.status_code == 201:
-            # Update canonical URL with actual post URL
             post_result = response.json()
             update_data = {
                 'meta': {
                     'rank_math_canonical_url': post_result['link']
                 }
             }
-            
+           
             update_response = requests.post(
                 f"{wp_config['url']}/wp-json/wp/v2/posts/{post_result['id']}",
                 auth=(wp_config['username'], wp_config['password']),
                 json=update_data
             )
-            
-            return post_result
+           
+            return post_result, None
         else:
-            print(f"Post creation failed: {response.text}")
-            return None
-            
+            return None, f"WordPress API 错误: {response.text}"
+           
     except Exception as e:
-        print(f"Error creating post: {str(e)}")
-        return None
+        return None, f"创建文章时出错: {str(e)}"
+
+# ========== Gradio UI 回调函数 ==========
+def parse_content_callback(content):
+    global parsed_data
+    result, msg = parse_gemini_content(content)
+    if result:
+        preview = f"""
+        ### 📊 解析结果
+        **标题:** {result['seo_title']}
+        **主关键词:** {result.get('primary_keyword', '无')}
+        **标签:** {', '.join(result.get('tags', []))}
+        **URL Slug:** {result.get('url_slug', '自动生成')}
+        **内容长度:** {len(result.get('content', ''))} 字符
+        """
+        return msg, preview
+    return msg, ""
+
+def update_image_description_callback(image):
+    global uploaded_image, image_description
+    if image:
+        uploaded_image = image
+        filename = image.name
+        name_without_ext = '.'.join(filename.split('.')[:-1])
+        readable_name = name_without_ext.replace('_', ' ').replace('-', ' ').title()
+        image_description = f"Featured image: {readable_name}"
+        return image_description, gr.update(value=image_description)
+    return "", gr.update()
+
+def save_config_callback(url, username, password, status):
+    global wp_config, wp_config_locked, post_status
+    if not wp_config_locked:
+        wp_config = {
+            'url': url.rstrip('/'),
+            'username': username,
+            'password': password,
+            'status': status
+        }
+        post_status = status
+        return "✅ 配置已保存！"
+    return "🔒 配置已锁定，无法保存"
 
-# ========== GRADIO UI FUNCTIONS ==========
-def parse_content(gemini_content, post_status):
-    """Parse Gemini content and return preview"""
-    if not gemini_content or not gemini_content.strip():
-        return "Please paste content first", "", "", "", "", "", "", {}
-    
-    parsed = parse_gemini_content(gemini_content, post_status)
-    
-    if 'error' in parsed:
-        return parsed['error'], "", "", "", "", "", "", {}
-    
-    # Ensure all fields exist
-    parsed.setdefault('seo_title', 'No title')
-    parsed.setdefault('meta_description', '')
-    parsed.setdefault('primary_keyword', '')
-    parsed.setdefault('tags', [])
-    parsed.setdefault('content', '')
-    parsed.setdefault('url_slug', '')
-    
-    # Format tags for display
-    if parsed['tags'] and len(parsed['tags']) > 0:
-        tags_display = ", ".join(parsed['tags'][:8])
-    else:
-        tags_display = "No tags found in content"
-    
-    # Format meta description preview
-    if parsed['meta_description']:
-        meta_preview = parsed['meta_description'][:120] + "..." if len(parsed['meta_description']) > 120 else parsed['meta_description']
-    else:
-        meta_preview = "No meta description found"
-    
-    return (
-        f"✅ Parsed: {parsed['seo_title'][:50]}..." if len(parsed['seo_title']) > 50 else f"✅ Parsed: {parsed['seo_title']}",
-        parsed['seo_title'] or "No title extracted",
-        parsed['url_slug'] or "No slug generated",
-        meta_preview,
-        parsed['primary_keyword'] or "No primary keyword",
-        tags_display,
-        f"Content Length: {len(parsed['content'])} characters",
-        parsed
-    )
+def toggle_lock_callback(locked):
+    global wp_config_locked
+    wp_config_locked = locked
+    return gr.update(interactive=not locked), gr.update(interactive=not locked), gr.update(interactive=not locked)
 
-def publish_post(parsed_data, wp_config, image_file, post_status, wp_url, wp_username, wp_password):
-    """Publish post to WordPress"""
-    if not parsed_data:
-        yield "❌ No parsed content available. Please parse content first.", None
-        return
+def publish_callback():
+    global parsed_data, uploaded_image, wp_config, post_status
     
-    # Use stored config or provided inputs
-    if not wp_config:
-        wp_config = {
-            'url': wp_url.rstrip('/') if wp_url else '',
-            'username': wp_username,
-            'password': wp_password,
-            'status': post_status
-        }
+    if not parsed_data:
+        return "❌ 请先解析内容", "", ""
     
-    # Validate config
-    if not all([wp_config['url'], wp_config['username'], wp_config['password']]):
-        yield "❌ WordPress configuration incomplete. Please provide URL, username, and password.", None
-        return
+    if not all([wp_config.get('url'), wp_config.get('username'), wp_config.get('password')]):
+        return "❌ WordPress 配置不完整", "", ""
     
-    try:
-        yield "🖼️ Uploading image to WordPress...", None
-        
-        # Step 1: Upload image
-        media_id = None
-        if image_file:
-            media_id = upload_image_to_wordpress(
-                image_file,
-                wp_config,
-                parsed_data['url_slug']
-            )
-        
-        yield "📤 Creating post with Rank Math meta...", None
-        
-        # Step 2: Create post
-        result = create_wordpress_post(
-            parsed_data,
+    yield "🖼️ 正在上传图片到 WordPress...", "", ""
+    media_id = None
+    if uploaded_image:
+        media_id, img_msg = upload_image_to_wordpress(
+            uploaded_image,
             wp_config,
-            media_id,
-            post_status
+            parsed_data.get('url_slug', 'post')
         )
-        
-        if result:
-            # Create download data
-            export_data = {
-                'post_id': result['id'],
-                'title': result['title']['rendered'],
-                'link': result['link'],
-                'status': result['status'],
-                'slug': result['slug'],
-                'published_at': datetime.now().isoformat()
-            }
-            
-            # Save JSON to file
-            json_file = f"post_{result['id']}.json"
-            with open(json_file, 'w') as f:
-                json.dump(export_data, f, indent=2)
-            
-            success_msg = f"""
-            ✅ Post published successfully!
-            
-            Post ID: {result['id']}
-            Status: {result['status']}
-            Date: {result['date'][:10]}
-            Link: {result['link']}
-            
-            Click the download button to get post data.
-            """
-            
-            yield success_msg, json_file
-        else:
-            yield "❌ Failed to publish post. Check WordPress configuration and try again.", None
-            
-    except Exception as e:
-        yield f"❌ Error: {str(e)}", None
-
-def clear_all():
-    """Clear all inputs"""
-    return (
-        "",  # gemini_content
-        None,  # image
-        "https://cdgarment.com",  # wp_url
-        "",  # wp_username
-        "",  # wp_password
-        "draft",  # post_status
-        "",  # parse_status
-        "",  # seo_title
-        "",  # url_slug
-        "",  # meta_description
-        "",  # primary_keyword
-        "",  # tags
-        "",  # content_length
-        "",  # publish_status
-        None,  # download_file
-    )
+        if not media_id:
+            return f"❌ {img_msg}", "", ""
+    
+    yield "📤 正在创建文章并设置 Rank Math 元数据...", "", ""
+    result, error = create_wordpress_post(parsed_data, wp_config, media_id)
+    
+    if result:
+        yield f"✅ 文章{'已发布' if post_status == 'publish' else '已保存为草稿'}！(ID: {result['id']})", f"""
+        ### 📝 发布成功
+        **文章ID:** {result['id']}
+        **状态:** {result['status']}
+        **日期:** {result['date'][:10]}
+        **链接:** {result['link']}
+        """, json.dumps({
+            'post_id': result['id'],
+            'title': result['title']['rendered'],
+            'link': result['link'],
+            'status': result['status'],
+            'slug': result['slug'],
+            'published_at': datetime.now().isoformat(),
+            'parse_method': parse_method
+        }, indent=2)
+    else:
+        yield f"❌ 发布失败: {error}", "", ""
 
-# ========== CREATE GRADIO INTERFACE ==========
-with gr.Blocks(title="CdGarment WordPress Publisher") as demo:
-    
-    # State variables
-    parsed_data_state = gr.State(None)
-    wp_config_state = gr.State(None)
+# ========== Gradio UI 布局 ==========
+with gr.Blocks(title="CdGarment WordPress Publisher", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🏭 CdGarment WordPress Publisher")
+    gr.Markdown("### ✓ 专为 Rank Math SEO 优化")
     
-    # Header
-    gr.HTML("""
-    <div style="text-align: center; margin: 20px 0;">
-        <h1 style="
-            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-            -webkit-background-clip: text;
-            -webkit-text-fill-color: transparent;
-            font-size: 2.5rem;
-            font-weight: 800;
-            margin-bottom: 10px;
-        ">🏭 CdGarment WordPress Publisher</h1>
-        <div style="
-            background: linear-gradient(135deg, #fee2e2 0%, #fecaca 100%);
-            color: #991B1B;
-            padding: 8px 16px;
-            border-radius: 20px;
-            font-weight: 600;
-            display: inline-block;
-            margin-bottom: 20px;
-        ">✓ Optimized for Rank Math SEO</div>
-    </div>
-    """)
-    
-    # Two main columns
     with gr.Row():
         with gr.Column(scale=2):
-            with gr.Group():
-                gr.Markdown("### 📋 Paste Gemini Content")
-                gr.Markdown("Include SEO package and article. Formatting preserved!")
-                
-                # Formatting toolbar
-                gr.HTML("""
-                <div style="background: #f8f9fa; padding: 10px; border-radius: 5px; margin-bottom: 10px; border: 1px solid #e9ecef;">
-                    <strong>Formatting:</strong>
-                    <button class="format-btn" data-wrap="# " data-type="H1" style="background: white; border: 1px solid #dee2e6; padding: 5px 10px; margin-right: 5px; border-radius: 4px; cursor: pointer;">H1</button>
-                    <button class="format-btn" data-wrap="## " data-type="H2" style="background: white; border: 1px solid #dee2e6; padding: 5px 10px; margin-right: 5px; border-radius: 4px; cursor: pointer;">H2</button>
-                    <button class="format-btn" data-wrap="**" data-type="Bold" style="background: white; border: 1px solid #dee2e6; padding: 5px 10px; margin-right: 5px; border-radius: 4px; cursor: pointer; font-weight: bold;">B</button>
-                    <button class="format-btn" data-wrap="*" data-type="Italic" style="background: white; border: 1px solid #dee2e6; padding: 5px 10px; margin-right: 5px; border-radius: 4px; cursor: pointer; font-style: italic;">I</button>
-                    <button class="format-btn" data-wrap="- " data-type="List" style="background: white; border: 1px solid #dee2e6; padding: 5px 10px; margin-right: 5px; border-radius: 4px; cursor: pointer;">• List</button>
-                </div>
-
-                <script>
-                function formatText(wrapText, type) {
-                    const textarea = document.querySelector('textarea[aria-label="Gemini Content"]');
-                    if (!textarea) return;
-                    
-                    const start = textarea.selectionStart;
-                    const end = textarea.selectionEnd;
-                    const selected = textarea.value.substring(start, end);
-                    
-                    if (selected) {
-                        textarea.value = textarea.value.substring(0, start) + wrapText + selected + wrapText + textarea.value.substring(end);
-                        textarea.selectionStart = start + wrapText.length;
-                        textarea.selectionEnd = end + wrapText.length;
-                    } else {
-                        textarea.value = textarea.value.substring(0, start) + wrapText + type + wrapText + textarea.value.substring(end);
-                        textarea.selectionStart = textarea.selectionEnd = start + wrapText.length + type.length;
-                    }
-                    textarea.focus();
-                    
-                    const event = new Event('input', { bubbles: true });
-                    textarea.dispatchEvent(event);
-                }
-
-                document.addEventListener('DOMContentLoaded', function() {
-                    const buttons = document.querySelectorAll('.format-btn');
-                    buttons.forEach(btn => {
-                        btn.addEventListener('click', function(e) {
-                            e.preventDefault();
-                            const wrap = this.getAttribute('data-wrap');
-                            const type = this.getAttribute('data-type');
-                            formatText(wrap, type);
-                        });
-                    });
-                });
-                </script>
-                """)
-                
-                # Markdown editor
-                example_content = """SEO Toolkit: Article #21
-
-Element
-Suggestion
-
-Primary Keyword
-Garment Supply Chain Security
-
-SEO Title
-Guaranteed Delivery: How CdGarment Solves the Global Sourcing Lead-Time Crisis
-
-Meta Description
-Eliminate delivery risks with CdGarment. Our advanced automated facilities and "In-Time" communication guarantee reliable lead times and supply chain security.
-
-Character Count
-156 Characters
+            gr.Markdown("### 📋 粘贴 Gemini 内容")
+            gr.Markdown("支持 JSON 和传统格式")
+            
+            example_content = """# Article #25: Smart Manufacturing: AI & Automation in our Humen Factory
 
-Tags
-reliable garment supplier, apparel lead time management, secure supply chain fashion, automated apparel production, CdGarment delivery guarantee
+The apparel industry is undergoing a digital revolution...
 
 ---
 
-## Guaranteed Delivery: Solving the Lead-Time Crisis Through Automation
-
-In 2026, the greatest risk to a fashion brand isn't design—it's **delivery**. Port delays, labor shortages, and manual production bottlenecks have made traditional "medium" manufacturers unreliable.
-
-CdGarment has solved this by rebuilding the manufacturing process around **Security and Stability**. When we commit to a date, our automated ecosystem ensures we meet it.
-
-### 1. Automation = Predictability
-
-Manual factories suffer from "human variance." If a worker is sick or a bundle is lost, the whole line stops.
-
-- **The Hanging System Advantage**: Our automated system ensures a continuous flow.
-- **Buffer Management**: Our integrated bases allow us to shift production loads.
-
-### 2. Real-Time Supply Chain Visibility
-
-Supply chain security comes from knowing, not guessing.
-
-- **Digital Tracking**: Every order is logged in our system.
-- **Proactive Logistics**: Our logistics team monitors global shipping lanes.
-
-### 3. Financial and Compliance Security
-
-A secure supplier is a compliant supplier. CdGarment maintains all global certifications."""
-
-                gemini_content = gr.Textbox(
-                    label="Gemini Content",
-                    value=example_content,
-                    lines=25,
-                    placeholder="""Paste your Gemini output here...
-
-Example format:
-SEO Title: Your Title Here
-Primary Keyword: Your Keyword
-Meta Description: Your description
-Tags: tag1, tag2, tag3
-
---- (or 📝 Article)
-
-## Article Title
-Your article content here...""",
-                    elem_classes="paste-box"
-                )
-                
-                with gr.Row():
-                    parse_btn = gr.Button("🔍 Parse Content", variant="primary", scale=2)
-                    clear_btn = gr.Button("🗑️ Clear", variant="secondary", scale=1)
+# Machine-Readable Data (For your Script)
+
+```json
+{
+    "post_id": "25",
+    "seo_toolkit": {
+        "primary_keyword": "AI in Garment Manufacturing",
+        "secondary_keywords": [
+            "Smart apparel factory Humen",
+            "Automated fabric cutting",
+            "Digital clothing production 2026"
+        ],
+        "seo_title": "Smart Manufacturing: AI & Automation in our Humen Factory | CdGarment",
+        "description": "Explore the future of fashion...",
+        "character_count": 159,
+        "tags": ["AI garment manufacturing", "smart factory", "automated apparel production", "digital fashion", "Humen factory"]
+    }
+}
+```"""
+            
+            content_input = gr.Textbox(
+                label="粘贴你的 Gemini 输出",
+                value=example_content,
+                lines=15,
+                placeholder="在此粘贴完整的 Gemini 输出（JSON 或传统格式）"
+            )
+            
+            parse_btn = gr.Button("🔍 解析内容", variant="primary")
+            parse_msg = gr.Markdown()
+            
+            preview_box = gr.Markdown()
         
         with gr.Column(scale=1):
-            # Image upload section
-            with gr.Group():
-                gr.Markdown("### 🖼️ Image & Upload")
-                image_input = gr.File(
-                    label="Upload Image",
-                    file_types=["image"],
-                    type="filepath"
-                )
-                image_preview = gr.Image(
-                    label="Image Preview",
-                    height=200,
-                    visible=False
-                )
-                image_description = gr.Textbox(
-                    label="Image Description",
-                    placeholder="Auto-generated description will appear here",
-                    lines=3
-                )
+            gr.Markdown("### 🖼️ 图片和上传")
             
-            # WordPress settings section
-            with gr.Group():
-                gr.Markdown("### ⚙️ WordPress Settings")
-                
-                # Load environment variables
-                env_config = load_environment_variables()
-                
-                wp_url = gr.Textbox(
-                    label="WordPress URL",
-                    value=env_config['url'],
-                    placeholder="https://cdgarment.com"
-                )
-                wp_username = gr.Textbox(
-                    label="Username",
-                    value=env_config['username'],
-                    placeholder="admin"
-                )
-                wp_password = gr.Textbox(
-                    label="Application Password",
-                    value=env_config['password'],
-                    type="password",
-                    placeholder="••••••••"
-                )
-                post_status = gr.Radio(
-                    label="Post Status",
-                    choices=["draft", "publish"],
-                    value=env_config['status']
-                )
-                
-                save_config_btn = gr.Button("💾 Save Configuration", variant="secondary")
-    
-    # Parse status and preview
-    parse_status = gr.Textbox(label="Parse Status", visible=False)
-    
-    with gr.Row(visible=False) as preview_row:
-        with gr.Column():
-            seo_title = gr.Textbox(label="SEO Title")
-            url_slug = gr.Textbox(label="URL Slug")
-        
-        with gr.Column():
-            meta_description = gr.Textbox(label="Meta Description")
-            primary_keyword = gr.Textbox(label="Primary Keyword")
-        
-        with gr.Column():
-            tags = gr.Textbox(label="Tags")
-            content_length = gr.Textbox(label="Content Length")
-    
-    # Publish section
-    gr.Markdown("---")
-    gr.Markdown("### 🚀 Ready to Publish")
-    
-    # Status indicators
-    with gr.Row():
-        content_status = gr.HTML("<span style='padding: 5px 10px; border-radius: 5px; font-weight: bold; margin: 2px; background-color: #fee2e2; color: #991b1b;'>❌ Content</span>")
-        image_status = gr.HTML("<span style='padding: 5px 10px; border-radius: 5px; font-weight: bold; margin: 2px; background-color: #fef3c7; color: #92400e;'>⚠️ Image Optional</span>")
-        wp_status = gr.HTML("<span style='padding: 5px 10px; border-radius: 5px; font-weight: bold; margin: 2px; background-color: #fee2e2; color: #991b1b;'>❌ WordPress</span>")
-        status_indicator = gr.HTML("<span style='padding: 5px 10px; border-radius: 5px; font-weight: bold; margin: 2px; background-color: #e5e7eb;'>📊 Status: DRAFT</span>")
-    
-    publish_btn = gr.Button("🚀 PUSH TO WORDPRESS", variant="primary", size="lg")
-    
-    # Publish results
-    publish_status = gr.Textbox(label="Publish Status", lines=5)
-    download_file = gr.File(label="Download Post Data", visible=False)
-    
-    # Footer
-    gr.Markdown("---")
-    gr.Markdown(f"CdGarment WordPress Publisher • {datetime.now().year}")
-    
-    # ========== EVENT HANDLERS ==========
-    
-    def update_status_indicators(has_content, has_image, has_wp_config, post_status_value):
-        """Update status indicators"""
-        content_color = "#d1fae5" if has_content else "#fee2e2"
-        content_text = "✅ Content" if has_content else "❌ Content"
-        content_html = f"<span style='padding: 5px 10px; border-radius: 5px; font-weight: bold; margin: 2px; background-color: {content_color}; color: #065f46;'>{content_text}</span>"
-        
-        image_color = "#d1fae5" if has_image else "#fef3c7"
-        image_text = "✅ Image" if has_image else "⚠️ Image Optional"
-        image_html = f"<span style='padding: 5px 10px; border-radius: 5px; font-weight: bold; margin: 2px; background-color: {image_color}; color: #92400e;'>{image_text}</span>"
-        
-        wp_color = "#d1fae5" if has_wp_config else "#fee2e2"
-        wp_text = "✅ WordPress" if has_wp_config else "❌ WordPress"
-        wp_html = f"<span style='padding: 5px 10px; border-radius: 5px; font-weight: bold; margin: 2px; background-color: {wp_color}; color: #065f46;'>{wp_text}</span>"
-        
-        status_html = f"<span style='padding: 5px 10px; border-radius: 5px; font-weight: bold; margin: 2px; background-color: #e5e7eb;'>📊 Status: {post_status_value.upper()}</span>"
-        
-        return content_html, image_html, wp_html, status_html
-    
-    def check_wp_config(wp_url_val, wp_username_val, wp_password_val):
-        """Check if WordPress config is complete"""
-        return all([wp_url_val, wp_username_val, wp_password_val])
-    
-    # Image upload handler
-    def update_image_preview(image_file):
-        if image_file:
-            return gr.update(visible=True, value=image_file), True
-        return gr.update(visible=False), False
-    
-    # Parse button handler
-    def on_parse_click(gemini_content_val, post_status_val):
-        if not gemini_content_val or not gemini_content_val.strip():
-            return (
-                gr.update(value="Please paste content first", visible=True),
-                gr.update(visible=False),
-                "", "", "", "", "",
-                None,
-                *update_status_indicators(False, False, False, post_status_val)
+            image_upload = gr.File(
+                label="上传特色图片",
+                file_types=["image"],
+                type="filepath"
             )
-        
-        result, parsed = parse_content(gemini_content_val, post_status_val)
-        
-        if isinstance(result, tuple):
-            parse_status_val, seo_title_val, url_slug_val, meta_desc_val, keyword_val, tags_val, content_len_val = result[:7]
             
-            # Debug output
-            print(f"DEBUG - Parsed tags: {parsed.get('tags', [])}")
-            print(f"DEBUG - Tags display: {tags_val}")
+            img_desc_input = gr.Textbox(
+                label="图片描述",
+                placeholder="输入图片的 alt 文本和描述",
+                value=""
+            )
+            
+            image_preview = gr.Image(label="图片预览", height=200)
+            
+            gr.Markdown("---")
+            gr.Markdown("### ⚙️ WordPress 设置")
             
-            return (
-                gr.update(value=parse_status_val, visible=True),
-                gr.update(visible=True),
-                seo_title_val,
-                url_slug_val,
-                meta_desc_val,
-                keyword_val,
-                tags_val if tags_val else "No tags extracted",
-                content_len_val,
-                parsed,
-                *update_status_indicators(True, False, False, post_status_val)
+            lock_toggle = gr.Checkbox(label="🔒 锁定配置", value=False)
+            
+            wp_url = gr.Textbox(
+                label="WordPress URL",
+                value=wp_config.get('url', 'https://cdgarment.com')
             )
-        else:
-            return (
-                gr.update(value=result, visible=True),
-                gr.update(visible=False),
-                "", "", "", "", "",
-                None,
-                *update_status_indicators(False, False, False, post_status_val)
+            
+            wp_username = gr.Textbox(
+                label="用户名",
+                value=wp_config.get('username', ''),
+                placeholder="admin"
             )
-    
-    # Save config handler
-    def on_save_config(wp_url_val, wp_username_val, wp_password_val, post_status_val):
-        if all([wp_url_val, wp_username_val, wp_password_val]):
-            wp_config = {
-                'url': wp_url_val.rstrip('/'),
-                'username': wp_username_val,
-                'password': wp_password_val,
-                'status': post_status_val
-            }
-            return (
-                gr.update(value="✅ Configuration saved!"),
-                wp_config,
-                *update_status_indicators(False, False, True, post_status_val)
+            
+            wp_password = gr.Textbox(
+                label="应用密码",
+                value=wp_config.get('password', ''),
+                type="password",
+                placeholder="••••••••"
             )
-        return (
-            gr.update(value="⚠️ Please fill all WordPress configuration fields"),
-            None,
-            *update_status_indicators(False, False, False, post_status_val)
-        )
-    
-    # Publish button handler
-    def on_publish_click(parsed_data, wp_config, image_file, post_status_val, wp_url_val, wp_username_val, wp_password_val):
-        if not parsed_data:
-            yield (
-                gr.update(value="❌ No parsed content available. Please parse content first"),
-                gr.update(visible=False),
-                *update_status_indicators(False, bool(image_file), bool(wp_config), post_status_val)
+            
+            status_select = gr.Radio(
+                choices=["draft", "publish"],
+                value=post_status,
+                label="文章状态"
             )
-            return
-        
-        wp_config_to_use = wp_config or {
-            'url': wp_url_val.rstrip('/') if wp_url_val else '',
-            'username': wp_username_val,
-            'password': wp_password_val,
-            'status': post_status_val
-        }
-        
-        # Update status during processing
-        yield (
-            gr.update(value="🖼️ Uploading image to WordPress..."),
-            gr.update(visible=False),
-            *update_status_indicators(True, bool(image_file), True, post_status_val)
-        )
-        
-        # Call the publish function
-        for status_msg, download_file_val in publish_post(parsed_data, wp_config_to_use, image_file, post_status_val, wp_url_val, wp_username_val, wp_password_val):
-            if download_file_val:
-                yield (
-                    gr.update(value=status_msg),
-                    gr.update(value=download_file_val, visible=True),
-                    *update_status_indicators(True, bool(image_file), True, post_status_val)
-                )
-            else:
-                yield (
-                    gr.update(value=status_msg),
-                    gr.update(visible=False),
-                    *update_status_indicators(True, bool(image_file), True, post_status_val)
-                )
-            break
-    
-    # Update status indicators when inputs change
-    def update_indicators_on_change(gemini_content_val, image_file, wp_url_val, wp_username_val, wp_password_val, post_status_val, parsed_data):
-        has_content = bool(parsed_data) or bool(gemini_content_val and gemini_content_val.strip())
-        has_image = bool(image_file)
-        has_wp = check_wp_config(wp_url_val, wp_username_val, wp_password_val)
-        return update_status_indicators(has_content, has_image, has_wp, post_status_val)
+            
+            save_btn = gr.Button("💾 保存配置")
+            save_msg = gr.Markdown()
     
-    # ========== BIND EVENT HANDLERS ==========
+    gr.Markdown("---")
+    gr.Markdown("### 🚀 准备发布")
     
-    # Image upload
-    image_input.change(
-        update_image_preview,
-        inputs=[image_input],
-        outputs=[image_preview, image_status]
+    with gr.Row():
+        content_status = gr.Markdown("❌ 内容")
+        image_status = gr.Markdown("⚠️ 图片可选")
+        wp_status = gr.Markdown("❓ WordPress")
+        status_display = gr.Markdown(f"📊 状态: {post_status.upper()}")
+    
+    publish_btn = gr.Button("🚀 推送到 WordPress", variant="primary", size="lg")
+    publish_progress = gr.Markdown()
+    publish_result = gr.Markdown()
+    
+    json_output = gr.JSON(label="文章数据", visible=False)
+    download_btn = gr.DownloadButton(
+        "📥 下载文章数据",
+        value="",
+        visible=False
     )
     
-    # Parse button
+    # ========== 事件绑定 ==========
     parse_btn.click(
-        on_parse_click,
-        inputs=[gemini_content, post_status],
-        outputs=[
-            parse_status,
-            preview_row,
-            seo_title,
-            url_slug,
-            meta_description,
-            primary_keyword,
-            tags,
-            content_length,
-            parsed_data_state,
-            content_status,
-            image_status,
-            wp_status,
-            status_indicator
-        ]
+        parse_content_callback,
+        inputs=[content_input],
+        outputs=[parse_msg, preview_box]
     )
     
-    # Save config button
-    save_config_btn.click(
-        on_save_config,
-        inputs=[wp_url, wp_username, wp_password, post_status],
-        outputs=[parse_status, wp_config_state, content_status, image_status, wp_status, status_indicator]
+    image_upload.change(
+        update_image_description_callback,
+        inputs=[image_upload],
+        outputs=[img_desc_input, img_desc_input]
     )
     
-    # Publish button
-    publish_btn.click(
-        on_publish_click,
-        inputs=[parsed_data_state, wp_config_state, image_input, post_status, wp_url, wp_username, wp_password],
-        outputs=[publish_status, download_file, content_status, image_status, wp_status, status_indicator]
+    lock_toggle.change(
+        toggle_lock_callback,
+        inputs=[lock_toggle],
+        outputs=[wp_url, wp_username, wp_password]
     )
     
-    # Clear button
-    clear_btn.click(
-        clear_all,
-        outputs=[
-            gemini_content,
-            image_input,
-            wp_url,
-            wp_username,
-            wp_password,
-            post_status,
-            parse_status,
-            seo_title,
-            url_slug,
-            meta_description,
-            primary_keyword,
-            tags,
-            content_length,
-            publish_status,
-            download_file
-        ]
+    save_btn.click(
+        save_config_callback,
+        inputs=[wp_url, wp_username, wp_password, status_select],
+        outputs=[save_msg]
     )
     
-    # Update status indicators on input changes
-    for input_component in [gemini_content, image_input, wp_url, wp_username, wp_password, post_status]:
-        input_component.change(
-            update_indicators_on_change,
-            inputs=[gemini_content, image_input, wp_url, wp_username, wp_password, post_status, parsed_data_state],
-            outputs=[content_status, image_status, wp_status, status_indicator]
-        )
+    publish_btn.click(
+        publish_callback,
+        inputs=[],
+        outputs=[publish_progress, publish_result, json_output]
+    ).then(
+        lambda: (gr.update(visible=True), gr.update(visible=True)),
+        outputs=[json_output, download_btn]
+    )
+    
+    gr.Markdown(f"---\nCdGarment WordPress Publisher • {datetime.now().year} • 格式: {parse_method.upper()}")
 
-# ========== LAUNCH THE APP ==========
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7862,
-        share=False,
-        debug=True,
-        ssr_mode=False,
-        css="""
-        .gradio-container { max-width: 1400px !important; }
-        .paste-box textarea { 
-            font-family: 'Courier New', monospace !important;
-            font-size: 14px !important;
-            line-height: 1.5 !important;
-        }
-        """
-    )
+    demo.launch(server_name="0.0.0.0", server_port=7860)