address feedback

NeonGeckoCom · Dec 22, 2023 · c83c332 · c83c332
1 parent cba09bb
commit c83c332
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 66 deletions.
diff --git a/.github/workflows/publish_test_websat_build.yml b/.github/workflows/publish_test_websat_build.yml
diff --git a/Dockerfile b/Dockerfile
@@ -40,7 +40,7 @@ COPY docker_overlay/ /
 # Expose port 8000 for websat
 EXPOSE 8000
 
-ENTRYPOINT ["iris"]
+# Set the ARG value as an environment variable
+ENV EXTRAS=${EXTRAS}
 
-# Default command
-CMD ["-h"]
+ENTRYPOINT ["/neon_iris/entrypoint.sh"]
diff --git a/README.md b/README.md
@@ -126,7 +126,7 @@ iris:
   webui_title: Neon AI
   webui_description: Chat with Neon
   webui_input_placeholder: Ask me something
-  webui_ws_url: wss://neonaialpha.com:8000/ws
+  webui_ws_url: wss://neonaialpha.com/ws
 ```
 
 ### Customization

diff --git a/entrypoint.sh b/entrypoint.sh
@@ -0,0 +1,11 @@
+#!/bin/bash
+set -e
+
+if [ "$EXTRAS" = "gradio" ]; then
+    exec iris start-gradio
+    elif [ "$EXTRAS" = "web_sat" ]; then
+    exec iris start-websat
+else
+    echo "No extras specified, showing help. To execute a command, use 'docker run iris <command>'"
+    exec iris -h
+fi