improved system prompt and container logs

app.py

@@ -34,10 +34,12 @@ print("Symptom index built successfully. Ready for queries.", flush=True)
 
 # ========== Prompt template ==========
 SYSTEM_PROMPT = (
-    "You are a medical assistant helping a user narrow down to the most likely ICD-10 code. "
-    "At each turn, either ask one focused clarifying question (e.g. 'Is your cough dry or productive?') "
-    "or if you have enough information, provide a final JSON with fields: {\"diagnoses\": [...], "
-    "\"confidences\": [...], \"follow_up\": [...]}. Output must be valid JSON with no trailing commas. Your output MUST be strictly valid JSON, starting with '{' and ending with '}', with no extra text outside the JSON."
+    "You are a medical assistant helping a user find the most relevant ICD-10 code based on their symptoms.\n",
+    "At each turn, determine the top three most relevant ICD-10 codes based on input from the user.\n",
+    "Additionally, provide a 1 through 100 score of how confident you are in the relevancy of each code.\n",
+    "Continually ask questions to the user to raise or lower your confidence of each code.\n",
+    "Replace low-confidence codes with new ones as you learn more.\n",
+    "Your goal is to find the most relevant codes with high confidence.\n",
 )
 
 # ========== Generator handler ==========

services/indexing.py

@@ -1,20 +1,20 @@
-from llama_index.core import SimpleDirectoryReader, VectorStoreIndex
-from llama_index.core import Settings
+# at top of file
+import logging, sys
+logging.basicConfig(stream=sys.stdout, level=logging.INFO, force=True)
+logger = logging.getLogger(__name__)
 
 def create_symptom_index():
-    """Create and return symptom index from ICD-10 data."""
-    print("build_symptom_index: Loading documents from data directory...")
+    logger.info("build_symptom_index: Loading documents from data directory…")
     documents = SimpleDirectoryReader(
         input_dir="data",
         filename_as_id=True
     ).load_data()
-    
-    print(f"build_symptom_index: Creating vector index from {len(documents)} documents...")
+
+    logger.info(f"build_symptom_index: Creating vector index from {len(documents)} documents…")
     symptom_index = VectorStoreIndex.from_documents(
         documents,
         show_progress=True
     )
-    
-    print("build_symptom_index: Symptom index created successfully")
 
+    logger.info("build_symptom_index: Symptom index created successfully")
     return symptom_index