Update main.py

main.py

@@ -0,0 +1,111 @@
+import os
+from langchain.agents import initialize_agent
+from langchain.llms import HuggingFacePipeline
+from langchain.tools import BaseTool
+from transformers import pipeline
+
+# Set up the Hugging Face pipeline to run on CPU (device=-1 ensures CPU usage)
+pipe = pipeline(
+    "text-generation",
+    model="bigscience/bloom",       # You can change to another CPU-friendly model if desired
+    tokenizer="bigscience/bloom",
+    device=-1,                      # -1 forces CPU mode
+    max_new_tokens=512
+)
+
+# Initialize the LangChain LLM with our Hugging Face pipeline
+llm = HuggingFacePipeline(pipeline=pipe)
+
+# Define a custom tool that compiles the research report
+class MachineryReportTool(BaseTool):
+    name = "machinery_report"
+    description = (
+        "This tool compiles a detailed report on the mini construction equipment project. "
+        "It includes specifications, market analysis, and production considerations for each model."
+    )
+    
+    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: Capability to add a 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, and Ditch Witch SK900 are priced from $38,000 to $45,000+.
+        - **Tariff Impact:**
+          - Example for DRT 450: Base price ~$6,300, plus a 25% tariff and an additional 10% (approx. $630) brings the final price to ~$9,000.
+          - Mini Skid Steer: Base price ~$9,300, shipping ~$5,000, and an additional 10% tariff adds ~$930, leading to a total of ~$16,000.
+        - **Competitive Advantage:**
+          - Even with tariffs, the Chinese-engineered equipment is available at 25-30% of the cost of American machines, offering higher ROI potential for small businesses.
+
+        ### 3. Designing and Building Locally in South Africa
+        - **Local Manufacturing Advantages:**
+          - Leveraging “Made in South Africa” branding appeals to local pride and supports regional economies.
+          - Potential for custom designs tailored to local farming and construction needs.
+        - **Production Considerations:**
+          - Consider assembling components locally to reduce tariff impacts.
+          - Explore partnerships with local engineering firms and manufacturing plants.
+          - Ensure compliance with South African standards and, if planning for export, international standards.
+        - **Market Entry Strategies:**
+          - Begin with a pilot production line, gather real-world feedback, and scale based on demand.
+          - Use targeted marketing to local contractors and farmers.
+          - Investigate government incentives for local manufacturing and innovation.
+
+        ### 4. Next Steps & Recommendations
+        - **R&D:** Develop prototypes and conduct market tests with local stakeholders.
+        - **Supply Chain:** Identify reliable local suppliers and consider modular designs for easier customization.
+        - **Financing:** Look into bank loans, government grants, or investor funding to support the initial build.
+        - **Compliance:** Prepare for necessary certifications and warranty support structures.
+        
+        ### Conclusion
+        Building your own machinery under a “Made in South Africa” banner not only differentiates your product but also meets a market need for cost-effective, durable equipment. With the right strategy, this approach could disrupt a market currently dominated by high-priced American equipment.
+        """
+        return report
+
+    def _arun(self, query: str) -> str:
+        raise NotImplementedError("Async method not implemented.")
+
+# Add the custom tool to the list of tools for the agent
+tools = [MachineryReportTool()]
+
+# Initialize the agent using a Zero-Shot approach (React-style)
+agent = initialize_agent(
+    tools, llm, agent="zero-shot-react-description", verbose=True
+)
+
+# Define your query to compile the detailed report
+query = (
+    "Using the available tools, please compile a detailed research report on the "
+    "mini construction equipment project, including design, market analysis, and production "
+    "specifications based on the provided machinery details."
+)
+
+# Run the agent and print the output
+result = agent.run(query)
+print(result)