Update app/main.py

app/main.py

@@ -1,17 +1,52 @@
-from fastapi import FastAPI
-from pydantic import BaseModel
-from opentrons.simulate import simulate, format_runlog, get_protocol_api
-import os
 import io
-import subprocess
+import os
+import traceback
+
+from fastapi import FastAPI, HTTPException, status
+from fastapi.concurrency import run_in_threadpool
+from pydantic import BaseModel
+from opentrons.simulate import simulate, format_runlog
+
 
 app = FastAPI()
 
 
+class Protocol(BaseModel):
+    name: str
+    content: str
+
 @app.get("/")
 async def root():
-    return {"message": "this is a test api server"}
+    return {"message": "Opentrons simulation API is running."}
 
+# no file store in memory, just simulate the protocol
+@app.post("/simulate")
+async def simulate_protocol(protocol: Protocol):
+    try:
+        protocol_file = io.StringIO(protocol.content)
+        run_log, _ = await run_in_threadpool(
+            simulate,
+            protocol_file=protocol_file,
+            file_name=protocol.name
+        )
+        # return execution log
+        return  {
+            'protocol_name': protocol.name,
+            'run_status': 'success',
+            'run_log': format_runlog(run_log)
+        }
+    except Exception as e:
+        print(f"Error simulating protocol: {e}")
+        
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail={
+                "message": "Error simulating protocol",
+                "error_type": type(e).__name__,
+                "error_details": str(e),
+                "traceback": traceback.format_exc()
+            }
+        )
 
 # @app.get("/protocols")
 # def read_protocols():
@@ -45,11 +80,6 @@ def get_file_names():
     return file_names
 
 
-class Protocol(BaseModel):
-    name: str
-    content: str
-
-
 @app.post("/protocol")
 def upload_protocol(protocol: Protocol):
     file_path = 'storage/' + protocol.name