diff --git a/docs/images/architecture.png b/docs/docs/images/architecture.png
similarity index 100%
rename from docs/images/architecture.png
rename to docs/docs/images/architecture.png
diff --git a/docs/images/chameleon.mp4 b/docs/docs/images/chameleon.mp4
similarity index 100%
rename from docs/images/chameleon.mp4
rename to docs/docs/images/chameleon.mp4
diff --git a/docs/images/chameleon.webp b/docs/docs/images/chameleon.webp
similarity index 100%
rename from docs/images/chameleon.webp
rename to docs/docs/images/chameleon.webp
diff --git a/docs/images/container.png b/docs/docs/images/container.png
similarity index 100%
rename from docs/images/container.png
rename to docs/docs/images/container.png
diff --git a/docs/images/omi-logo.png b/docs/docs/images/omi-logo.png
similarity index 100%
rename from docs/images/omi-logo.png
rename to docs/docs/images/omi-logo.png
diff --git a/docs/images/omi-symbol.png b/docs/docs/images/omi-symbol.png
similarity index 100%
rename from docs/images/omi-symbol.png
rename to docs/docs/images/omi-symbol.png
diff --git a/docs/index.md b/docs/docs/index.md
similarity index 100%
rename from docs/index.md
rename to docs/docs/index.md
diff --git a/docs/rationale.md b/docs/docs/rationale.md
similarity index 100%
rename from docs/rationale.md
rename to docs/docs/rationale.md
diff --git a/docs/spec.md b/docs/docs/spec.md
similarity index 100%
rename from docs/spec.md
rename to docs/docs/spec.md
diff --git a/docs/docs/stylesheets/extra.css b/docs/docs/stylesheets/extra.css
new file mode 100644
index 0000000..6ebcdca
--- /dev/null
+++ b/docs/docs/stylesheets/extra.css
@@ -0,0 +1,12 @@
+[data-md-color-scheme="omi"] {
+    /* Sets the primary color (i.e., background of top nav) */
+    --md-primary-fg-color:       #11284F;
+  
+    /* Sets the links color */
+    --md-typeset-a-color:        #046aa5;
+      /* --md-typeset-mark-color:     #0092DF;
+      --md-typeset-kbd-color:      #0092DF; */
+  
+    /* Sets the accent color (i.e., links on hover) */
+    --md-accent-fg-color:        #024a74;
+  }
\ No newline at end of file
diff --git a/mkdocs.yml b/docs/mkdocs.yml
similarity index 81%
rename from mkdocs.yml
rename to docs/mkdocs.yml
index d26fc2b..dcc3f35 100644
--- a/mkdocs.yml
+++ b/docs/mkdocs.yml
@@ -6,8 +6,9 @@ nav:
 theme:
     name: material
     palette:
-        primary: orange
-        accent: blue
+        scheme: omi
+        # primary: orange
+        # accent: blue
     logo: images/omi-symbol.png
     favicon: images/omi-symbol.png
     features:
@@ -21,3 +22,5 @@ markdown_extensions:
 repo_url: https://github.com/modzy/openmodelinterface
 repo_name: modzy/openmodelinterface
 edit_uri: edit/main/docs/
+extra_css:
+    - stylesheets/extra.css
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..b896862
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,17 @@
+chassisml
+flask>=2.0.2
+requests>=2.27.1
+kubernetes>=21.7.0
+loguru>=0.6.0
+pytest>=6.2.5
+pylint>=2.12.2
+flake8>=4.0.1
+mkdocs>=1.2.3
+mkdocs-autorefs>=0.3.1
+mkdocs-material>=8.1.6
+mkdocs-material-extensions>=1.0.3
+mkdocstrings>=0.17.0
+mlflow>=1.22.0
+numpy>=1.22.0
+pandas>=1.4.0
+python-dotenv>=0.19.2