Trying to fix basic functionality again.

scripts/test_module_exists.py

@@ -0,0 +1,123 @@
+#!/usr/bin/env python3
+"""
+Simple script to check if a module exists for a given disease and generate it if not.
+"""
+
+import os
+import sys
+import json
+import subprocess
+import re
+
+# Constants
+DISEASE_NAME = sys.argv[1] if len(sys.argv) > 1 else "Excessive frequent and irregular menstruation"
+MODULES_DIR = "src/main/resources/modules"
+RUN_MODULE_GENERATOR_PATH = "src/main/python/run_module_generator.py"
+DISEASE_LIST_PATH = "src/main/resources/disease_list.json"
+
+# Function to normalize disease name for filenames
+def normalize_filename(name):
+    """Convert disease name to normalized filename format."""
+    filename = name.lower()
+    filename = re.sub(r'[^a-zA-Z0-9]', '_', filename)
+    filename = re.sub(r'_+', '_', filename)
+    filename = filename.strip('_')
+    return filename
+
+# Main function
+def main():
+    print(f"Checking if module exists for: {DISEASE_NAME}")
+    
+    # Normalize disease name for filename
+    normalized_name = normalize_filename(DISEASE_NAME)
+    module_path = os.path.join(MODULES_DIR, f"{normalized_name}.json")
+    
+    # Check if module already exists
+    if os.path.exists(module_path):
+        print(f"✅ Module already exists at: {module_path}")
+        return
+    
+    print(f"❌ Module not found at: {module_path}")
+    print(f"Creating disease list entry for {DISEASE_NAME}...")
+    
+    # Create temporary disease list with just this disease
+    create_disease_list(DISEASE_NAME)
+    
+    # Run the module generator
+    print(f"Running module generator for {DISEASE_NAME}...")
+    try:
+        result = subprocess.run(
+            [sys.executable, RUN_MODULE_GENERATOR_PATH, "--batch-size", "1"],
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        
+        print("Module generator output:")
+        print(result.stdout)
+        
+        # Check if module was created
+        if os.path.exists(module_path):
+            print(f"✅ Module created successfully at: {module_path}")
+        else:
+            print(f"❌ Module generation failed, file not found at: {module_path}")
+            print("Error output:")
+            print(result.stderr)
+    except subprocess.CalledProcessError as e:
+        print(f"Error running module generator: {e}")
+        print("STDOUT:")
+        print(e.stdout)
+        print("STDERR:")
+        print(e.stderr)
+
+def create_disease_list(disease_name):
+    """Create a temporary disease list with just the requested disease."""
+    disease_list = []
+    
+    # Try to read original disease list if it exists
+    if os.path.exists(DISEASE_LIST_PATH):
+        try:
+            with open(DISEASE_LIST_PATH, 'r') as f:
+                original_diseases = json.load(f)
+            
+            # Check if our disease already exists
+            name_lower = disease_name.lower()
+            for disease in original_diseases:
+                if disease.get("disease_name", "").lower() == name_lower:
+                    disease_list.append(disease)
+                    print(f"Found existing disease entry for '{disease_name}'")
+                    break
+            else:
+                # Disease not found in list
+                disease_list.append({
+                    "id": "",  # ICD-10 code (empty as we don't have it)
+                    "disease_name": disease_name,
+                    "ICD-10_name": disease_name
+                })
+                print(f"Created new disease entry for '{disease_name}'")
+        except Exception as e:
+            print(f"Error reading disease list: {e}")
+            # Create new entry
+            disease_list.append({
+                "id": "",
+                "disease_name": disease_name,
+                "ICD-10_name": disease_name
+            })
+    else:
+        # Create new entry
+        disease_list.append({
+            "id": "",
+            "disease_name": disease_name,
+            "ICD-10_name": disease_name
+        })
+        print(f"Created new disease list with '{disease_name}'")
+    
+    # Save disease list
+    os.makedirs(os.path.dirname(DISEASE_LIST_PATH), exist_ok=True)
+    with open(DISEASE_LIST_PATH, 'w') as f:
+        json.dump(disease_list, f, indent=2)
+    
+    print(f"Saved disease list with {len(disease_list)} entries.")
+
+if __name__ == "__main__":
+    main()