Update utils.py

utils.py

@@ -1,3 +1,4 @@
+import os
 import io
 
 import librosa
@@ -57,15 +58,19 @@ def bytes_to_array(audio_bytes):
     )
     return audio_array
 
-def start_server(ssh_key, dns_name):
-    pkey = paramiko.RSAKey.from_private_key(io.StringIO(ssh_key))
+def start_server():
+    if os.getenv('TUNNEL_STATUS') == "200":
+        return
+    
+    pkey = paramiko.RSAKey.from_private_key(io.StringIO(os.getenv('PRIVATE_KEY')))
 
     server = SSHTunnelForwarder(
-        ssh_address_or_host=dns_name,
+        ssh_address_or_host=os.getenv('SERVER_DNS_NAME'),
         ssh_username="ec2-user",
         ssh_pkey=pkey,  
-        local_bind_address=("127.0.0.1", local_port),
+        local_bind_address=("127.0.0.1", int(os.getenv('LOCAL_PORT'))),
         remote_bind_address=("127.0.0.1", 8000)
     )
-
+    
+    os.environ['TUNNEL_STATUS'] = "200"
     server.start()