From c72958fb70fddfe00dce1841d1da70a7453ba3f0 Mon Sep 17 00:00:00 2001
From: Kumaran Rajendhiran <kumaran@airt.ai>
Date: Thu, 21 Nov 2024 13:35:51 +0530
Subject: [PATCH] Polish deployment related issues

---
 {{cookiecutter.project_slug}}/fly.toml                       | 2 +-
 ...kiecutter.app_type %}nats{% else %}fastapi{% endif %}.py" | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/{{cookiecutter.project_slug}}/fly.toml b/{{cookiecutter.project_slug}}/fly.toml
index efa646e..a7a7287 100644
--- a/{{cookiecutter.project_slug}}/fly.toml
+++ b/{{cookiecutter.project_slug}}/fly.toml
@@ -10,7 +10,7 @@ primary_region = 'ams'
   dockerfile = 'docker/Dockerfile'
 
 [http_service]
-  internal_port = 8888
+  internal_port = {% if cookiecutter.app_type == 'fastapi' %}8008{% else %}8888{% endif %}
   force_https = true
   auto_stop_machines = 'stop'
   auto_start_machines = true
diff --git "a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/deployment/main_1_{% if \"nats\" in cookiecutter.app_type %}nats{% else %}fastapi{% endif %}.py" "b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/deployment/main_1_{% if \"nats\" in cookiecutter.app_type %}nats{% else %}fastapi{% endif %}.py"
index 282c942..d3b17eb 100644
--- "a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/deployment/main_1_{% if \"nats\" in cookiecutter.app_type %}nats{% else %}fastapi{% endif %}.py"	
+++ "b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/deployment/main_1_{% if \"nats\" in cookiecutter.app_type %}nats{% else %}fastapi{% endif %}.py"	
@@ -38,8 +38,8 @@
       <div id="workflows"></div>
       <ul id="messages"></ul>
       <script>
-         const API_URL = 'http://127.0.0.1:8008/fastagency';
-         const WS_URL = 'ws://127.0.0.1:8008/fastagency/ws'; // nosemgrep
+         const API_URL = location.protocol + '//' + location.host + '/fastagency';// Use wss:// if location.protocol is https else use ws://
+         const WS_URL = (location.protocol === 'https:' ? 'wss://' : 'ws://') + location.host + '/fastagency/ws'; // nosemgrep
          let socket;
 
          async function fetchWorkflows() {
@@ -115,6 +115,7 @@
 """
 
 
+# Optional basic HTML page to interact with the workflows
 @app.get("/")
 async def get() -> HTMLResponse:
     return HTMLResponse(html)