fixes 6

Dockerfile

@@ -22,6 +22,11 @@ RUN mkdir -p /app/models && \
 
 COPY . .
 
+RUN mkdir -p /app/database
+RUN useradd -m -u 1000 user && chown -R user:user /app
+
+USER user
+
 RUN python init_db.py
 
 EXPOSE 7860

app.py

@@ -23,9 +23,11 @@ app.secret_key = os.getenv("FLASK_SECRET_KEY")
 # ------------------------------------------------------------------
 
 DATABASE_URL = os.getenv("DB_URL")
-
 engine = create_engine(DATABASE_URL)
 
+with engine.connect() as startup_conn:
+    startup_conn.execute(text("SELECT 1;"))
+
 registry = extract(engine)
 schema_context = registry.to_prompt_context()
 

nl_to_sql.py

@@ -3,14 +3,14 @@ import re
 from llama_cpp import Llama
 # from datetime import datetime
 
-# LOG_FILE= "app.log"
+LOG_FILE= "app.log"
 
 # Load the model directly into memory from the local container path
 llm = Llama(
     model_path="/app/models/qwen.gguf",
     n_ctx=2048,
     verbose=False,
-    n_threads=2
+    n_threads=4
 )
 
 def clean_llm_sql(sql: str) -> str: