htmlgen-mcp 0.2.5__py3-none-any.whl → 0.3.1__py3-none-any.whl

htmlgen_mcp/agents/smart_web_agent.py

@@ -32,6 +32,7 @@ else:
 
 # 导入工具函数
 from .web_tools import *
+from .quick_generator import QuickSiteGenerator
 
 
 class SmartWebAgent:
@@ -440,6 +441,24 @@ class SmartWebAgent:
         self, user_input: str, max_retries: int = 3
     ) -> Optional[dict]:
         """获取执行计划，带重试机制"""
+        
+        # 判断是否使用快速模式
+        if self._should_use_quick_mode(user_input):
+            print("⚡ 启用快速模式：生成单页面网站")
+            site_type = self._extract_site_type(user_input)
+            project_name = self._extract_project_name(user_input, site_type)
+            
+            plan = QuickSiteGenerator.create_single_page_plan(
+                project_name=project_name,
+                site_type=site_type,
+                description=user_input
+            )
+            
+            # 进一步优化速度
+            plan = QuickSiteGenerator.optimize_for_speed(plan)
+            self.current_plan = plan
+            return plan
+        
         # 若无客户端，直接走离线计划
         if self.client is None:
             plan = self._build_fallback_plan(user_input)
@@ -480,12 +499,16 @@ class SmartWebAgent:
         planning_prompt_template = """你是一个专业的网页设计与前端开发专家，精通现代Web技术、设计系统和用户体验。
 请分析用户需求并制定执行计划，创建高质量、现代化、专业的网站。
 
-🎯 核心设计理念：
+🎨 核心设计理念：
 • 现代美观：采用当前流行的设计趋势（如新拟态、毛玻璃、渐变、3D效果）
 • 用户体验：移动优先、快速加载、流畅交互、无障碍访问
 • 视觉层次：合理的留白、清晰的信息架构、引导性的视觉流
 • 品牌一致：统一的设计语言、配色方案、字体系统
-• 多页面协同：根据需求规划多页面站点结构，若用户未指定，也默认覆盖首页、关于、服务/产品、联系等关键页面，所有页面需保持导航一致性与互链
+
+📊 **页面生成策略**：
+• 快速模式（默认）：仅生成一个完整的首页，包含所有核心内容部分
+• 多页面模式：仅当用户明确提到"多页面"、"多个页面"、"分别创建"或列出具体页面名称时启用
+• 判断规则：如果用户只说"咖啡店网站"、"企业官网"等，默认使用单页面滚动式设计
 
 用户需求：<<USER_INPUT>>
 工作目录：<<PROJECT_DIR>>
@@ -531,7 +554,9 @@ class SmartWebAgent:
 
 📋 执行规范：
 
-1. **项目结构**（严格遵循）：
+⚠️ **重要提示**：优先速度！除非用户明确要求，否则只生成一个页面。
+
+1. **项目结构**（简化流程）：
    - 第1步：create_project_structure - 创建完整目录结构
    - 第2步：create_css_file - 创建样式文件(assets/css/style.css)
    - 第3步：create_js_file - 创建脚本文件(assets/js/main.js)
@@ -539,9 +564,9 @@ class SmartWebAgent:
    - 第5步：add_bootstrap - 添加框架支持
    - 第6步：create_responsive_navbar - 创建导航组件
    - 🎨 **第7步：inject_images - 智能图片注入（必须包含！）**
-   - 第8-9步：验证和检查
-   - 第10步：open_in_browser - 预览效果
-   - 如需求涉及多页面，请为每个页面单独安排 create_html_file → create_responsive_navbar（带跨页面链接）→ inject_images / 验证等步骤，确保导航项指向正确的 html 文件
+   - 第8步：验证HTML（可选）
+   - 第9步：open_in_browser - 预览效果
+   - 仅在用户明确要求多页面时，才为每个页面单独安排步骤
 
 ⚠️ **重要：第7步图片注入是必需的！**
 使用 inject_images 为网站添加美观的AI生成图片：
@@ -572,12 +597,12 @@ class SmartWebAgent:
    - 团队介绍：人员卡片+职位信息
    - 合作伙伴：Logo墙+滚动动画
 
-   🍔 **餐厅网站 / 咖啡店**（强烈建议使用专用工具）：
-   - Hero区：餐厅氛围图+预约/到店CTA
-   - 菜单展示：使用 create_menu_page（分类Tab+价格标签+推荐标记）
-   - 关于我们：使用 create_about_page（品牌故事/理念/团队）
-   - 联系我们：使用 create_contact_page（营业时间+地址+表单+地图占位）
-   - 图片：为菜单/关于/联系分别注入咖啡/甜点/门店环境类主题
+   🍔 **餐厅网站 / 咖啡店**：
+   - 默认生成单页面版本：Hero区+菜单展示+营业信息+联系方式（全部在首页）
+   - 多页面版本（仅在用户要求时）：
+     * 菜单页：使用 create_menu_page
+     * 关于页：使用 create_about_page  
+     * 联系页：使用 create_contact_page
 
    🛍️ **电商着陆页**：
    - Hero区：产品大图+限时优惠倒计时
@@ -753,6 +778,87 @@ class SmartWebAgent:
         s = "".join(slug).strip("-")
         return s[:32] or default
 
+    def _should_use_quick_mode(self, user_input: str) -> bool:
+        """判断是否应该使用快速模式
+        
+        快速模式条件：
+        1. 用户未明确要求多页面
+        2. 用户未列出具体的页面名称
+        3. 用户未要求复杂的功能
+        """
+        lower_input = user_input.lower()
+        
+        # 检查是否明确要求多页面
+        multi_page_keywords = [
+            "多页面", "多个页面", "分别创建", "分别生成",
+            "菜单页", "关于页", "联系页", "产品页",
+            "multiple pages", "separate pages", "menu page", "about page"
+        ]
+        
+        for keyword in multi_page_keywords:
+            if keyword in lower_input:
+                return False
+        
+        # 检查是否有复杂功能要求
+        complex_keywords = [
+            "复杂", "详细", "完整", "全面", "多功能",
+            "complex", "detailed", "complete", "comprehensive"
+        ]
+        
+        for keyword in complex_keywords:
+            if keyword in lower_input:
+                return False
+        
+        # 默认使用快速模式
+        return True
+    
+    def _extract_site_type(self, user_input: str) -> str:
+        """从用户输入中提取网站类型"""
+        lower_input = user_input.lower()
+        
+        type_keywords = {
+            "咖啡店": ["咖啡", "coffee", "cafe"],
+            "餐厅": ["餐厅", "餐饮", "美食", "restaurant", "dining"],
+            "企业": ["企业", "公司", "商业", "company", "corporate", "business"],
+            "作品集": ["作品集", "个人", "portfolio", "personal"],
+            "电商": ["电商", "商店", "购物", "shop", "store", "ecommerce"],
+            "博客": ["博客", "文章", "内容", "blog", "article"]
+        }
+        
+        for site_type, keywords in type_keywords.items():
+            for keyword in keywords:
+                if keyword in lower_input:
+                    return site_type
+        
+        return "通用网站"
+    
+    def _extract_project_name(self, user_input: str, site_type: str) -> str:
+        """从用户输入提取或生成项目名称"""
+        # 尝试提取引号中的名称
+        import re
+        match = re.search(r'[“"「]([\w\s\u4e00-\u9fa5]+)[”"」]', user_input)
+        if match:
+            name = match.group(1)
+            # 转换为英文项目名
+            name = name.replace(' ', '-').lower()
+            # 如果是中文，使用类型作为前缀
+            if any(ord(c) > 127 for c in name):
+                return f"{site_type.replace('网站', '').lower()}-site"
+            return name
+        
+        # 默认项目名
+        type_to_name = {
+            "咖啡店": "coffee-shop",
+            "餐厅": "restaurant",
+            "企业": "corporate",
+            "作品集": "portfolio",
+            "电商": "ecommerce",
+            "博客": "blog",
+            "通用网站": "modern-site"
+        }
+        
+        return type_to_name.get(site_type, "website")
+    
     def _build_fallback_plan(self, user_input: str) -> dict:
         """在无网络/无密钥时的本地执行计划：生成一个现代化的基础站点骨架"""
         project_name = self._slugify(user_input)

htmlgen_mcp/agents/web_tools/__init__.py

@@ -4,12 +4,22 @@ from __future__ import annotations
 from .bootstrap import add_bootstrap
 from .browser import open_in_browser, start_live_server
 from .css import create_css_file
-from .html_templates import (
-    create_about_page,
-    create_contact_page,
-    create_html_file,
-    create_menu_page,
-)
+# 使用改进版的页面生成工具
+try:
+    from .html_templates_improved import (
+        create_html_file,
+        create_menu_page,
+        create_about_page,
+        create_contact_page,
+    )
+except ImportError:
+    # 如果改进版不存在，回退到原版
+    from .html_templates import (
+        create_html_file,
+        create_menu_page,
+        create_about_page,
+        create_contact_page,
+    )
 from .images import fetch_generated_images, inject_images
 from .js import create_js_file
 from .navigation import create_responsive_navbar