[SAM-260013]: feat: enhance Streamlit app command with port option and update main function to accept port parameter

Author: Heisenberg208

Makefile

@@ -1,5 +1,6 @@
 .PHONY: install lint format type test run api clean
 
+PORT ?= 8501
 install:
 	poetry install
 
@@ -20,7 +21,7 @@ type:
 
 
 dev:
-	poetry run sample dev
+	poetry run sample dev --port $(PORT)
 
 api:
 	poetry run sample api

src/sample/__main__.py

@@ -8,37 +8,33 @@
 from . import __version__
 
 
-def main():
+def main(port: int = 8501):
     """
     Entrypoint for the Streamlit 'dev' app.
     """
     print("🏷️ Sample version:", __version__)
     logging.info("Starting sample dev script...")
 
-    # Paths
     Sample_dir = Path(__file__).resolve().parent
-    dev_root = Sample_dir.parent  # src/
-    wheel_root = Sample_dir.parent  # same in wheel
+    dev_root = Sample_dir.parent
+    wheel_root = Sample_dir.parent
 
-    # Add correct root to sys.path
-    if "site-packages" in str(Sample_dir):  # running from wheel
+    if "site-packages" in str(Sample_dir):
         if str(wheel_root) not in sys.path:
             sys.path.append(str(wheel_root))
             logging.info(f"Added wheel root to sys.path: {wheel_root}")
-    else:  # dev mode
+    else:
         if str(dev_root) not in sys.path:
             sys.path.append(str(dev_root))
             logging.info(f"Added dev src root to sys.path: {dev_root}")
 
-    # Locate streamlit_app.py
     streamlit_app_path = Sample_dir / "streamlit_app.py"
     logging.info(f"Streamlit app path: {streamlit_app_path}")
 
     if not streamlit_app_path.exists():
         logging.error(f"Streamlit app not found at: {streamlit_app_path}")
         return
 
-    # Run Streamlit app
     python_path = sys.executable
     logging.info(f"Using Python executable: {python_path}")
 
@@ -50,11 +46,11 @@ def main():
             "run",
             str(streamlit_app_path.resolve()),
             "--server.port",
-            "8501",
+            str(port),
         ],
         check=True,
     )
 
 
 if __name__ == "__main__":
-    main()
+    main(port=8501)

src/sample/cli.py

@@ -22,10 +22,13 @@ def cli(ctx, version):
 
 
 @cli.command(help="Run the Sample Streamlit app.")
-def dev():
+@click.option(
+    "--port", default=8501, show_default=True, help="Port to run the Streamlit app on."
+)
+def dev(port: int):
     from sample.__main__ import main
 
-    main()
+    main(port)
 
 
 @cli.command(help="Run the Sample FastAPI backend.")