diff --git a/python/packages/autogen-studio/autogenstudio/cli.py b/python/packages/autogen-studio/autogenstudio/cli.py
index 46299b86f..8bf27915c 100644
--- a/python/packages/autogen-studio/autogenstudio/cli.py
+++ b/python/packages/autogen-studio/autogenstudio/cli.py
@@ -1,5 +1,4 @@
 import os
-import tempfile
 import warnings
 from typing import Optional
 
@@ -91,6 +90,7 @@ def serve(
     host: str = "127.0.0.1",
     port: int = 8084,
     workers: int = 1,
+    reload: Annotated[bool, typer.Option("--reload")] = False,
     docs: bool = False,
 ):
     """
@@ -118,7 +118,7 @@ def serve(
         host=host,
         port=port,
         workers=workers,
-        reload=False,
+        reload=reload,
     )
 
 
diff --git a/python/packages/autogen-studio/autogenstudio/web/initialization.py b/python/packages/autogen-studio/autogenstudio/web/initialization.py
index e7830f761..b3be95e8c 100644
--- a/python/packages/autogen-studio/autogenstudio/web/initialization.py
+++ b/python/packages/autogen-studio/autogenstudio/web/initialization.py
@@ -1,7 +1,6 @@
 # api/initialization.py
 import os
 from pathlib import Path
-from typing import Dict
 
 from dotenv import load_dotenv
 from loguru import logger
diff --git a/python/packages/autogen-studio/autogenstudio/web/serve.py b/python/packages/autogen-studio/autogenstudio/web/serve.py
index 35f5f8424..8dbbf1823 100644
--- a/python/packages/autogen-studio/autogenstudio/web/serve.py
+++ b/python/packages/autogen-studio/autogenstudio/web/serve.py
@@ -1,7 +1,6 @@
 import os
-from typing import cast
 
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI
 
 from ..datamodel import Response
 from ..teammanager import TeamManager