Update app.py via AI Editor

app.py

@@ -19,7 +19,6 @@ from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores import Chroma
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 
-# ========== GLOBALS AND LOGGING ==========
 logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(threadName)s %(message)s")
 logger = logging.getLogger("AskTricare")
 
@@ -35,7 +34,6 @@ os.makedirs(VECTOR_DB_DIR, exist_ok=True)
 
 openai.api_key = os.environ.get("OPENAI_API_KEY")
 
-# ========== VECTOR DB SHARED ==========
 chroma_client = chromadb.Client(Settings(
     chroma_db_impl="duckdb+parquet",
     persist_directory=VECTOR_DB_DIR,
@@ -87,7 +85,6 @@ vectordb = Chroma(
     client_settings=Settings(chroma_db_impl="duckdb+parquet", persist_directory=VECTOR_DB_DIR),
 )
 
-# ========== SESSION MANAGEMENT ==========
 def get_session_id():
     sid = flask_request.cookies.get("asktricare_session_id")
     if not sid:
@@ -126,7 +123,15 @@ def load_session_state(session_id):
         with open(path, "r") as f:
             SESSION_DATA[session_id] = json.load(f)
 
-# ========== APP SETUP ==========
+def load_system_prompt():
+    prompt_path = os.path.join(os.getcwd(), "system_prompt.txt")
+    try:
+        with open(prompt_path, "r", encoding="utf-8") as f:
+            return f.read().strip()
+    except Exception as e:
+        logger.error(f"Failed to load system prompt: {e}")
+        return "You are Ask Tricare, a helpful assistant for TRICARE health benefits. Respond conversationally, and cite relevant sources when possible. If you do not know, say so."
+
 app = dash.Dash(
     __name__,
     server=app_flask,
@@ -135,7 +140,6 @@ app = dash.Dash(
     update_title="Ask Tricare"
 )
 
-# ========== LAYOUT ==========
 def chat_message_card(msg, is_user):
     align = "end" if is_user else "start"
     color = "primary" if is_user else "secondary"
@@ -229,7 +233,6 @@ app.layout = html.Div([
     ], style={"display": "flex"})
 ])
 
-# ========== CALLBACKS ==========
 @app.callback(
     Output("session-id", "data"),
     Input("url", "href"),
@@ -266,7 +269,6 @@ def main_callback(session_id, send_clicks, file_contents, file_names, user_input
         error = ""
         loading = False
 
-        # File upload
         if trigger == "file-upload" and file_contents and file_names:
             uploads = []
             if not isinstance(file_contents, list):
@@ -286,12 +288,10 @@ def main_callback(session_id, send_clicks, file_contents, file_names, user_input
             save_session_state(session_id)
             logger.info(f"Session {session_id}: Uploaded files {[u['name'] for u in uploads]}")
 
-        # Chat send
         if trigger == "send-btn" and user_input and user_input.strip():
             loading = True
             state["messages"].append({"role": "user", "content": user_input})
             try:
-                # RAG: retrieve relevant docs
                 docs = []
                 try:
                     retr = vectordb.similarity_search(user_input, k=3)
@@ -299,8 +299,9 @@ def main_callback(session_id, send_clicks, file_contents, file_names, user_input
                 except Exception as e:
                     logger.warning(f"Vector search failed: {e}")
                 context = "\n\n".join(docs)
+                system_prompt = load_system_prompt()
                 messages = [
-                    {"role": "system", "content": "You are Ask Tricare, a helpful assistant for TRICARE health benefits. Respond conversationally, and cite relevant sources when possible. If you do not know, say so."},
+                    {"role": "system", "content": system_prompt},
                 ]
                 for m in state["messages"]:
                     messages.append({"role": m["role"], "content": m["content"]})
@@ -328,7 +329,6 @@ def main_callback(session_id, send_clicks, file_contents, file_names, user_input
         right = right_main(chat_history, loading, error)
         return left, right
 
-# ========== SESSION COOKIE ==========
 @app_flask.after_request
 def set_session_cookie(resp):
     sid = flask_request.cookies.get("asktricare_session_id")
@@ -337,7 +337,6 @@ def set_session_cookie(resp):
         resp.set_cookie("asktricare_session_id", sid, max_age=60*60*24*7, path="/")
     return resp
 
-# ========== CLEANUP ==========
 def cleanup_sessions(max_age_hours=48):
     now = datetime.datetime.utcnow()
     for sid in os.listdir(SESSION_DIR_BASE):
@@ -354,7 +353,6 @@ def cleanup_sessions(max_age_hours=48):
         except Exception as e:
             logger.error(f"Cleanup error for {sid}: {e}")
 
-# ========== CUDA/GPU SETUP ==========
 try:
     import torch
     if torch.cuda.is_available():
@@ -363,9 +361,7 @@ try:
 except Exception as e:
     logger.warning(f"CUDA config failed: {e}")
 
-# ========== RUN ==========
 if __name__ == '__main__':
     print("Starting the Dash application...")
     app.run(debug=True, host='0.0.0.0', port=7860, threaded=True)
-    print("Dash application has finished running.")
-```
+    print("Dash application has finished running.")