htmlgen-mcp 0.2.2__tar.gz → 0.3.1__tar.gz

{htmlgen_mcp-0.2.2/htmlgen_mcp.egg-info → htmlgen_mcp-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: htmlgen-mcp
-Version: 0.2.2
+Version: 0.3.1
 Summary: AI-powered HTML website generator with auto-upload functionality via Model Context Protocol
 Author-email: HTML Generator Team <contact@htmlgen-mcp.com>
 License: MIT

{htmlgen_mcp-0.2.2 → htmlgen_mcp-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "htmlgen-mcp"
-version = "0.2.2"
+version = "0.3.1"
 description = "AI-powered HTML website generator with auto-upload functionality via Model Context Protocol"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -48,7 +48,7 @@ dev = [
 ]
 
 [project.scripts]
-htmlgen-mcp = "MCP.web_agent_server:main"
+htmlgen-mcp = "htmlgen_mcp.web_agent_server:main"
 
 [project.urls]
 Homepage = "https://github.com/your-username/htmlgen-mcp"
@@ -57,11 +57,11 @@ Issues = "https://github.com/your-username/htmlgen-mcp/issues"
 Documentation = "https://github.com/your-username/htmlgen-mcp/blob/main/README.md"
 
 [tool.setuptools.packages.find]
-where = ["."]
-include = ["MCP*", "agents*"]
+where = ["src"]
+include = ["htmlgen_mcp*", "agents*"]
 
 [tool.setuptools.package-data]
-MCP = ["*.json", "*.yaml", "*.yml"]
+htmlgen_mcp = ["*.json", "*.yaml", "*.yml"]
 agents = ["*.json", "*.yaml", "*.yml"]
 
 [dependency-groups]

htmlgen_mcp-0.3.1/src/htmlgen_mcp/__init__.py

@@ -0,0 +1,7 @@
+"""
+MCP web agent server module.
+"""
+from .web_agent_server import main
+
+__version__ = "0.3.1"
+__all__ = ["main"]

htmlgen_mcp-0.3.1/src/htmlgen_mcp/agents/ai_content_generator.py

@@ -0,0 +1,328 @@
+"""AI 内容生成辅助模块 - 为页面生成个性化内容"""
+from __future__ import annotations
+
+import json
+from typing import Dict, Any, Optional
+
+
+class AIContentGenerator:
+    """AI 内容生成器 - 为各种页面生成个性化内容"""
+    
+    def __init__(self, agent_instance):
+        """初始化生成器
+        
+        Args:
+            agent_instance: SmartWebAgent 实例
+        """
+        self.agent = agent_instance
+    
+    def generate_menu_content(
+        self, 
+        project_name: str, 
+        context: Dict[str, Any]
+    ) -> Dict[str, Any]:
+        """为菜单页面生成个性化内容"""
+        
+        prompt = f"""你是一个专业的网页内容创作者。
+为"{project_name}"生成一个详细的菜单内容。
+
+项目信息：
+- 名称：{project_name}
+- 描述：{context.get('description', '')}
+- 特色：{context.get('features', '')}
+- 位置：{context.get('location', '')}
+- 目标客户：{context.get('target_audience', '')}
+
+请生成一个完整的菜单，包含：
+1. 2-4个类别（如：热饮、冷饮、轻食、甜点）
+2. 每个类别3-5个项目
+3. 每个项目包含：
+   - name: 产品名称（有创意且符合品牌）
+   - price: 价格（符合定位，使用"¥"符号）
+   - description: 描述（简短吸引人，20字以内）
+   - image_topic: 图片主题描述（英文，用于生成图片）
+
+返回 JSON 格式：
+{{
+  "categories": [
+    {{
+      "name": "类别名称",
+      "items": [
+        {{
+          "name": "产品名称",
+          "price": "¥ XX",
+          "description": "产品描述",
+          "image_topic": "image description"
+        }}
+      ]
+    }}
+  ],
+  "description": "菜单页面的副标题描述"
+}}
+
+重要：
+- 内容要符合项目的定位和风格
+- 价格要合理且有梯度
+- 描述要诱人且真实
+- 直接返回 JSON，不要其他内容
+"""
+        
+        try:
+            response = self.agent.client.chat.completions.create(
+                model=self.agent.model,
+                messages=[
+                    {"role": "system", "content": "你是一个专业的内容创作者，擅长为各类商业项目创作个性化内容。"},
+                    {"role": "user", "content": prompt}
+                ],
+                temperature=0.7,
+                max_tokens=2000
+            )
+            
+            content = response.choices[0].message.content.strip()
+            # 清理可能的 markdown 代码块标记
+            if content.startswith("```"):
+                lines = content.split('\n')
+                content = '\n'.join(lines[1:-1] if lines[-1] == "```" else lines[1:])
+            
+            return json.loads(content)
+            
+        except Exception as e:
+            print(f"AI 生成菜单内容失败：{e}")
+            # 返回空内容，让页面使用默认值
+            return {}
+    
+    def generate_about_content(
+        self, 
+        project_name: str, 
+        context: Dict[str, Any]
+    ) -> Dict[str, Any]:
+        """为关于页面生成个性化内容"""
+        
+        prompt = f"""为"{project_name}"生成"关于我们"页面的内容。
+
+项目信息：
+- 名称：{project_name}
+- 描述：{context.get('description', '')}
+- 使命：{context.get('mission', '')}
+- 特色：{context.get('features', '')}
+- 创始年份：{context.get('year_founded', '2020')}
+
+请生成以下内容（JSON格式）：
+{{
+  "headline": "页面主标题",
+  "subtitle": "副标题（一句话介绍）",
+  "story": {{
+    "title": "品牌故事标题",
+    "content": "品牌故事内容（100-150字）",
+    "image_topic": "story image description"
+  }},
+  "values": ["价值观1", "价值观2", "价值观3", "价值观4"],
+  "team": [
+    {{
+      "name": "团队成员姓名",
+      "role": "职位",
+      "description": "简介（20字以内）",
+      "image_topic": "team member photo description"
+    }}
+  ],
+  "achievements": [
+    {{"value": "数值", "label": "标签"}},
+    {{"value": "数值", "label": "标签"}}
+  ],
+  "cta_title": "行动号召标题",
+  "cta_text": "行动号召文字"
+}}
+
+要求：
+- 内容真实可信，避免夸大
+- 团队成员生成2-3个核心岗位即可
+- 成就数据要合理
+- 符合品牌调性
+"""
+        
+        try:
+            response = self.agent.client.chat.completions.create(
+                model=self.agent.model,
+                messages=[
+                    {"role": "user", "content": prompt}
+                ],
+                temperature=0.7,
+                max_tokens=1500
+            )
+            
+            content = response.choices[0].message.content.strip()
+            if content.startswith("```"):
+                lines = content.split('\n')
+                content = '\n'.join(lines[1:-1] if lines[-1] == "```" else lines[1:])
+            
+            return json.loads(content)
+            
+        except Exception as e:
+            print(f"AI 生成关于内容失败：{e}")
+            return {}
+    
+    def generate_contact_content(
+        self, 
+        project_name: str, 
+        context: Dict[str, Any]
+    ) -> Dict[str, Any]:
+        """为联系页面生成个性化文案"""
+        
+        prompt = f"""为"{project_name}"生成联系页面的文案。
+
+项目信息：
+- 名称：{project_name}
+- 类型：{context.get('type', '咖啡店')}
+- 特色：{context.get('features', '')}
+
+生成以下文案（JSON格式）：
+{{
+  "headline": "联系页面标题",
+  "subtitle": "副标题（欢迎语）",
+  "form_title": "表单标题",
+  "form_text": "表单说明文字",
+  "response_time": "响应时间承诺"
+}}
+
+要求文案：
+- 友好亲切
+- 专业可信
+- 简洁明了
+"""
+        
+        try:
+            response = self.agent.client.chat.completions.create(
+                model=self.agent.model,
+                messages=[
+                    {"role": "user", "content": prompt}
+                ],
+                temperature=0.6,
+                max_tokens=500
+            )
+            
+            content = response.choices[0].message.content.strip()
+            if content.startswith("```"):
+                lines = content.split('\n')
+                content = '\n'.join(lines[1:-1] if lines[-1] == "```" else lines[1:])
+            
+            return json.loads(content)
+            
+        except Exception as e:
+            print(f"AI 生成联系内容失败：{e}")
+            return {}
+    
+    def enhance_project_context(
+        self, 
+        project_name: str, 
+        description: str
+    ) -> Dict[str, Any]:
+        """增强项目上下文信息，补充缺失的细节"""
+        
+        prompt = f"""分析项目"{project_name}"并补充必要的商业信息。
+
+项目描述：{description}
+
+请推断并生成以下信息（JSON格式）：
+{{
+  "type": "项目类型（如：咖啡店、餐厅、科技公司等）",
+  "target_audience": "目标客户描述",
+  "features": "核心特色（3-5个关键词）",
+  "mission": "使命宣言（一句话）",
+  "address": "合理的示例地址",
+  "phone": "示例电话（使用 010-XXXX-XXXX 格式）",
+  "email": "示例邮箱",
+  "business_hours": "营业时间"
+}}
+"""
+        
+        try:
+            response = self.agent.client.chat.completions.create(
+                model=self.agent.model,
+                messages=[
+                    {"role": "user", "content": prompt}
+                ],
+                temperature=0.5,
+                max_tokens=500
+            )
+            
+            content = response.choices[0].message.content.strip()
+            if content.startswith("```"):
+                lines = content.split('\n')
+                content = '\n'.join(lines[1:-1] if lines[-1] == "```" else lines[1:])
+            
+            return json.loads(content)
+            
+        except Exception as e:
+            print(f"增强上下文失败：{e}")
+            return {
+                "type": "商业",
+                "target_audience": "大众客户",
+                "features": "品质、服务、创新",
+                "mission": f"为客户提供优质的产品与服务",
+                "address": "北京市朝阳区示例路123号",
+                "phone": "010-8888-8888",
+                "email": f"info@{project_name.lower().replace(' ', '')}.com",
+                "business_hours": "周一至周五 9:00-18:00"
+            }
+
+
+def integrate_ai_generation(agent_instance, tool_name: str, params: Dict[str, Any]) -> Dict[str, Any]:
+    """集成 AI 生成到工具调用中
+    
+    Args:
+        agent_instance: SmartWebAgent 实例
+        tool_name: 工具名称
+        params: 工具参数
+    
+    Returns:
+        增强后的参数，包含 AI 生成的内容
+    """
+    
+    # 只处理特定的页面生成工具
+    if tool_name not in ["create_menu_page", "create_about_page", "create_contact_page"]:
+        return params
+    
+    # 提取项目信息
+    project_name = params.get('project_name', '')
+    file_path = params.get('file_path', '')
+    
+    # 获取或创建上下文
+    context = params.get('context', {})
+    if not context:
+        # 尝试从 agent 的执行历史中提取上下文
+        context = {
+            'description': agent_instance.latest_user_request or '',
+            'project_directory': agent_instance.project_directory
+        }
+    
+    # 创建 AI 内容生成器
+    generator = AIContentGenerator(agent_instance)
+    
+    # 根据工具类型生成相应内容
+    ai_content = {}
+    if tool_name == "create_menu_page":
+        ai_content = generator.generate_menu_content(project_name, context)
+    elif tool_name == "create_about_page":
+        ai_content = generator.generate_about_content(project_name, context)
+    elif tool_name == "create_contact_page":
+        ai_content = generator.generate_contact_content(project_name, context)
+    
+    # 如果需要，增强上下文信息
+    if not context.get('address') or not context.get('phone'):
+        enhanced_context = generator.enhance_project_context(
+            project_name,
+            context.get('description', '')
+        )
+        context.update(enhanced_context)
+    
+    # 更新参数
+    params['context'] = context
+    params['ai_content'] = ai_content
+    
+    return params
+
+
+__all__ = [
+    'AIContentGenerator',
+    'integrate_ai_generation'
+]

htmlgen_mcp-0.3.1/src/htmlgen_mcp/agents/quick_generator.py

@@ -0,0 +1,270 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""快速单页网站生成器 - 优化生成速度"""
+
+from typing import Dict, Any, Optional, List
+import json
+
+
+class QuickSiteGenerator:
+    """快速生成单页面网站的辅助类"""
+    
+    @staticmethod
+    def create_single_page_plan(
+        project_name: str,
+        site_type: str,
+        description: str
+    ) -> Dict[str, Any]:
+        """创建单页面网站的简化计划
+        
+        Args:
+            project_name: 项目名称
+            site_type: 网站类型（咖啡店、企业、作品集等）
+            description: 用户需求描述
+            
+        Returns:
+            简化的执行计划
+        """
+        
+        # 基础步骤（所有类型通用）
+        base_steps = [
+            {
+                "step": 1,
+                "tool": "create_project_structure",
+                "params": {
+                    "project_name": project_name,
+                    "project_path": "."
+                },
+                "description": "创建项目目录结构",
+                "rationale": "建立规范的项目结构"
+            },
+            {
+                "step": 2,
+                "tool": "create_css_file",
+                "params": {
+                    "file_path": "assets/css/style.css"
+                },
+                "description": "创建样式文件",
+                "rationale": "定义网站视觉风格"
+            },
+            {
+                "step": 3,
+                "tool": "create_js_file",
+                "params": {
+                    "file_path": "assets/js/main.js"
+                },
+                "description": "创建交互脚本",
+                "rationale": "添加动态效果和交互"
+            },
+            {
+                "step": 4,
+                "tool": "create_html_file",
+                "params": {
+                    "file_path": "index.html",
+                    "title": f"{project_name} - 首页"
+                },
+                "description": "创建单页面网站主文件",
+                "rationale": "生成包含所有内容的单页面"
+            },
+            {
+                "step": 5,
+                "tool": "add_bootstrap",
+                "params": {
+                    "project_path": "."
+                },
+                "description": "添加Bootstrap框架",
+                "rationale": "加速响应式开发"
+            },
+            {
+                "step": 6,
+                "tool": "create_responsive_navbar",
+                "params": {
+                    "file_path": "index.html",
+                    "brand_name": project_name,
+                    "nav_items": QuickSiteGenerator._get_nav_items(site_type)
+                },
+                "description": "创建导航栏",
+                "rationale": "页面内导航锚点"
+            },
+            {
+                "step": 7,
+                "tool": "inject_images",
+                "params": {
+                    "file_path": "index.html",
+                    "provider": "pollinations",
+                    "topics": QuickSiteGenerator._get_image_topics(site_type),
+                    "size": "1920x1080",
+                    "save": True
+                },
+                "description": "注入AI生成图片",
+                "rationale": "添加视觉内容"
+            },
+            {
+                "step": 8,
+                "tool": "open_in_browser",
+                "params": {
+                    "file_path": "index.html"
+                },
+                "description": "浏览器预览",
+                "rationale": "查看最终效果"
+            }
+        ]
+        
+        # 根据网站类型定制配色
+        color_scheme = QuickSiteGenerator._get_color_scheme(site_type)
+        
+        return {
+            "task_analysis": f"快速生成{site_type}单页面网站",
+            "project_name": project_name,
+            "site_type": site_type,
+            "design_style": "现代简洁",
+            "color_scheme": color_scheme,
+            "estimated_time": "2-3分钟",
+            "tools_sequence": base_steps
+        }
+    
+    @staticmethod
+    def _get_nav_items(site_type: str) -> List[Dict[str, str]]:
+        """根据网站类型返回导航项（锚点链接）"""
+        
+        nav_templates = {
+            "咖啡店": [
+                {"name": "首页", "link": "#hero"},
+                {"name": "菜单", "link": "#menu"},
+                {"name": "关于", "link": "#about"},
+                {"name": "联系", "link": "#contact"}
+            ],
+            "企业": [
+                {"name": "首页", "link": "#hero"},
+                {"name": "服务", "link": "#services"},
+                {"name": "关于", "link": "#about"},
+                {"name": "联系", "link": "#contact"}
+            ],
+            "作品集": [
+                {"name": "首页", "link": "#hero"},
+                {"name": "作品", "link": "#portfolio"},
+                {"name": "技能", "link": "#skills"},
+                {"name": "联系", "link": "#contact"}
+            ],
+            "餐厅": [
+                {"name": "首页", "link": "#hero"},
+                {"name": "菜单", "link": "#menu"},
+                {"name": "预约", "link": "#booking"},
+                {"name": "位置", "link": "#location"}
+            ]
+        }
+        
+        # 默认导航
+        default = [
+            {"name": "首页", "link": "#hero"},
+            {"name": "介绍", "link": "#about"},
+            {"name": "服务", "link": "#services"},
+            {"name": "联系", "link": "#contact"}
+        ]
+        
+        # 查找最匹配的类型
+        for key in nav_templates:
+            if key in site_type:
+                return nav_templates[key]
+        
+        return default
+    
+    @staticmethod
+    def _get_image_topics(site_type: str) -> List[str]:
+        """根据网站类型返回图片主题"""
+        
+        topics_map = {
+            "咖啡": ["coffee shop interior modern", "latte art", "coffee beans", "cozy cafe"],
+            "餐厅": ["restaurant interior elegant", "gourmet food", "dining table", "chef cooking"],
+            "企业": ["modern office", "business team", "corporate building", "technology workspace"],
+            "作品集": ["creative workspace", "design portfolio", "artistic studio", "digital art"],
+            "电商": ["product showcase", "online shopping", "ecommerce", "shopping cart"],
+            "博客": ["writing desk", "laptop workspace", "books and coffee", "minimal desk setup"]
+        }
+        
+        # 查找匹配的主题
+        for key, topics in topics_map.items():
+            if key in site_type:
+                return topics
+        
+        # 默认主题
+        return ["modern website hero", "business concept", "technology background", "professional workspace"]
+    
+    @staticmethod
+    def _get_color_scheme(site_type: str) -> Dict[str, str]:
+        """根据网站类型返回配色方案"""
+        
+        schemes = {
+            "咖啡": {
+                "primary": "#6F4E37",  # 咖啡棕
+                "secondary": "#C8A882",  # 奶泡色
+                "accent": "#D2691E"  # 焦糖色
+            },
+            "餐厅": {
+                "primary": "#8B0000",  # 深红
+                "secondary": "#FFD700",  # 金色
+                "accent": "#228B22"  # 森林绿
+            },
+            "企业": {
+                "primary": "#003366",  # 企业蓝
+                "secondary": "#F0F0F0",  # 浅灰
+                "accent": "#FF6B35"  # 橙色
+            },
+            "作品集": {
+                "primary": "#2C3E50",  # 深蓝灰
+                "secondary": "#ECF0F1",  # 云白
+                "accent": "#E74C3C"  # 红色
+            },
+            "电商": {
+                "primary": "#FF6B6B",  # 珊瑚红
+                "secondary": "#4ECDC4",  # 青绿
+                "accent": "#FFE66D"  # 黄色
+            }
+        }
+        
+        # 查找匹配的配色
+        for key, scheme in schemes.items():
+            if key in site_type:
+                return scheme
+        
+        # 默认配色（现代通用）
+        return {
+            "primary": "#3B82F6",  # 蓝色
+            "secondary": "#F3F4F6",  # 浅灰
+            "accent": "#10B981"  # 绿色
+        }
+    
+    @staticmethod
+    def optimize_for_speed(plan: Dict[str, Any]) -> Dict[str, Any]:
+        """优化计划以提升生成速度
+        
+        - 移除不必要的验证步骤
+        - 简化图片生成
+        - 减少工具调用次数
+        """
+        optimized_steps = []
+        
+        for step in plan.get("tools_sequence", []):
+            tool = step.get("tool", "")
+            
+            # 跳过验证类工具（可选）
+            if tool in ["validate_html", "check_mobile_friendly"]:
+                continue
+                
+            # 简化图片参数
+            if tool == "inject_images":
+                params = step.get("params", {})
+                # 限制图片数量
+                if "topics" in params and len(params["topics"]) > 3:
+                    params["topics"] = params["topics"][:3]
+                # 使用较小的尺寸
+                if "size" in params:
+                    params["size"] = "1280x720"
+                step["params"] = params
+            
+            optimized_steps.append(step)
+        
+        plan["tools_sequence"] = optimized_steps
+        plan["estimated_time"] = "1-2分钟"
+        
+        return plan