动手搭建 AI 硬件设计平台!本地 LLM 一句话生成完整 Arduino 项目(接线图 + BOM + 代码 + 步骤)

见文末代码二

见文末代码二

见文末代码三

# 1. 安装 Ollama
curl -fsSL https://ollama.com/install.sh | sh
ollama pull qwen2:7b     # 或 llama3.1:8b

# 2. 安装Python依赖
pip install ollama gradio schemdraw sqlite3

# 自建 AI 硬件设计 Agent Mini 版 - 完全本地
import ollama
import gradio as gr

##代码二##

# 简单硬件知识库（可后续扩展为SQLite或向量数据库）
HARDWARE_DB = {
    "arduino_uno": "引脚 9/10/11 支持PWM，LED需串联220Ω电阻限流，避免烧毁",
    "pir_sensor": "输出数字信号，可接任意数字引脚",
    "button": "建议使用INPUT_PULLUP模式，接GND触发",
}

def generate_hardware(prompt):
    system_prompt = """你是一个经验丰富的硬件设计专家。
用户输入需求后，请严格按照以下格式输出：
1. 物料清单 (BOM)，包含估算成本
2. 详细文字接线说明
3. 完整、可直接上传的Arduino代码
4. 安全注意事项
严格遵守电气安全规则，不输出危险或不可靠的方案。"""

    response = ollama.chat(model='qwen2', messages=[
        {'role': 'system', 'content': system_prompt},
        {'role': 'user', 'content': prompt + "\n参考硬件知识：" + str(HARDWARE_DB)}
    ])
    return response['message']['content']

# Gradio聊天界面
iface = gr.Interface(
    fn=generate_hardware,
    inputs=gr.Textbox(lines=2, placeholder="例如：用Arduino Uno做一个带按钮的红黄绿交通灯"),
    outputs=gr.Markdown(),
    title="我的本地硬件AI设计平台",
    description="输入需求，一键生成接线图、BOM、代码和步骤（纯本地运行）"
)
iface.launch()

##代码三##

// Arduino 交通灯 - 本地AI平台生成代码
const int redPin = 9, yellowPin = 10, greenPin = 11, buttonPin = 2;
int mode = 0;

void setup() {
  pinMode(redPin, OUTPUT);
  pinMode(yellowPin, OUTPUT);
  pinMode(greenPin, OUTPUT);
  pinMode(buttonPin, INPUT_PULLUP);
  Serial.begin(9600);
  Serial.println("交通灯启动！按按钮切换模式，或通过串口输入1/2/3手动控制");
}

void loop() {
  if (digitalRead(buttonPin) == LOW) {
    mode = !mode;
    delay(300);
  }
  if (mode == 0) {
    // 正常交通灯循环
    digitalWrite(greenPin, HIGH); delay(5000); digitalWrite(greenPin, LOW);
    digitalWrite(yellowPin, HIGH); delay(2000); digitalWrite(yellowPin, LOW);
    digitalWrite(redPin, HIGH); delay(5000); digitalWrite(redPin, LOW);
  } else if (Serial.available()) {
    char cmd = Serial.read();
    digitalWrite(redPin, cmd=='3');
    digitalWrite(yellowPin, cmd=='2');
    digitalWrite(greenPin, cmd=='1');
  }
}