Create logistics.py

playground/demos/logistics/logistics.py

@@ -0,0 +1,67 @@
+from swarms.structs import Agent
+import os
+from dotenv import load_dotenv
+from swarms.models import GPT4VisionAPI
+from swarms.prompts.logistics import (
+    Health_Security_Agent_Prompt,
+    Quality_Control_Agent_Prompt,
+    Productivity_Agent_Prompt,
+    Safety_Agent_Prompt,
+    Security_Agent_Prompt,
+    Sustainability_Agent_Prompt,
+    Efficiency_Agent_Prompt
+)
+
+load_dotenv()
+api_key = os.getenv("OPENAI_API_KEY")
+
+llm = GPT4VisionAPI(openai_api_key=api_key)
+
+# Image for analysis
+factory_image = "factory_image1.jpg"
+
+# Initialize agents with respective prompts
+health_security_agent = Agent(
+    llm=llm, sop=Health_Security_Agent_Prompt, max_loops=3, multi_modal=True
+)
+quality_control_agent = Agent(
+    llm=llm, sop=Quality_Control_Agent_Prompt, max_loops=3, multi_modal=True
+)
+productivity_agent = Agent(
+    llm=llm, sop=Productivity_Agent_Prompt, max_loops=3, multi_modal=True
+)
+safety_agent = Agent(
+    llm=llm, sop=Safety_Agent_Prompt, max_loops=3, multi_modal=True
+)
+security_agent = Agent(
+    llm=llm, sop=Security_Agent_Prompt, max_loops=3, multi_modal=True
+)
+sustainability_agent = Agent(
+    llm=llm, sop=Sustainability_Agent_Prompt, max_loops=3, multi_modal=True
+)
+efficiency_agent = Agent(
+    llm=llm, sop=Efficiency_Agent_Prompt, max_loops=3, multi_modal=True
+)
+
+# Run agents with respective tasks on the same image
+health_analysis = health_security_agent.run(
+    "Analyze the safety of this factory", factory_image
+)
+quality_analysis = quality_control_agent.run(
+    "Examine product quality in the factory", factory_image
+)
+productivity_analysis = productivity_agent.run(
+    "Evaluate factory productivity", factory_image
+)
+safety_analysis = safety_agent.run(
+    "Inspect the factory's adherence to safety standards", factory_image
+)
+security_analysis = security_agent.run(
+    "Assess the factory's security measures and systems", factory_image
+)
+sustainability_analysis = sustainability_agent.run(
+    "Examine the factory's sustainability practices", factory_image
+)
+efficiency_analysis = efficiency_agent.run(
+    "Analyze the efficiency of the factory's manufacturing process", factory_image
+)