testing v-0.0.5

app.py

@@ -4,7 +4,7 @@ import streamlit as st
 import os
 import stat
 import socket
-# import time
+import time
 
 # Get current permissions and add the execute bit for the owner
 current_permissions = os.stat("init.sh").st_mode
@@ -38,6 +38,11 @@ if env_var not in os.environ:
         print(result.stdout)
         # st.write("Script output:")
         # st.write(result.stdout)
+        while not is_port_in_use(host, port):
+            print(f"Not listening on port: {host}:{port}")
+            print("Waiting 10 seconds before retrying...")
+            time.sleep(10)
+        print(f"Listening on port: {host}:{port}")
     except subprocess.CalledProcessError as e:
         print("An error occurred:")
         print(e.stderr)

init.sh

@@ -40,5 +40,5 @@ cd build
 
 echo "llama-server launched. Waiting for the server to initialize..."
 # (Optional) Wait a few seconds for the server to start before launching Streamlit
-sleep 15
+sleep 45
 echo "Initialization complete. Proceeding with Streamlit app startup..."

init2.sh

@@ -17,7 +17,7 @@ if [ -d "llama.cpp" ]; then
 
         echo "llama-server launched. Waiting for the server to initialize..."
         # (Optional) Wait a few seconds for the server to start before launching Streamlit
-        sleep 15
+        sleep 45
         echo "Initialization complete. Proceeding with Streamlit app startup..."
     else
         echo "Directory 'build' not found inside 'llama.cpp'"
@@ -41,7 +41,7 @@ elif [ "$(basename "$PWD")" = "llama.cpp" ]; then
 
         echo "llama-server launched. Waiting for the server to initialize..."
         # (Optional) Wait a few seconds for the server to start before launching Streamlit
-        sleep 15
+        sleep 45
         echo "Initialization complete. Proceeding with Streamlit app startup..."
     else
         echo "Directory 'build' not found in the current 'llama.cpp' folder"