Update app.py

app.py

@@ -10,22 +10,39 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-
 class ProjectScopingTool(Tool):
     name = "project_scoping_tool"
     description = """
     Handles project scoping by collecting key details such as type, domain, budget, and timeline.
     """
-    def __init__(self):
-        self.project_details = {
-            "type": None,
-            "domain": None,
-            "technical_requirements": {},
-            "budget_range": {},
-            "timeline": {}
-        }
-    
-    def collect_project_details(self, project_type, domain, budget_min, budget_max, timeline_months):
+
+    inputs = {
+        "project_type": {
+            "type": "string",
+            "description": "The type of project (e.g., 'web', 'blockchain', 'ai').",
+        },
+        "domain": {
+            "type": "string",
+            "description": "The industry or field related to the project (e.g., 'finance', 'healthcare').",
+        },
+        "budget_min": {
+            "type": "float",
+            "description": "The minimum budget allocated for the project.",
+        },
+        "budget_max": {
+            "type": "float",
+            "description": "The maximum budget allocated for the project.",
+        },
+        "timeline_months": {
+            "type": "int",
+            "description": "The expected duration of the project in months.",
+        },
+    }
+
+    output_type = "json"
+
+    def forward(self, project_type: str, domain: str, budget_min: float, budget_max: float, timeline_months: int):
+
         """
         Collects and structures project details from user input.
         """
@@ -37,13 +54,23 @@ class ProjectScopingTool(Tool):
         }
         return json.dumps(self.project_details, indent=2)
 
-
 class TechnicalArchitectureTool(Tool):
     name = "technical_architecture_tool"
     description = """
     Determines the appropriate technical architecture based on the project type.
     """
-    def generate_architecture(self, project_type):
+
+    inputs = {
+        "project_type": {
+            "type": "string",
+            "description": "The type of project (e.g., 'web', 'blockchain', 'ai').",
+        }
+    }
+
+    output_type = "json"
+
+    def forward(self, project_type: str):
+
         architectures = {
             "web": {
                 "frontend": ["React", "Next.js", "Tailwind"],
@@ -63,13 +90,27 @@ class TechnicalArchitectureTool(Tool):
         }
         return json.dumps(architectures.get(project_type, "Invalid project type"), indent=2)
 
-
 class CostEstimationTool(Tool):
     name = "cost_estimation_tool"
     description = """
     Estimates the cost based on the architecture complexity and timeline.
     """
-    def estimate_project_cost(self, architecture_size, timeline_months):
+
+    inputs = {
+        "architecture_size": {
+            "type": "int",
+            "description": "The estimated complexity of the architecture on a scale from 1 to 10.",
+        },
+        "timeline_months": {
+            "type": "int",
+            "description": "The project duration in months.",
+        }
+    }
+
+    output_type = "json"
+
+    def forward(self, architecture_size: int, timeline_months: int):
+
         base_costs = {
             "development": 5000 * (architecture_size * 0.5),
             "infrastructure": 500 * (architecture_size * 0.3),
@@ -78,13 +119,18 @@ class CostEstimationTool(Tool):
         total_cost = sum(base_costs.values())
         return json.dumps({"total_estimated_cost": total_cost, "breakdown": base_costs}, indent=2)
 
-
 class DeploymentTool(Tool):
     name = "deployment_tool"
     description = """
     Suggests a deployment strategy based on best practices.
     """
-    def generate_deployment_strategy(self):
+
+    inputs = {}
+
+    output_type = "json"
+
+    def forward(self):
+
         return json.dumps({
             "container_strategy": "Docker + Kubernetes",
             "cloud_provider": "AWS",
@@ -92,13 +138,23 @@ class DeploymentTool(Tool):
             "ci_cd_pipeline": "GitHub Actions"
         }, indent=2)
 
-
 class MeetingPreparationTool(Tool):
     name = "meeting_preparation_tool"
     description = """
     Provides meeting agendas depending on the project stage.
     """
-    def generate_meeting_agenda(self, project_stage):
+
+    inputs = {
+        "project_stage": {
+            "type": "string",
+            "description": "The current stage of the project (e.g., 'initial_discovery', 'architecture_review').",
+        }
+    }
+
+    output_type = "json"
+
+    def forward(self, project_stage: str):
+
         agendas = {
             "initial_discovery": [
                 "Project Vision Validation",