Overhauls

.gitignore

@@ -8,3 +8,5 @@ wheels/
 
 # Virtual environments
 .venv
+
+.env.local

Dockerfile

@@ -4,6 +4,8 @@ RUN apk add curl python3
 
 RUN curl -fsSL https://raw.githubusercontent.com/arduino/arduino-cli/master/install.sh | sh
 
+RUN arduino-cli core install arduino:avr
+
 RUN adduser -u 1000 -D user
 
 USER user
@@ -11,6 +13,7 @@ USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin/:$PATH
 
+
 WORKDIR $HOME/app
 
 COPY --chown=user pyproject.toml .
@@ -19,9 +22,9 @@ RUN uv sync
 
 COPY --chown=user *.py .
 
-ENV GOOGLE_GENAI_USE_VERTEXAI=TRUE
-ENV GOOGLE_CLOUD_LOCATION=us-central1
-ENV GOOGLE_CLOUD_PROJECT=vibecoder-459110
+ENV GOOGLE_GENAI_USE_VERTEXAI=FALSE
+# ENV GOOGLE_CLOUD_LOCATION=us-central1
+# ENV GOOGLE_CLOUD_PROJECT=vibecoder-459110
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 
 CMD ["uv", "run", "main.py"]

agent.py

@@ -4,8 +4,6 @@ from google.adk.runners import Runner
 from google.adk.tools import (
     ToolContext,
     FunctionTool,
-    built_in_code_execution,
-    google_search,
 )
 from google.genai import types
 import asyncio
@@ -13,18 +11,19 @@ import dotenv
 import os
 import subprocess
 import shutil
+import tempfile
 
 dotenv.load_dotenv()
 
 
-def read_code(tool_context: ToolContext, start_line: int = 0, end_line: int = 9999999):
+def read_code(tool_context: ToolContext, start_line: int, end_line: int):
     """
     Read the code written so far. The code is arduino code.
     You can optionally select which part of the code you would like to get.
 
     Args:
-        start_line (int): The first line of the code part you want (defaults to 0)
-        end_line (int): The last line of the code part you want (defaults to infinity)
+        start_line (int): The first line of the code part you want
+        end_line (int): The last line of the code part you want
 
     Returns:
         str: The (part of the) code written so far.
@@ -304,28 +303,34 @@ def compile_with_arduino(
 
     code = tool_context.state.get("code")
     if not code:
-        return {"error": "No code to compile"}
-
+        return {
+            "success": False,
+            "error": "No code to compile"}
+    
 
-    shutil.rmtree('/tmp/arduino_project')
-    os.mkdir('/tmp/arduino_project')    
+    project_dir = tempfile.mkdtemp()
+    project_dir = f"{project_dir}/arduino_project"
+    os.makedirs(project_dir, exist_ok=True)
 
-    with open("/tmp/arduino_project/arduino_project.ino", "w") as f:
+    with open(f"{project_dir}/arduino_project.ino", "w") as f:
         f.write(code)
 
     # Compile with the Arduino CLI
     result = subprocess.run(
-        ["arduino-cli", "compile", "--fqbn", "arduino:avr:uno", "/tmp/arduino_project", "--build-path", "/tmp/arduino_project/output"],
+        ["arduino-cli", "compile", "--fqbn", "arduino:avr:uno", project_dir, "--build-path", f"{project_dir}/output"],
         capture_output=True,
     )
     if result.returncode != 0:
+        print(f"Compilation failed: {result.stderr.decode('utf-8')}")
         return {
+            "success": False,
             "error": "Compilation failed",
             "output": result.stderr.decode("utf-8"),
         }
     return {
         "success": True,
         "output": result.stdout.decode("utf-8"),
+        "project_dir": project_dir,
     }
 
 def search_library(

main.py

@@ -127,21 +127,11 @@ async def interact_with_agent(
             yield messages, None, None, None
         for call in event.get_function_responses():
             if call.name == "compile_with_arduino" and call.response["success"]:
-                temp = tempfile.gettempdir()
-
-                temp_uuid = str(uuid4())
-                temp = os.path.join(temp, f"{temp_uuid}.ino.elf")
-
-
-                with open(temp, "wb") as f:
-                    with open(
-                        "/tmp/arduino_project/output/arduino_project.ino.elf", "rb"
-                    ) as f2:
-                        f.write(f2.read())
+                path = call.response["project_dir"]
 
                 yield messages, None, None, gr.DownloadButton(
                     "Download ELF",
-                    temp,
+                    os.path.join(path, "output", "arduino_project.ino.elf"),
                     variant="primary",
                     render=False,
                     interactive=True,