Upload 61 files

app/api/routes/projects.py

@@ -0,0 +1,140 @@
+"""
+Projects API Routes - Workspace management
+"""
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
+from typing import Optional, List
+from datetime import datetime
+
+from app.core.database import get_db
+from app.models import Project, Entity, Relationship
+
+
+router = APIRouter(prefix="/projects", tags=["Projects"])
+
+
+class ProjectCreate(BaseModel):
+    name: str
+    description: Optional[str] = None
+    color: str = "#00d4ff"
+    icon: str = "folder"
+
+
+class ProjectResponse(BaseModel):
+    id: str
+    name: str
+    description: Optional[str]
+    color: str
+    icon: str
+    entity_count: int = 0
+    created_at: datetime
+    
+    class Config:
+        from_attributes = True
+
+
+@router.get("", response_model=List[ProjectResponse])
+def list_projects():
+    """List all projects"""
+    db = next(get_db())
+    projects = db.query(Project).order_by(Project.created_at.desc()).all()
+    
+    result = []
+    for p in projects:
+        entity_count = db.query(Entity).filter(Entity.project_id == p.id).count()
+        result.append(ProjectResponse(
+            id=p.id,
+            name=p.name,
+            description=p.description,
+            color=p.color,
+            icon=p.icon,
+            entity_count=entity_count,
+            created_at=p.created_at
+        ))
+    
+    return result
+
+
+@router.post("", response_model=ProjectResponse)
+def create_project(project: ProjectCreate):
+    """Create a new project"""
+    db = next(get_db())
+    
+    new_project = Project(
+        name=project.name,
+        description=project.description,
+        color=project.color,
+        icon=project.icon
+    )
+    db.add(new_project)
+    db.commit()
+    db.refresh(new_project)
+    
+    return ProjectResponse(
+        id=new_project.id,
+        name=new_project.name,
+        description=new_project.description,
+        color=new_project.color,
+        icon=new_project.icon,
+        entity_count=0,
+        created_at=new_project.created_at
+    )
+
+
+@router.get("/{project_id}", response_model=ProjectResponse)
+def get_project(project_id: str):
+    """Get project by ID"""
+    db = next(get_db())
+    project = db.query(Project).filter(Project.id == project_id).first()
+    
+    if not project:
+        raise HTTPException(status_code=404, detail="Project not found")
+    
+    entity_count = db.query(Entity).filter(Entity.project_id == project_id).count()
+    
+    return ProjectResponse(
+        id=project.id,
+        name=project.name,
+        description=project.description,
+        color=project.color,
+        icon=project.icon,
+        entity_count=entity_count,
+        created_at=project.created_at
+    )
+
+
+@router.delete("/{project_id}")
+def delete_project(project_id: str):
+    """Delete project and optionally its entities"""
+    db = next(get_db())
+    project = db.query(Project).filter(Project.id == project_id).first()
+    
+    if not project:
+        raise HTTPException(status_code=404, detail="Project not found")
+    
+    # Set entities and relationships to no project (null)
+    db.query(Entity).filter(Entity.project_id == project_id).update({"project_id": None})
+    db.query(Relationship).filter(Relationship.project_id == project_id).update({"project_id": None})
+    
+    db.delete(project)
+    db.commit()
+    
+    return {"message": f"Project '{project.name}' deleted"}
+
+
+@router.put("/{project_id}")
+def update_project(project_id: str, project: ProjectCreate):
+    """Update project"""
+    db = next(get_db())
+    existing = db.query(Project).filter(Project.id == project_id).first()
+    
+    if not existing:
+        raise HTTPException(status_code=404, detail="Project not found")
+    
+    existing.name = project.name
+    existing.description = project.description
+    existing.color = project.color
+    existing.icon = project.icon
+    db.commit()
+    
+    return {"message": "Project updated"}

app/core/database.py

@@ -54,4 +54,19 @@ def init_db():
             print("✅ Added event_date column to relationships table")
         except Exception:
             pass  # Column already exists
-
+        
+        # Add project_id to entities if not exists
+        try:
+            conn.execute(text("ALTER TABLE entities ADD COLUMN project_id VARCHAR(36)"))
+            conn.commit()
+            print("✅ Added project_id column to entities table")
+        except Exception:
+            pass  # Column already exists
+        
+        # Add project_id to relationships if not exists
+        try:
+            conn.execute(text("ALTER TABLE relationships ADD COLUMN project_id VARCHAR(36)"))
+            conn.commit()
+            print("✅ Added project_id column to relationships table")
+        except Exception:
+            pass  # Column already exists

app/main.py

@@ -8,7 +8,7 @@ from contextlib import asynccontextmanager
 
 from app.config import settings
 from app.core.database import init_db
-from app.api.routes import entities, relationships, events, search, ingest, analyze, graph, research, chat, investigate, dados_publicos, timeline
+from app.api.routes import entities, relationships, events, search, ingest, analyze, graph, research, chat, investigate, dados_publicos, timeline, projects
 
 
 @asynccontextmanager
@@ -62,6 +62,7 @@ app.include_router(chat.router, prefix="/api/v1")
 app.include_router(investigate.router, prefix="/api/v1")
 app.include_router(dados_publicos.router, prefix="/api/v1")
 app.include_router(timeline.router, prefix="/api/v1")
+app.include_router(projects.router, prefix="/api/v1")
 
 
 @app.get("/")

app/models/__init__.py

@@ -1,2 +1,3 @@
 # Models module
 from app.models.entity import Entity, Relationship, Event, Document
+from app.models.project import Project

app/models/entity.py

@@ -21,6 +21,7 @@ class Entity(Base):
     __tablename__ = "entities"
     
     id = Column(String(36), primary_key=True, default=generate_uuid)
+    project_id = Column(String(36), ForeignKey("projects.id"), nullable=True, index=True)
     type = Column(String(50), nullable=False, index=True)  # person, organization, location, etc
     name = Column(String(255), nullable=False, index=True)
     description = Column(Text, nullable=True)
@@ -62,6 +63,7 @@ class Relationship(Base):
     __tablename__ = "relationships"
     
     id = Column(String(36), primary_key=True, default=generate_uuid)
+    project_id = Column(String(36), ForeignKey("projects.id"), nullable=True, index=True)
     source_id = Column(String(36), ForeignKey("entities.id"), nullable=False)
     target_id = Column(String(36), ForeignKey("entities.id"), nullable=False)
     type = Column(String(50), nullable=False, index=True)  # works_for, knows, owns, etc

app/models/project.py

@@ -0,0 +1,29 @@
+"""
+Project Model - Workspaces for organizing investigations
+"""
+from sqlalchemy import Column, String, Text, DateTime
+from datetime import datetime
+import uuid
+
+from app.core.database import Base
+
+
+def generate_uuid():
+    return str(uuid.uuid4())
+
+
+class Project(Base):
+    """
+    Projeto/Workspace - agrupa entidades e relacionamentos por investigação
+    """
+    __tablename__ = "projects"
+    
+    id = Column(String(36), primary_key=True, default=generate_uuid)
+    name = Column(String(255), nullable=False)
+    description = Column(Text, nullable=True)
+    color = Column(String(7), default="#00d4ff")  # Hex color for UI
+    icon = Column(String(50), default="folder")  # Icon name
+    
+    # Timestamps
+    created_at = Column(DateTime, default=datetime.utcnow)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)