fixer upper

D2/run_d2.py

@@ -3,17 +3,12 @@
 import asyncio
 import os
 
-current_directory = os.getcwd()
-print(f"Current directory: {current_directory}")
-
-# Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 async def run_go_script(input_data: str):
     try:
         print("Running go script")
-        # Use asyncio's subprocess support to run the Go script
         process = await asyncio.create_subprocess_exec(
             './D2/main', 'encode', input_data,
             stdout=asyncio.subprocess.PIPE,

app.py

@@ -49,11 +49,11 @@ class DiagramRequest(BaseModel):
     lang: str
     type: str
     code: str
-    theme: str = None
+    theme: str = ""
 
     @validator('lang')
     def validate_lang(cls, v):
-        valid_langs = ["plantuml", "mermaid", "mermaidjs" "d2lang", "D2", "d2", "terrastruct", "graphviz"]
+        valid_langs = ["plantuml", "mermaid", "mermaidjs", "d2lang", "D2", "d2", "terrastruct", "graphviz"]
         if v not in valid_langs:
             raise ValidationError(f"Invalid diagram language: {v}")
         return v

test_app.py

@@ -45,6 +45,7 @@ def test_generate_d2_diagram_endpoint():
         "type": "class",
         "code": "class Test{}"
     })
+    assert response.json() == {"error": "An error occurred while generating the diagram."}
     assert response.status_code == 200
     assert "url" in response.json()
 
@@ -130,4 +131,7 @@ async def test_run_go_script():
     test_input = "test input"
 
     # Call the function with the test input
-    await run_go_script(test_input)
+    url, content, playground = await run_go_script(test_input)
+    assert url is not None
+    assert content is not None
+    assert playground is not None