More components

agent.py

@@ -126,6 +126,55 @@ COMPONENT_INFO_DICT = {
             "ka": "Kathode",
         },
     },
+    "DHT22": {
+        "pins": {
+            "VCC": "Positive voltage",
+            "SDA": "Digital data pin (input/output)",
+            "NC": "Not connected",
+            "GND": "Ground",
+        }
+    },
+    "HC-SR04": {
+        "pins": {
+            "VCC": "Supply Voltage (5V)",
+            "TRIG": "Trigger Pin (input)",
+            "ECHO": "Echo Pin (output)",
+            "GND": "Ground",
+        }
+    },
+    "DS1307": {
+        "pins": {
+            "5V": "Supply Voltage (5V)",
+            "SCL": "I2C Clock Line",
+            "SDA": "I2C Data Line",
+            "GND": "Ground",
+        }
+    },
+    "MPU6050": {
+        "pins": {
+            "VCC": "Supply Voltage (3.3V or 5V)",
+            "SCL": "I2C Clock Line",
+            "SDA": "I2C Data Line",
+            "GND": "Ground",
+            "AD0": "Address select pin (optional)",
+            "INT": "Interrupt pin (optional)",
+        }
+    },
+    "Slide Switch": {
+        "pins": {
+            "one": "Left terminal",
+            "two": "Common terminal",
+            "three": "Right terminal"
+        }
+    },
+    "RGB LED": {
+        "pins": {
+            "R": "Red anode",
+            "G": "Green anode",
+            "B": "Blue anode",
+            "GND": "Common cathode"
+        }
+    }
 }
 
 

main.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from agent import session_service, root_agent, APP_NAME, runner
+from agent import session_service, root_agent, APP_NAME, runner, COMPONENT_INFO_DICT
 from google.genai import types
 from pprint import pprint, pformat
 from uuid import uuid4
@@ -175,11 +175,19 @@ with gr.Blocks() as demo:
         interactive=False,
         render=False,
     )
+    COMPONENT_NAMES_MD = "\n".join([f'- {name}' for name in COMPONENT_INFO_DICT.keys()])
 
     with gr.Row():
         gr.Markdown(
-            """
-# Arduino Hardware and Software
+            f"""
+# Arduino Hardware and Software Vibecoder
+
+This is a prototype of a vibecoder (like Lovable) that can generate Arduino code and schematics for you.
+Given a prompt, it well look up the components it has availble and generate a high-level schematic and code.
+It compiles the code to make sure it works and then gives you a download link for the ELF file that can then be uploaded to the Arduino.
+
+It has the following components:
+{COMPONENT_NAMES_MD}
                     """
         )
     with gr.Row():