Fastapi refactor update

app.py

@@ -1,30 +1,25 @@
-import os
-
-import click
-import uvicorn
-from dotenv import load_dotenv
-
-
-@click.command()
-@click.option(
-    "--env",
-    type=click.Choice(["local", "dev", "prod"], case_sensitive=False),
-    default="local",
-)
-def main(env: str):
-    load_dotenv()
-    os.environ["ENV"] = env
-    if env == "prod":
-        raise Exception("Production environment not supported yet!")
-
-    uvicorn.run(
-        app="app.server:app",
-        host="localhost",
-        port=8000,
-        reload=True if env != "prod" else False,
-        workers=1,
-    )
-
-
-if __name__ == "__main__":
-    main()
+import click
+import uvicorn
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+@click.command()
+@click.option(
+    "--env",
+    type=click.Choice(["local", "staging", "production"], case_sensitive=False),
+    default="staging",
+)
+def main(env: str):
+    uvicorn.run(
+        app="app.server:app",
+        host="localhost",
+        port=8000,
+        reload=True if env != "production" else False,
+        workers=1,
+    )
+
+
+if __name__ == "__main__":
+    main()