htmlgen-mcp 0.3.7__tar.gz → 0.3.9__tar.gz

{htmlgen_mcp-0.3.7 → htmlgen_mcp-0.3.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: htmlgen-mcp
-Version: 0.3.7
+Version: 0.3.9
 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.3.7 → htmlgen_mcp-0.3.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "htmlgen-mcp"
-version = "0.3.7"
+version = "0.3.9"
 description = "AI-powered HTML website generator with auto-upload functionality via Model Context Protocol"
 readme = "README.md"
 requires-python = ">=3.10"

{htmlgen_mcp-0.3.7 → htmlgen_mcp-0.3.9}/src/htmlgen_mcp/web_agent_server.py

@@ -44,8 +44,8 @@ DEFAULT_PROJECT_ROOT = os.path.abspath(
 )
 # 是否自动生成项目子目录（可通过环境变量控制）
 AUTO_CREATE_PROJECT_DIR = os.environ.get("AUTO_CREATE_PROJECT_DIR", "true").lower() == "true"
-# 默认的文件上传URL（固定值）
-DEFAULT_UPLOAD_URL = "https://www.mcpcn.cc/api/fileUploadAndDownload/uploadMcpFile"
+# 默认的文件上传URL（可通过环境变量 UPLOAD_URL 覆盖）
+DEFAULT_UPLOAD_URL = os.environ.get("UPLOAD_URL", "https://www.mcpcn.cc/api/fileUploadAndDownload/uploadMcpFile")
 DEFAULT_MODEL = os.environ.get("WEB_AGENT_MODEL", "qwen3-coder-plus-2025-09-23")
 DEFAULT_BASE_URL = os.environ.get(
     "OPENAI_BASE_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1"
@@ -82,6 +82,7 @@ _PROGRESS_LOG_BY_ID: dict[str, str] = {}
 _PROGRESS_LOG_BY_JOB: dict[str, str] = {}
 _JOB_REGISTRY: dict[str, Dict[str, Any]] = {}
 _CONTEXT_CACHE_BY_ID: dict[str, Dict[str, Any]] = {}
+_CONTEXT_ID_BY_PLAN: dict[str, str] = {}
 
 
 def _job_state_path(job_id: str) -> Path:
@@ -140,6 +141,18 @@ def _load_job_states() -> None:
 _load_job_states()
 
 
+def _load_job_state_from_disk(job_id: str) -> Optional[Dict[str, Any]]:
+    path = _job_state_path(job_id)
+    if not path.exists():
+        return None
+    try:
+        data = json.loads(path.read_text(encoding="utf-8"))
+        data.setdefault("job_id", job_id)
+        return data
+    except Exception:
+        return None
+
+
 def _context_cache_path(context_id: str) -> Path:
     return CONTEXT_CACHE_DIR / f"{context_id}.json"
 
@@ -453,8 +466,6 @@ async def execute_plan(
     # verbose: bool = False,  # 后台执行时详细日志意义不大
     save_output: bool = False,
     progress_log: Optional[str] = None,
-    auto_upload: bool = False,
-    upload_url: Optional[str] = None,
 ) -> Dict[str, Any]:
     """执行网页构建计划，始终以后台模式运行。
 
@@ -479,21 +490,11 @@ async def execute_plan(
         如果都未指定：不记录进度日志
         路径可以是绝对路径或相对于 project_root 的相对路径
 
-    - auto_upload: 是否在构建完成后自动上传到MCP服务器
-        True: 构建完成后自动打包为ZIP并上传，返回访问URL
-        False: 仅构建，不上传
-        默认为 False
-
-    - upload_url: 文件上传API地址（可选）
-        默认为 "https://www.mcpcn.cc/api/fileUploadAndDownload/uploadMcpFile"
-        只在 auto_upload=True 时生效，一般不需要修改
-
     执行流程：
     - 任务始终在后台异步执行，立即返回 job_id 和 progress_log 路径
     - 使用 get_progress(job_id=..., limit=...) 可以实时查询执行状态和进度
     - 支持通过进度日志文件追踪详细的执行步骤和结果
-    - 执行完成后可以获取完整的执行报告和生成的文件列表
-    - 🎯 新增：auto_upload=True时，构建完成(100%)后自动上传并返回访问URL
+    - 执行完成后可以获取完整的执行报告、生成文件列表，以及自动上传结果
     """
     try:
         # 如果没有提供project_root，尝试从缓存中获取
@@ -638,38 +639,26 @@ async def execute_plan(
 
         _persist_job_state(job_id)
 
-        # 使用默认URL如果没有提供
-        effective_upload_url = upload_url or DEFAULT_UPLOAD_URL
-        
         asyncio.create_task(
             _run_execution_job(
                 job_id,
                 agent,
                 plan_dict,
                 progress_log_path=progress_log_path,
-                auto_upload=auto_upload,
-                upload_url=effective_upload_url,
             )
         )
 
-        if auto_upload:
-            message = (
-                "执行已在后台启动（含自动上传）：调用 get_progress(job_id='{}', limit=20) "
-                "可获取构建和上传进度。完成后将返回网站访问URL。"
-            ).format(job_id)
-        else:
-            message = (
-                "执行已在后台启动：调用 get_progress(job_id='{}', limit=20) "
-                "或传入 progress_log='{}' 可获取实时进度"
-            ).format(job_id, progress_log_path or "<未启用进度日志>")
+        message = (
+            "执行已在后台启动（含自动上传）：调用 get_progress(job_id='{}', limit=20) "
+            "或传入 progress_log='{}' 可获取实时进度与上传结果"
+        ).format(job_id, progress_log_path or "<未启用进度日志>")
 
         return {
             "status": "started",
             "job_id": job_id,
             "plan_id": plan_id,
             "progress_log": progress_log_path,
-            "auto_upload": auto_upload,
-            "upload_url": upload_url if auto_upload else None,
+            "upload_url": DEFAULT_UPLOAD_URL,
             "message": message,
         }
     except Exception as exc:
@@ -686,8 +675,6 @@ async def _run_execution_job(
     plan_dict: Dict[str, Any],
     *,
     progress_log_path: Optional[str],
-    auto_upload: bool = False,
-    upload_url: str = None,
 ) -> None:
     job_info = _JOB_REGISTRY.get(job_id)
     if not job_info:
@@ -705,52 +692,7 @@ async def _run_execution_job(
         job_info["completed_at"] = time.time()
         _persist_job_state(job_id)
 
-        # 🎯 关键：任务完成后自动上传
-        if auto_upload and job_info.get("project_directory"):
-            try:
-                # 记录上传开始
-                job_info["upload_status"] = "uploading"
-                _persist_job_state(job_id)
-
-                # 只在不是默认值时传递upload_url
-                upload_params = {"folder_path": job_info["project_directory"]}
-                if upload_url and upload_url != DEFAULT_UPLOAD_URL:
-                    upload_params["upload_url"] = upload_url
-                
-                upload_result = await upload_project_to_mcp_server(**upload_params)
-
-                # 记录上传结果
-                job_info["upload_result"] = upload_result
-                job_info["upload_status"] = upload_result["status"]
-                if upload_result["status"] == "success":
-                    job_info["website_url"] = upload_result["url"]
-                    job_info["upload_completed_at"] = time.time()
-
-                    # 添加到进度日志
-                    if progress_log_path:
-                        upload_event = {
-                            "timestamp": time.time(),
-                            "type": "upload_completed",
-                            "status": "success",
-                            "website_url": upload_result["url"],
-                            "message": upload_result["message"],
-                        }
-                        try:
-                            with open(
-                                progress_log_path, "a", encoding="utf-8"
-                            ) as log_file:
-                                log_file.write(
-                                    json.dumps(upload_event, ensure_ascii=False)
-                                )
-                                log_file.write("\n")
-                        except Exception:
-                            pass
-
-                _persist_job_state(job_id)
-            except Exception as upload_exc:
-                job_info["upload_status"] = "failed"
-                job_info["upload_error"] = str(upload_exc)
-                _persist_job_state(job_id)
+        await _handle_auto_upload(job_id, job_info, progress_log_path)
 
     except Exception as exc:
         job_info["status"] = "failed"
@@ -764,37 +706,70 @@ async def _run_execution_job(
         _persist_job_state(job_id)
 
 
+async def _handle_auto_upload(
+    job_id: str,
+    job_info: Dict[str, Any],
+    progress_log_path: Optional[str],
+) -> None:
+    project_dir = job_info.get("project_directory")
+    if not project_dir:
+        return
+
+    job_info["upload_status"] = "uploading"
+    _persist_job_state(job_id)
+
+    try:
+        upload_result = await upload_project_to_mcp_server(
+            folder_path=project_dir,
+            upload_url=DEFAULT_UPLOAD_URL,
+        )
+
+        job_info["upload_result"] = upload_result
+        job_info["upload_status"] = upload_result.get("status")
+
+        if upload_result.get("status") == "success":
+            job_info["website_url"] = upload_result.get("url")
+            job_info["upload_completed_at"] = time.time()
+
+            if progress_log_path:
+                upload_event = {
+                    "timestamp": time.time(),
+                    "type": "upload_completed",
+                    "status": "success",
+                    "website_url": upload_result.get("url"),
+                    "message": upload_result.get("message"),
+                }
+                try:
+                    with open(progress_log_path, "a", encoding="utf-8") as log_file:
+                        log_file.write(json.dumps(upload_event, ensure_ascii=False))
+                        log_file.write("\n")
+                except Exception:
+                    pass
+
+    except Exception as exc:
+        job_info["upload_status"] = "failed"
+        job_info["upload_error"] = str(exc)
+    finally:
+        _persist_job_state(job_id)
+
+
 @mcp.tool()
 async def create_simple_site(
     description: str,
     site_title: str = "我的网站",
-    project_root: Optional[str] = None,
-    model: Optional[str] = None,
-    context_id: Optional[str] = None,
     context_content: Optional[str] = None,
 ) -> Dict[str, Any]:
     """使用AI分析需求，生成简单但美观的网站计划。
 
-    ⚠️ 重要参数说明：
+    参数说明：
     - description: 网站需求描述，例如"个人作品展示网站"、"小餐厅官网"、"博客网站"等
-    - site_title: 网站标题
-    - context_content: 🔥 核心参数！用于传递网页制作所需的所有原始数据内容
+    - site_title: 网站标题，默认为"我的网站"
+    - context_content: 可选，用于传递网页制作所需的所有原始数据内容
       * 例如：咖啡馆列表、产品介绍、菜单内容、地址信息、营业时间等
       * 这是AI获取具体业务信息的唯一渠道，请务必将查询到的详细信息完整传入
       * 如果有地图查询结果、API返回数据等，都应该放在这个参数中
       * 格式可以是文本、JSON字符串或结构化数据的字符串表示
 
-    其他参数：
-    - project_root: 项目根目录，缺省使用默认目录
-    - model: 使用的AI模型，缺省使用默认模型
-    - context_id: 可选，引用已缓存的上下文快照以复用历史资料
-
-    图片配置参数（可选）：
-    - image_style: 图片风格 (professional|artistic|minimal|vibrant|luxury)
-    - image_topics: 用户自定义的图片主题列表，如 ["modern office", "team collaboration"]
-    - include_gallery: 是否在网站中包含图片画廊功能
-    - image_provider: 图片提供商 (pollinations|dicebear|robohash)
-
     返回值说明：
     - status: 操作状态 ("success" 或 "error")
     - plan_id: 生成的计划唯一标识符，用于后续执行
@@ -813,22 +788,17 @@ async def create_simple_site(
     这样AI就能基于真实数据来生成个性化的网站内容。
     """
     try:
-        # 使用指定模型或默认模型
-        used_model = model or DEFAULT_MODEL
+        # 使用默认模型
+        used_model = DEFAULT_MODEL
 
-        # 使用新的路径解析逻辑
-        # 如果没有提供 project_root，使用 site_title 作为项目名
-        project_directory = _resolve_project_directory(project_root, site_title)
+        # 使用新的路径解析逻辑，默认在共享目录下按标题创建项目
+        project_directory = _resolve_project_directory(None, site_title)
 
         # 处理上下文
         context_data = ""
-        actual_context_id = context_id
+        actual_context_id: Optional[str] = None
 
-        if context_id and context_id in _CONTEXT_CACHE_BY_ID:
-            # 使用缓存的上下文
-            cached_context = _CONTEXT_CACHE_BY_ID[context_id]
-            context_data = cached_context.get("content", "")
-        elif context_content:
+        if context_content:
             # 使用新提供的上下文内容
             context_data = context_content
             # 生成新的上下文ID并缓存
@@ -919,13 +889,16 @@ async def create_simple_site(
             "plan_id": plan_id,
         }
 
+        if actual_context_id:
+            cached_entry["context_id"] = actual_context_id
+
         _PLAN_CACHE_BY_ID[plan_id] = cached_entry
         cache_key = (project_directory, description)
         _PLAN_CACHE[cache_key] = cached_entry
 
         # 将上下文信息关联到计划
         if actual_context_id:
-            _PROGRESS_LOG_BY_ID[plan_id] = actual_context_id
+            _CONTEXT_ID_BY_PLAN[plan_id] = actual_context_id
 
         # 保存计划到文件
         plan_filename = f"simple_site_plan_{plan_id}.json"
@@ -1029,6 +1002,20 @@ async def get_progress(
 
         if job_id:
             job_info = _JOB_REGISTRY.get(job_id)
+            if not job_info:
+                disk_state = _load_job_state_from_disk(job_id)
+                if disk_state:
+                    _JOB_REGISTRY[job_id] = disk_state
+                    job_info = disk_state
+                    progress_log_from_state = job_info.get("progress_log")
+                    if progress_log_from_state:
+                        progress_log_str = str(progress_log_from_state)
+                        _PROGRESS_LOG_BY_JOB[job_id] = progress_log_str
+                        plan_in_state = job_info.get("plan_id")
+                        if plan_in_state:
+                            _PROGRESS_LOG_BY_ID.setdefault(
+                                plan_in_state, progress_log_str
+                            )
             if job_info and not plan_id:
                 plan_id = job_info.get("plan_id")
             if job_id in _PROGRESS_LOG_BY_JOB:
@@ -1122,14 +1109,12 @@ async def get_progress(
                     "report": job_info.get("result", {}).get("report"),
                     "created_files": job_info.get("result", {}).get("created_files"),
                 }
-                # 添加上传结果信息
                 if job_info.get("upload_result"):
                     job_snapshot["upload_result"] = job_info.get("upload_result")
 
             if job_info.get("status") == "failed":
                 job_snapshot["error"] = job_info.get("error")
 
-            # 添加上传错误信息
             if job_info.get("upload_error"):
                 job_snapshot["upload_error"] = job_info.get("upload_error")
 
@@ -1147,15 +1132,11 @@ async def get_progress(
 @mcp.tool()
 async def upload_project_to_mcp_server(
     folder_path: str,
-    upload_url: Optional[str] = None,
 ) -> Dict[str, Any]:
     """将项目文件夹打包成ZIP并上传到MCP服务器。
 
     参数说明：
     - folder_path: 项目文件夹的绝对路径
-    - upload_url: 上传API地址（可选）
-        默认为 "https://www.mcpcn.cc/api/fileUploadAndDownload/uploadMcpFile"
-        一般不需要修改
 
     返回值：
     - status: 上传状态 ("success" 或 "error")
@@ -1164,9 +1145,6 @@ async def upload_project_to_mcp_server(
     - message: 状态信息
     """
     try:
-        # 使用默认URL如果没有提供
-        effective_upload_url = upload_url or DEFAULT_UPLOAD_URL
-        
         # 验证文件夹路径
         if not os.path.exists(folder_path):
             return {"status": "error", "message": f"项目文件夹不存在: {folder_path}"}
@@ -1206,7 +1184,7 @@ async def upload_project_to_mcp_server(
                     "file", f, filename=zip_filename, content_type="application/zip"
                 )
 
-                async with session.post(effective_upload_url, data=data) as response:
+                async with session.post(DEFAULT_UPLOAD_URL, data=data) as response:
                     response_text = await response.text()
 
                     if response.status != 200:

{htmlgen_mcp-0.3.7 → htmlgen_mcp-0.3.9}/src/htmlgen_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: htmlgen-mcp
-Version: 0.3.7
+Version: 0.3.9
 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