Update main.py

main.py

@@ -4,7 +4,8 @@ import os
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf"
 os.environ["HF_HOME"] = "/tmp/hf"
 
-from langchain.llms import HuggingFacePipeline
+from langchain_community.llms import HuggingFacePipeline
+from langchain.agents import initialize_agent
 from langchain.agents.agent_types import AgentType
 from langchain.tools import BaseTool
 from transformers import pipeline
@@ -33,36 +34,30 @@ class MachineryReportTool(BaseTool):
     def _run(self, query: str) -> str:
         report = """
         ## Comprehensive Report on Mini Construction Equipment Project
-
         ### Overview
         The project involves designing and building construction machinery tailored for the local South African market. 
         The focus is on developing cost-effective, high-performance machines that can compete with expensive American-made 
         equipment while leveraging local manufacturing strengths.
-
         ### 1. Equipment Details
         - **Basic Gas-Powered Auction Unit**
           - Price: $3,700
           - Engine: 14 horsepower gas motor
           - Configuration: Two pump system
           - Capabilities: Digging, Scooping, Auger Operation, Trenching, Pallet Fork Operation
-
         - **DRT 450**
           - Original Price: $8,500 (current base price: $5,500-6,300)
           - Engine: Twin cylinder Honda gas motor
           - Pump System: Triple pump configuration
           - Compatible Attachments: Mulcher, Brush Cutter, Harley Rake, Trencher
-
         - **Mini Skid Steer (New Acquisition)**
           - Price: $15,000
           - Engine: Kubota diesel
           - Control: Pilot controls with vertical lift arms
           - Notable Feature: Mulcher attachment ($2,000)
-
         - **Mini Excavators (Kimron Units)**
           - Unit 1: One-ton unit with Briggs & Stratton engine
           - Unit 2: 3,500 lb unit with Yanmar diesel engine
           - Supplier: K&R Equipment (Oklahoma)
-
         ### 2. Market Comparison and Cost Impact
         - **American Equivalents:** Bobcat MT100, Kubota SCCL1000, Ditch Witch SK900 ($38,000 to $45,000+).
         - **Tariff Impact:** 
@@ -70,7 +65,6 @@ class MachineryReportTool(BaseTool):
           - Mini Skid Steer: ~$9,300 + ~$5,000 shipping + 10% tariff (~$930) = ~$16,000.
         - **Competitive Advantage:** 
           Even with tariffs, Chinese equipment can be 25-30% the cost of American machines.
-
         ### 3. Designing and Building Locally in South Africa
         - **Local Manufacturing Advantages:**
           - "Made in South Africa" branding for local pride.
@@ -83,13 +77,11 @@ class MachineryReportTool(BaseTool):
           - Pilot production, gather feedback, scale.
           - Target local contractors/farmers.
           - Check government incentives for local manufacturing.
-
         ### 4. Next Steps & Recommendations
         - **R&D:** Prototypes + real-world testing.
         - **Supply Chain:** Local suppliers, modular designs.
         - **Financing:** Bank loans, grants, investor funding.
         - **Compliance:** Prepare for certifications and warranties.
-
         ### Conclusion
         Building "Made in South Africa" machinery meets a real need for affordable, durable equipment, 
         potentially disrupting a market dominated by high-priced American brands.
@@ -119,4 +111,4 @@ query = (
 
 result = agent.run(query)
 print("\n===== AGENT OUTPUT =====")
-print(result)
+print(result)