Update app.py

app.py

@@ -11,7 +11,6 @@ model = HfApiModel(
 )
 
 class ProjectScopingTool(Tool):
-
     name = "project_scoping_tool"
     description = """
     Handles project scoping by collecting key details such as type, domain, budget, and timeline.
@@ -26,36 +25,30 @@ class ProjectScopingTool(Tool):
             "type": "string",
             "description": "The industry or field related to the project (e.g., 'finance', 'healthcare').",
         },
-        "budget_min"
-: {
-            "type": "number",
+        "budget_min": {
+            "type": "float",
             "description": "The minimum budget allocated for the project.",
         },
         "budget_max": {
-            "type": "number",
+            "type": "float",
             "description": "The maximum budget allocated for the project.",
         },
         "timeline_months": {
-            "type": "number",
+            "type": "int",
             "description": "The expected duration of the project in months.",
         },
     }
 
-    output_type = "string"
+    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.
-        """
-        self.project_details = {
+        project_details = {
             "type": project_type,
             "domain": domain,
-
             "budget_range": {"min": budget_min, "max": budget_max},
             "timeline": {"months": timeline_months}
         }
-        return json.dumps(self.project_details, indent=2)
+        return json.dumps(project_details, indent=2)
 
 class TechnicalArchitectureTool(Tool):
     name = "technical_architecture_tool"
@@ -70,10 +63,9 @@ class TechnicalArchitectureTool(Tool):
         }
     }
 
-    output_type = "string"
+    output_type = "json"
 
     def forward(self, project_type: str):
-
         architectures = {
             "web": {
                 "frontend": ["React", "Next.js", "Tailwind"],
@@ -101,19 +93,18 @@ class CostEstimationTool(Tool):
 
     inputs = {
         "architecture_size": {
-            "type": "number",
+            "type": "int",
             "description": "The estimated complexity of the architecture on a scale from 1 to 10.",
         },
         "timeline_months": {
-            "type": "number",
+            "type": "int",
             "description": "The project duration in months.",
         }
     }
 
-    output_type = "string"
+    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),
@@ -130,10 +121,9 @@ class DeploymentTool(Tool):
 
     inputs = {}
 
-    output_type = "string"
+    output_type = "json"
 
     def forward(self):
-
         return json.dumps({
             "container_strategy": "Docker + Kubernetes",
             "cloud_provider": "AWS",
@@ -154,10 +144,9 @@ class MeetingPreparationTool(Tool):
         }
     }
 
-    output_type = "string"
+    output_type = "json"
 
     def forward(self, project_stage: str):
-
         agendas = {
             "initial_discovery": [
                 "Project Vision Validation",
@@ -173,24 +162,18 @@ class MeetingPreparationTool(Tool):
         return json.dumps(agendas.get(project_stage, "No agenda found for this stage"), indent=2)
 
 # Instantiate tools
-scoping_tool = ProjectScopingTool()
-tech_arch_tool = TechnicalArchitectureTool()
-cost_tool = CostEstimationTool()
-deploy_tool = DeploymentTool()
-meeting_tool = MeetingPreparationTool()
-
-# Define the agent
 tools = [
-    scoping_tool,
-    tech_arch_tool,
-    cost_tool,
-    deploy_tool,
-    meeting_tool
+    ProjectScopingTool(),
+    TechnicalArchitectureTool(),
+    CostEstimationTool(),
+    DeploymentTool(),
+    MeetingPreparationTool()
 ]
 
+# Define the agent
 agent = CodeAgent(
     model=model,
-    tools=[tools], # add your tools here (don't remove final_answer)
+    tools=tools,  # Fixed tools list format
     max_steps=6,
     verbosity_level=1,
     name="webchainai",
@@ -217,11 +200,11 @@ with gr.Blocks() as demo:
     meeting_output = gr.Textbox(label="Meeting Agenda", interactive=False)
 
     def process_inputs(p_type, dom, min_b, max_b, months):
-        project_details = scoping_tool.collect_project_details(p_type, dom, min_b, max_b, months)
-        architecture = tech_arch_tool.generate_architecture(p_type)
-        estimated_cost = cost_tool.estimate_project_cost(len(architecture), months)
-        deployment_strategy = deploy_tool.generate_deployment_strategy()
-        meeting_agenda = meeting_tool.generate_meeting_agenda("initial_discovery")
+        project_details = tools[0].forward(p_type, dom, min_b, max_b, months)
+        architecture = tools[1].forward(p_type)
+        estimated_cost = tools[2].forward(5, months)  # Example architecture size
+        deployment_strategy = tools[3].forward()
+        meeting_agenda = tools[4].forward("initial_discovery")
         return project_details, architecture, estimated_cost, deployment_strategy, meeting_agenda
 
     process_button = gr.Button("Generate Estimates")
@@ -229,5 +212,4 @@ with gr.Blocks() as demo:
                          outputs=[project_output, arch_output, cost_output, deploy_output, meeting_output])
 
 # Launch the Gradio App
-demo(agent).launch()
-
+demo.launch()