With UI changes and Server Fixes

README.md

@@ -11,22 +11,53 @@ pinned: false
 
 # 🧠 NotionTaskSense
 
-Turn vague goals into structured, actionable tasks — and push them directly into Notion.
+NotionTaskSense is an AI-powered tool that helps you organize your goals into structured tasks in Notion. It uses natural language processing to break down your goals into actionable tasks and automatically creates them in your Notion workspace.
 
-## ✨ What It Does
+## Features
 
-You type messy thoughts like:
+- 🤖 AI-powered task breakdown
+- 📝 Automatic Notion task creation
+- 🎯 Smart task organization
+- 🔄 Real-time processing
+- 🌐 API access available
 
-> I want to fix my resume, apply to jobs, study cloud certs, and message seniors.
+## How to Use
 
-And NotionTaskSense will:
+1. Enter your goals or tasks in natural language
+2. The AI will analyze and break them down into structured tasks
+3. Tasks are automatically created in your Notion workspace
+4. Access your organized tasks in Notion
 
-✅ Understand your intent  
-✅ Break it into clear tasks  
-✅ Add categories, priorities, due dates  
-✅ Push each task into a clean Notion database
+## API Usage
 
-No spreadsheets. No task lists. Just clarity.
+You can also use NotionTaskSense via its API endpoint:
+
+```bash
+curl -X POST http://your-space-url/api/predict \
+  -H "Content-Type: application/json" \
+  -d '{"data": ["Your task description here"], "fn_index": 0}'
+```
+
+## Environment Variables Required
+
+- `ANTHROPIC_API_KEY`: Your Anthropic API key for Claude
+- `NOTION_API_KEY`: Your Notion integration token
+- `NOTION_DATABASE_ID`: (Optional) Default Notion database ID
+
+## Local Development
+
+1. Clone the repository
+2. Install dependencies: `pip install -r requirements.txt`
+3. Set up environment variables
+4. Run the app: `python app.py`
+
+## Deployment
+
+This app is deployed on Hugging Face Spaces and can be accessed at [space-url].
+
+## License
+
+MIT License
 
 ---
 

app.py

@@ -1,39 +1,80 @@
 import gradio as gr
 import os
+from typing import Dict, List, Union, Any
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 from llm_parser import llm_parse_tasks
 from notion_handler import create_tasks_database, push_tasks_to_notion
 
-def organize_goals_into_notion(user_input: str) -> str:
+def organize_goals_into_notion(user_input: str) -> Dict[str, Any]:
+    """
+    Process a single goal into Notion tasks.
+    Returns a dictionary with success status and detailed response.
+    """
     try:
+        logger.info(f"Processing user input: {user_input}")
         tasks = llm_parse_tasks(user_input)
         if isinstance(tasks, str):
-            return "Failed to parse tasks properly:\n\n" + tasks
+            logger.error(f"Failed to parse tasks: {tasks}")
+            return {
+                "success": False,
+                "error": "Failed to parse tasks",
+                "details": tasks
+            }
         
         db_id = create_tasks_database()
         push_tasks_to_notion(tasks, db_id)
 
         task_names = [t.get("task", "Untitled") for t in tasks]
-        return f"✅ {len(task_names)} tasks added to Notion:\n- " + "\n- ".join(task_names)
+        logger.info(f"Successfully created {len(task_names)} tasks")
+        return {
+            "success": True,
+            "message": f"✅ {len(task_names)} tasks added to Notion",
+            "tasks": task_names,
+            "database_id": db_id
+        }
     except Exception as e:
-        return f"❌ Error: {e}"
+        logger.error(f"Error processing request: {str(e)}")
+        return {
+            "success": False,
+            "error": str(e),
+            "details": "An error occurred while processing your request"
+        }
 
-# Gradio MCP app
-# with gr.Blocks() as demo:
-#     gr.Markdown("# 🧠 NotionTaskSense – Turn your goals into Notion tasks")
-#     inp = gr.Textbox(placeholder="What do you want to get done this week?", lines=4)
-#     out = gr.Textbox(label="Agent Output")
+# Create Gradio interface
+demo = gr.Interface(
+    fn=organize_goals_into_notion,
+    inputs=gr.Textbox(
+        lines=4,
+        placeholder="What do you want to get done this week?",
+        label="Your Goals"
+    ),
+    outputs=gr.JSON(label="Results"),
+    title="🧠 NotionTaskSense",
+    description="Turn your goals into structured Notion tasks",
+    examples=[
+        ["I need to update my resume, apply to 3 jobs, and study for AWS certification"],
+        ["Meet with team about project timeline, prepare presentation for client, and follow up on pending emails"],
+    ],
+    theme=gr.themes.Soft()
+)
 
-#     btn = gr.Button("Organize into Notion")
-#     btn.click(fn=organize_goals_into_notion, inputs=inp, outputs=out)
+# Add MCP metadata
+organize_goals_into_notion.mcp_type = "tool"
+organize_goals_into_notion.mcp_description = "Organize goals into structured Notion tasks"
+organize_goals_into_notion.mcp_input_type = "string"
+organize_goals_into_notion.mcp_output_type = "json"
 
-demo = gr.Interface(fn=organize_goals_into_notion,
-                    inputs=gr.Textbox(lines=4, placeholder="What do you want to get done this week?"),
-                    outputs=gr.Textbox(label="Agent Output"),
-                    title="🧠 NotionTaskSense",
-                    description="Turn your goals into Notion tasks")
+if __name__ == "__main__":
+    demo.queue().launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )
 
 
 
-if __name__ == "__main__":
-    demo.launch()

llm_parser.py

@@ -1,40 +1,72 @@
 import anthropic
 import os
 import json
+from typing import List, Dict, Union, Any
 from dotenv import load_dotenv
 
 load_dotenv()  # Load environment variables from .env file
 
+# Check for required environment variables
+if not os.getenv("ANTHROPIC_API_KEY"):
+    raise ValueError("ANTHROPIC_API_KEY environment variable is not set")
+
 client = anthropic.Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
 
-def llm_parse_tasks(user_input):
+def llm_parse_tasks(user_input: str) -> Union[List[Dict[str, Any]], str]:
+    """
+    Parse user input into structured tasks using Claude 3.
+    
+    Args:
+        user_input (str): The user's natural language input describing their goals
+        
+    Returns:
+        Union[List[Dict[str, Any]], str]: Either a list of task dictionaries or an error message
+    """
     prompt = f"""
 You are a helpful assistant that turns vague personal goals into clear, actionable tasks.
 
 Break this input into tasks with:
-- task name
-- category (Career, Learning, Personal, Outreach, etc.)
-- priority (High, Medium, Low)
-- optional due date (if mentioned or inferred)
-- short notes if relevant
+- task name (key: "task")
+- category (key: "category") - Choose from: Career, Learning, Personal, Outreach, Health, Finance
+- priority (key: "priority") - Choose from: High, Medium, Low
+- optional due date (key: "dueDate") - Use ISO format YYYY-MM-DD if date is mentioned or inferred
+- notes (key: "notes") - Add relevant context or subtasks
 
-Return the output as a JSON array of task objects.
+Return ONLY a valid JSON array of task objects with the exact keys mentioned above.
+Each task MUST have at least task, category, and priority fields.
 
 Input: \"\"\"{user_input}\"\"\"
 """
 
-    response = client.messages.create(
-        model="claude-3-haiku-20240307",
-        max_tokens=1024,
-        temperature=0.4,
-        messages=[
-            {"role": "user", "content": prompt}
-        ]
-    )
+    try:
+        response = client.messages.create(
+            model="claude-3-haiku-20240307",
+            max_tokens=1024,
+            temperature=0.4,
+            messages=[
+                {"role": "user", "content": prompt}
+            ]
+        )
+
+        content = response.content[0].text.strip()
+        tasks = json.loads(content)
 
-    content = response.content[0].text.strip()
+        # Validate required fields
+        for task in tasks:
+            if not all(k in task for k in ["task", "category", "priority"]):
+                return "Error: Each task must have task, category, and priority fields"
+            
+            # Validate category and priority values
+            if task["category"] not in ["Career", "Learning", "Personal", "Outreach", "Health", "Finance"]:
+                return f"Error: Invalid category '{task['category']}' in task '{task['task']}'"
+            
+            if task["priority"] not in ["High", "Medium", "Low"]:
+                return f"Error: Invalid priority '{task['priority']}' in task '{task['task']}'"
 
-    try:
-        return json.loads(content)
+        return tasks
+    except anthropic.APIError as e:
+        return f"Claude API Error: {str(e)}"
     except json.JSONDecodeError:
-        return content
+        return "Error: Failed to parse Claude's response as JSON"
+    except Exception as e:
+        return f"Unexpected error: {str(e)}"

notion_handler.py

@@ -1,50 +1,103 @@
 import os
+from typing import List, Dict, Any
 from notion_client import Client
 from datetime import datetime
 from dotenv import load_dotenv
 
 load_dotenv()  # Load environment variables from .env file
 
+# Check for required environment variables
+if not os.getenv("NOTION_TOKEN"):
+    raise ValueError("NOTION_TOKEN environment variable is not set")
+if not os.getenv("NOTION_PARENT_PAGE_ID"):
+    raise ValueError("NOTION_PARENT_PAGE_ID environment variable is not set")
+
 notion = Client(auth=os.getenv("NOTION_TOKEN"))
 parent_page_id = os.getenv("NOTION_PARENT_PAGE_ID")
 
-# Reusable Notion DB schema
-def create_tasks_database():
-    response = notion.databases.create(
-        parent={"type": "page_id", "page_id": parent_page_id},
-        title=[{"type": "text", "text": {"content": "NotionTaskSense Tasks"}}],
-        properties={
-            "Task": {"title": {}},
-            "Category": {"select": {"options": []}},  # Options added dynamically
-            "Priority": {"select": {"options": []}},
-            "Due Date": {"date": {}},
-            "Status": {"select": {"options": [{"name": "To Do"}, {"name": "In Progress"}, {"name": "Done"}]}},
-            "Notes": {"rich_text": {}}
-        }
-    )
-    return response["id"]
-
-# Add or update tasks into the database
-def push_tasks_to_notion(tasks, db_id):
-    for task in tasks:
-        properties = {
-            "Task": {"title": [{"text": {"content": task.get("taskName", "Untitled Task")}}]},
-            "Category": {"select": {"name": task.get("category", "General")}},
-            "Priority": {"select": {"name": task.get("priority", "Medium")}},
-            "Status": {"select": {"name": "To Do"}},
-        }
-
-        # Optional fields
-        if "dueDate" in task:
-            try:
+def create_tasks_database() -> str:
+    """
+    Create a new Notion database for tasks.
+    
+    Returns:
+        str: The ID of the created database
+    
+    Raises:
+        Exception: If database creation fails
+    """
+    try:
+        response = notion.databases.create(
+            parent={"type": "page_id", "page_id": parent_page_id},
+            title=[{"type": "text", "text": {"content": f"NotionTaskSense Tasks - {datetime.now().strftime('%Y-%m-%d')}"}}],
+            properties={
+                "Task": {"title": {}},
+                "Category": {
+                    "select": {
+                        "options": [
+                            {"name": "Career", "color": "blue"},
+                            {"name": "Learning", "color": "green"},
+                            {"name": "Personal", "color": "orange"},
+                            {"name": "Outreach", "color": "purple"},
+                            {"name": "Health", "color": "red"},
+                            {"name": "Finance", "color": "yellow"}
+                        ]
+                    }
+                },
+                "Priority": {
+                    "select": {
+                        "options": [
+                            {"name": "High", "color": "red"},
+                            {"name": "Medium", "color": "yellow"},
+                            {"name": "Low", "color": "blue"}
+                        ]
+                    }
+                },
+                "Due Date": {"date": {}},
+                "Status": {
+                    "select": {
+                        "options": [
+                            {"name": "To Do", "color": "gray"},
+                            {"name": "In Progress", "color": "blue"},
+                            {"name": "Done", "color": "green"}
+                        ]
+                    }
+                },
+                "Notes": {"rich_text": {}}
+            }
+        )
+        return response["id"]
+    except Exception as e:
+        raise Exception(f"Failed to create Notion database: {str(e)}")
+
+def push_tasks_to_notion(tasks: List[Dict[str, Any]], db_id: str) -> None:
+    """
+    Add or update tasks in the Notion database.
+    
+    Args:
+        tasks (List[Dict[str, Any]]): List of task dictionaries
+        db_id (str): Notion database ID
+    
+    Raises:
+        Exception: If pushing tasks fails
+    """
+    try:
+        for task in tasks:
+            properties = {
+                "Task": {"title": [{"text": {"content": task.get("task", "Untitled Task")}}]},
+                "Category": {"select": {"name": task.get("category", "Personal")}},
+                "Priority": {"select": {"name": task.get("priority", "Medium")}},
+                "Status": {"select": {"name": "To Do"}},
+            }
+
+            if "dueDate" in task and task["dueDate"]:
                 properties["Due Date"] = {"date": {"start": task["dueDate"]}}
-            except:
-                pass
 
-        if "notes" in task:
-            properties["Notes"] = {"rich_text": [{"text": {"content": task["notes"]}}]}
+            if "notes" in task and task["notes"]:
+                properties["Notes"] = {"rich_text": [{"text": {"content": task["notes"]}}]}
 
-        notion.pages.create(
-            parent={"database_id": db_id},
-            properties=properties
-        )
+            notion.pages.create(
+                parent={"database_id": db_id},
+                properties=properties
+            )
+    except Exception as e:
+        raise Exception(f"Failed to push tasks to Notion: {str(e)}")

requirements.txt

@@ -7,11 +7,11 @@ charset-normalizer==3.4.2
 click==8.2.1
 distro==1.9.0
 exceptiongroup==1.3.0
-fastapi==0.115.12
+fastapi>=0.110.0
 ffmpy==0.6.0
 filelock==3.18.0
 fsspec==2025.5.1
-gradio==5.33.0
+gradio>=4.19.2
 gradio_client==1.10.2
 groovy==0.1.2
 h11==0.16.0
@@ -25,7 +25,7 @@ jiter==0.10.0
 markdown-it-py==3.0.0
 MarkupSafe==3.0.2
 mdurl==0.1.2
-notion-client==2.3.0
+notion-client>=2.2.1
 numpy==2.2.6
 openai==1.85.0
 orjson==3.10.18
@@ -37,7 +37,7 @@ pydantic_core==2.33.2
 pydub==0.25.1
 Pygments==2.19.1
 python-dateutil==2.9.0.post0
-python-dotenv==1.1.0
+python-dotenv>=1.0.1
 python-multipart==0.0.20
 pytz==2025.2
 PyYAML==6.0.2
@@ -57,5 +57,5 @@ typing-inspection==0.4.1
 typing_extensions==4.14.0
 tzdata==2025.2
 urllib3==2.4.0
-uvicorn==0.34.3
+uvicorn>=0.27.1
 websockets==15.0.1