modified: .gitignore

modified: README.md
modified: app.py
modified: prompt.txt
renamed: ai_transform.py -> src/ai_transform.py
renamed: utils.py -> src/utils.py
new file: templates/index.html

.gitignore

@@ -1,5 +1,6 @@
 secret.env
 test*.py
+test/
 notion_data.json
 prompt_1.txt
 prompt_2.txt

README.md

@@ -7,10 +7,16 @@ sdk: docker
 pinned: false
 license: apache-2.0
 short_description: Push my data to web and transfer to Notion
+version: 2.0.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
+# update2
+build a async frontend for visualization
+set a debugging mode, which skips the entire Notion API interaction loop.(accept debug as a query)
+reconstruct the modules.
+
 # update
 - gpt 3.5 to classify the project.
 - generalize the model usage.
@@ -23,6 +29,10 @@ curl -X POST https://beanundlaugh-notion-pusher.hf.space/ \
 
 content should be a string with form "content"
 
+or
+
+visit the root and directly enter the content.
+
 # test locally
 ## get_all function
 

app.py

@@ -1,34 +1,18 @@
-from flask import Flask, request, jsonify
-from ai_transform import gemini_get_answer, classify_task_with_ai
-from utils import *
+from flask import Flask, request, jsonify, render_template
+from src.ai_transform import gemini_get_answer, classify_task_with_ai, modified_with_ai
+from src.utils import *
 import requests, os, datetime
-import time
 import json
+
+# for offline debugging
+if os.path.exists("secret.env"):
+    from dotenv import load_dotenv
+    load_dotenv("secret.env")
 app = Flask(__name__)
 
 NOTION_TOKEN = os.environ.get("NOTION_TOKEN")
 DB_ID = os.environ.get("DB_ID")
-with open("prompt.txt", encoding="utf-8") as f:
-    system_content = f.read()
 
-def time_cali():
-    # 拿到 UTC 时间戳
-    ts = time.time()
-    # 手动加 8 小时
-    beijing_ts = ts + 8*3600
-    # 再转回 struct_time
-    beijing_t = time.gmtime(beijing_ts)
-    # print(time.strftime('%Y-%m-%d', beijing_t))
-    return time.strftime('%Y-%m-%d', beijing_t)
-def modified_with_ai(items):
-    # 在这里调用AI模型对item进行修改
-    # today_date = time.strftime("%Y-%m-%d", time.localtime())
-    today_date = time_cali()
-    system_instu = system_content.replace("today", today_date)
-    items = gemini_get_answer(items, system_instu)
-    print(f"--- AI modification complete ---")
-    # print(items)
-    return items
 def get_property(it, task_name, url, api_key, model):
     properties = {
         "Task name": {
@@ -81,23 +65,38 @@ def init_items(items):
     # items = ",".join(line.strip() for line in items.splitlines() if line.strip())
     return items
 
+@app.route("/", methods=["GET"])
+def index():
+    return render_template("index.html")
+
 @app.route("/", methods=["POST"])
 def push():
     # items = request.get_json(force=True)
-    items = request.get_data(as_text=True)
+    if request.content_type == 'text/plain':
+        items = request.get_data(as_text=True)
+    else:  # application/json
+        items = request.get_json()['text']
+    # implement AI
+    ai_implemented = "gpt"
+    if ai_implemented is not None:
+        ai = load_ai_config(ai_implemented)
     items = init_items(items)
-    items= modified_with_ai(items)
+    items = modified_with_ai(items, url=ai["url"], api_key=ai["api_key"], model=ai["model"])
     # change to json format
     items = json.loads(items)
+    debug_mode = request.args.get("debug") == "true"
+    print("Debug mode:", debug_mode)
+    if debug_mode:
+        return jsonify({"items": items, "debug": True, "ok": True})
+
+    # push to notion
+
     url = "https://api.notion.com/v1/pages"
     headers = {
         "Authorization": f"Bearer {NOTION_TOKEN}",
         "Content-Type": "application/json",
         "Notion-Version": "2022-06-28"
     }
-    ai_implemented = "gpt"
-    if ai_implemented is not None:
-        ai = load_ai_config(ai_implemented)
     # print("ai_config:", ai)
     for it in items:
         task_name = it.get("Task name", "Untitled Task")
@@ -108,7 +107,8 @@ def push():
         else:
             print(f"Successfully pushed item: {it.get('Task name', 'Untitled Task')}")
         # print(r.status_code, r.text)
-    return jsonify({"ok": True, "inserted": len(items)})
+    return jsonify({"ok": True, "inserted": len(items), "debug": False, "items": items})
+
 
 @app.route("/get", methods=["GET"])
 def get_all_data():
@@ -143,4 +143,4 @@ def get_all_data():
             
     return jsonify({"ok": True, "data": all_pages, "count": len(all_pages)})
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860)# , debug=True

prompt.txt

@@ -29,6 +29,7 @@
     *   **傍晚/夜晚关键词**: "dinner", "supper", "evening", "night", "lecturing", "relaxing", "streaming" 等词语旁的数字，应优先推断为 **PM (12:00 - 23:00)**。
     *   **早晨/白天关键词**: "breakfast", "morning", "晨会", "standup", "meeting", "work" 等词语旁的数字，应优先推断为 **AM 或工作时间 (07:00 - 18:00)**。
     *   **无明确线索**: 如果没有任何上下文线索，根据你的世界知识做出最合理的判断（例如，一个在“2点”的“商务会议”极有可能是下午14:00，而不是凌晨02:00）。
+    *   注意：不要返回{current_date}字符，要根据给定的日期替换。
 *   可以推理device，如streaming为Phone，project working一般是电脑（涉及编程）
 
 ### 5. 示例 (必须严格模仿)

ai_transform.py → src/ai_transform.py

@@ -3,13 +3,28 @@ import json
 import requests
 import os
 import google.generativeai as genai
-from utils import *
-# from dotenv import load_dotenv
-# load_dotenv("secret.env")
+from src.utils import *
 ST_API_PASS = os.environ.get("ST_API_PASSWORD")
 ST_api_key = f"Bearer {ST_API_PASS}"
 ST_url = "https://spark-api-open.xf-yun.com/v1/chat/completions"
 
+with open("prompt.txt", encoding="utf-8") as f:
+    system_content = f.read()
+
+def modified_with_ai(items, url=ST_url, api_key=ST_api_key, model=None):
+    # 在这里调用AI模型对item进行修改
+    # today_date = time.strftime("%Y-%m-%d", time.localtime())
+    today_date = time_cali()
+    system_instu = system_content.replace("today", today_date)
+    if "gemini" in url:
+        items = gemini_get_answer(items, system_instu)
+    else:
+        items = _get_answer(items, url, api_key, model, system_instu)
+    items = items.replace("{current_date}", today_date)
+    print(f"--- AI modification complete ---")
+    # print(items)
+    return items
+
 def classify_task_with_ai(task_info: str, url=ST_url, api_key=ST_api_key, model=None, categories_filepath: str = "related_tasks.txt", config = None) -> str:
     """
     使用 ai 模型和 RAG 方式对任务名称进行分类。
@@ -25,7 +40,7 @@ def classify_task_with_ai(task_info: str, url=ST_url, api_key=ST_api_key, model=
     api_key : str, optional
         API 请求的密钥，默认为 ST_api_key。
     model : str, optional
-        使用的模型，默认为 None。
+        使用的模型-> 使用init_model根据url自动选择, 如:gpt-3.5-turbo, 默认为 None。
     返回
     ----
     str
@@ -79,6 +94,7 @@ def gemini_get_answer(user_input: str, system_prompt: str = "") -> str:
     # 2. 创建模型实例
     model = genai.GenerativeModel(
         model_name="gemini-2.5-flash",
+        # model_name = "gemini-2.0-flash-lite",
         system_instruction=system_prompt or None  # 空字符串时传 None，避免多余字段
     )
 
@@ -94,10 +110,10 @@ def _init_model(url):
         model = "lite"
     # could be extended
     # elif "gemini" in url:
-    #     model = "gemini-2.5-flash"
+    #     model = "gemini-2.5-lite"
     return model
 
-def _get_answer(message, url=ST_url, api_key=ST_api_key, model=None):
+def _get_answer(message, url=ST_url, api_key=ST_api_key, model=None, system_instr=None):
     """
     Accepts a single string `message` as the prompt and returns the model-generated response.
     """
@@ -115,14 +131,19 @@ def _get_answer(message, url=ST_url, api_key=ST_api_key, model=None):
     if model is None:
         model = _init_model(url)
     # 这里的 message 应该是我们之前设计好的、包含示例的完整提示词
+    message_list =[]
+    if system_instr is not None:
+        message_list.append({
+            "role": "system",
+            "content": system_instr
+        })
+    message_list.append({
+        "role": "user",
+        "content": message
+    })
     body = {
         "model": model,
-        "messages": [
-            {
-                "role": "user",
-                "content": message
-            },
-        ],
+        "messages": message_list,
         "stream": True
     }
     

utils.py → src/utils.py

@@ -1,4 +1,6 @@
 import os
+import time
+
 def load_task_mapping_from_txt(filepath: str = "task_mapping.txt") -> dict:
     """
     从 txt 文件中读取 "类别名称: ID" 格式的数据，并返回一个字典。
@@ -53,6 +55,16 @@ def load_ai_config(name = "default"):
     config["api_key"] = "Bearer " + os.environ.get(f"{API_MODEL}_KEY")
     return config
 
+def time_cali():
+    # 拿到 UTC 时间戳
+    ts = time.time()
+    # 手动加 8 小时
+    beijing_ts = ts + 8*3600
+    # 再转回 struct_time
+    beijing_t = time.gmtime(beijing_ts)
+    # print(time.strftime('%Y-%m-%d', beijing_t))
+    return time.strftime('%Y-%m-%d', beijing_t)
+
 if __name__ == "__main__":
     mapping = load_task_mapping_from_txt("related_tasks.txt")
     # print(mapping.keys())

templates/index.html

@@ -0,0 +1,220 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <title>Notion Task Pusher</title>
+    <style>
+        /* 黑色主题的样式 */
+        body { 
+            font-family: sans-serif; 
+            max-width: 80%; 
+            margin: 40px auto; 
+            padding: 20px; 
+            border: 1px solid #444; 
+            border-radius: 8px; 
+            background-color: #1a1a1a; /* 深色背景 */
+            color: #f0f0f0; /* 浅色文本 */
+        }
+        h1 {
+            color: #00bcd4; /* 强调色，例如青色 */
+        }
+        p {
+            color: #ccc;
+        }
+        textarea { 
+            width: 100%; 
+            height: 150px; 
+            padding: 10px; 
+            box-sizing: border-box; 
+            margin-bottom: 10px; 
+            border: 1px solid #333; 
+            background-color: #2c2c2c; /* 输入框背景 */
+            color: #f0f0f0; /* 输入框文本颜色 */
+        }
+        button { 
+            padding: 10px 15px; 
+            background-color: #00bcd4; /* 按钮背景使用强调色 */
+            color: #1a1a1a; /* 按钮文本为深色 */
+            border: none; 
+            border-radius: 5px; 
+            cursor: pointer; 
+            transition: background-color 0.3s;
+        }
+        button:hover { 
+            background-color: #008ba3; /* 鼠标悬停时的深一点的强调色 */
+        }
+        /* #jsonOutput {
+            height: 300px;
+            margin-top: 20px;
+            color: #00bcd4;
+            background-color: #111;
+        } */
+        .container {
+            display: flex;
+            gap: 20px;          /* 左右间距 */
+            /* max-width: 1200px;  视情况可删 */
+            width: 100%;
+            margin: 0 auto;     /* 居中 */
+        }
+
+        /* 左侧主体：自适应剩余宽度 */
+        .left-main {
+            flex: 1;
+        }
+
+        /* 右侧边栏：固定 340px 宽 */
+        .right-aside {
+            width: 40%;
+        }
+        .Output {
+            width: 100%;
+            /* height: 400px; */
+            height: 30vh;
+            resize: none;
+            color: #00bcd4;
+            background-color: #111;
+            border: 1px solid #00bcd4;
+            border-radius: 4px;
+            padding: 8px;
+            font-family: monospace;
+        }
+        .Input {
+            width: 100%;
+            height: 80%;      /* 想撑满父级可用 calc(100vh - 40px) */
+            resize: none;
+            color: #f0f0f0;
+            background-color: #2c2c2c;
+            border: 1px solid #444;
+            border-radius: 4px;
+            padding: 8px;
+            font-family: monospace;
+        }
+        #response {
+            margin-top: 20px;
+            padding: 10px;
+            border-left: 5px solid #00bcd4; /* 响应区使用强调色 */
+            background-color: #2c2c2c; 
+            white-space: pre-wrap; 
+            color: #f0f0f0;
+        }
+    </style>
+</head>
+<body>
+    <h1>Notion 任务快速录入</h1>
+    <div class="container">
+    <!-- 左侧主体内容 -->
+    <main class="left-main">
+        <textarea class="Input" id="taskInput" placeholder="例如：明天上午10点 设计评审会, 下午4-6点 团队建设 high priority, 用pad"></textarea>
+        <button onclick="submitTasks()">提交到 Notion</button>
+        <label style="margin-left: 20px; color: #ccc;">
+            <input type="checkbox" id="debugMode" style="vertical-align: middle;"> 调试模式
+        </label>
+        <div id="response"></div>
+    </main>
+
+    <!-- 右侧边栏 -->
+    <aside class="right-aside">
+        <textarea class="Output" id="jsonOutput" placeholder="AI 转换结果将在此处显示..." readonly></textarea>
+    </aside>
+    </div>
+    <script>
+        async function submitTasks() {
+            const input = document.getElementById('taskInput').value;
+            const debugMode = document.getElementById('debugMode').checked; 
+            const responseDiv = document.getElementById('response');
+            const jsonOutputArea = document.getElementById('jsonOutput'); 
+            responseDiv.innerHTML = '正在处理...';
+            responseDiv.style.display = 'block';
+            jsonOutputArea.value = ''; // 清空之前的内容
+            // 构建 URL, 如果是调试模式，则添加 ?debug=true
+            const url = debugMode ? '/?debug=true' : '/';
+
+            try {
+                const response = await fetch(url, {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'text/plain' 
+                    },
+                    body: input
+                });
+
+                if (!response.ok) {
+                    const text = await response.text();
+                    // 抛出错误，将被 catch 块捕获
+                    throw new Error(`HTTP Error ${response.status}: ${text}`);
+                }
+                
+                const data = await response.json();
+                
+                if (data.ok) {
+                    if (data.debug) {
+                        // 调试模式的特殊显示
+                        responseDiv.innerHTML = `✅ **调试模式开启：未推送到 Notion**<br><br>AI 转换后的 JSON 数据`;
+                        responseDiv.style.borderLeftColor = 'orange';
+                    } else {
+                        // 正常模式的显示
+                        responseDiv.innerHTML = `✅ 提交成功! 共插入 ${data.inserted} 个任务。`;
+                        responseDiv.style.borderLeftColor = '#00bcd4';
+                    }
+                    // 在右侧文本区域显示 AI 转换结果
+                    jsonOutputArea.value = JSON.stringify(data.items, null, 2);
+                } else {
+                    // 业务逻辑错误
+                    responseDiv.innerHTML = `❌ 提交失败: ${data.error || '未知错误'}`;
+                    responseDiv.style.borderLeftColor = 'red';
+                }
+
+            } catch (error) {
+                // 网络或代码/JSON 解析错误
+                responseDiv.innerHTML = `❌ 发生错误: ${error.message}`;
+                responseDiv.style.borderLeftColor = 'red';
+                console.error('Error:', error);
+            }
+        }
+        function _submitTasks() {
+            const input = document.getElementById('taskInput').value;
+            const debugMode = document.getElementById('debugMode').checked; // 获取调试模式状态
+            const responseDiv = document.getElementById('response');
+            responseDiv.innerHTML = '正在处理...';
+
+            // 构建 URL, 如果是调试模式，则添加 ?debug=true
+            const url = debugMode ? '/?debug=true' : '/';
+
+            fetch(url, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'text/plain' 
+                },
+                body: input
+            })
+            .then(response => {
+                if (!response.ok) {
+                    return response.text().then(text => Promise.reject(new Error(`HTTP Error ${response.status}: ${text}`)));
+                }
+                // 无论是否调试模式，都期望返回 JSON
+                return response.json(); 
+            })
+            .then(data => {
+                if (data.ok) {
+                    if (data.debug) {
+                        // 调试模式的特殊显示
+                        responseDiv.innerHTML = `✅ **调试模式开启：未推送到 Notion**<br><br>AI 转换后的 JSON 数据：\n${JSON.stringify(data.ai_output, null, 2)}`;
+                        responseDiv.style.borderLeftColor = 'orange';
+                    } else {
+                        responseDiv.innerHTML = `✅ 提交成功! 共插入 ${data.inserted} 个任务。`;
+                        responseDiv.style.borderLeftColor = '#00bcd4';
+                    }
+                } else {
+                    responseDiv.innerHTML = `❌ 提交失败: ${data.error || '未知错误'}`;
+                    responseDiv.style.borderLeftColor = 'red';
+                }
+            })
+            .catch(error => {
+                responseDiv.innerHTML = `❌ 发生错误: ${error.message}`;
+                responseDiv.style.borderLeftColor = 'red';
+                console.error('Error:', error);
+            });
+        }
+    </script>
+</body>
+</html>