Update app.py

app.py

@@ -46,14 +46,12 @@ class RAGSystem:
     @lru_cache(maxsize=1)  # Cache the knowledge base
     def load_knowledge_base() -> List[Dict]:
         """Load and preprocess knowledge base"""
-        # Your existing knowledge base code...
         kb = {
             "spalling": [
                 {
                     "severity": "Critical",
                     "description": "Severe concrete spalling with exposed reinforcement",
                     "repair_method": "Remove deteriorated concrete, clean reinforcement",
-                    "estimated_cost": "Very High ($15,000+)",
                     "immediate_action": "Evacuate area, install support",
                     "prevention": "Regular inspections, waterproofing"
                 }
@@ -63,7 +61,6 @@ class RAGSystem:
                     "severity": "High",
                     "description": "Active structural cracks >5mm width",
                     "repair_method": "Structural analysis, epoxy injection",
-                    "estimated_cost": "High ($10,000-$20,000)",
                     "immediate_action": "Install crack monitors",
                     "prevention": "Regular monitoring, load management"
                 }
@@ -73,13 +70,75 @@ class RAGSystem:
                     "severity": "Medium",
                     "description": "Surface scaling and deterioration",
                     "repair_method": "Surface preparation, patch repair",
-                    "estimated_cost": "Medium ($5,000-$10,000)",
                     "immediate_action": "Document extent, plan repairs",
                     "prevention": "Surface sealers, proper drainage"
                 }
+            ],
+            "corrosion": [
+                {
+                    "severity": "High",
+                    "description": "Corrosion of reinforcement leading to cracks",
+                    "repair_method": "Remove rust, apply inhibitors",
+                    "immediate_action": "Isolate affected area",
+                    "prevention": "Anti-corrosion coatings, proper drainage"
+                }
+            ],
+            "efflorescence": [
+                {
+                    "severity": "Low",
+                    "description": "White powder deposits on concrete surfaces",
+                    "repair_method": "Surface cleaning, sealant application",
+                    "immediate_action": "Identify moisture source",
+                    "prevention": "Improve waterproofing, reduce moisture ingress"
+                }
+            ],
+            "delamination": [
+                {
+                    "severity": "Medium",
+                    "description": "Separation of layers in concrete",
+                    "repair_method": "Resurface or replace delaminated sections",
+                    "immediate_action": "Inspect bonding layers",
+                    "prevention": "Proper curing and bonding agents"
+                }
+            ],
+            "honeycombing": [
+                {
+                    "severity": "Medium",
+                    "description": "Voids in concrete caused by improper compaction",
+                    "repair_method": "Grout injection, patch repair",
+                    "immediate_action": "Assess structural impact",
+                    "prevention": "Proper vibration during pouring"
+                }
+            ],
+            "water_leakage": [
+                {
+                    "severity": "High",
+                    "description": "Water ingress through cracks or joints",
+                    "repair_method": "Injection grouting, waterproofing membranes",
+                    "immediate_action": "Stop water flow, apply sealants",
+                    "prevention": "Drainage systems, joint sealing"
+                }
+            ],
+            "settlement_cracks": [
+                {
+                    "severity": "High",
+                    "description": "Cracks due to uneven foundation settlement",
+                    "repair_method": "Foundation underpinning, grouting",
+                    "immediate_action": "Monitor movement, stabilize foundation",
+                    "prevention": "Soil compaction, proper foundation design"
+                }
+            ],
+            "shrinkage_cracks": [
+                {
+                    "severity": "Low",
+                    "description": "Minor cracks caused by shrinkage during curing",
+                    "repair_method": "Sealant application, surface repairs",
+                    "immediate_action": "Monitor cracks",
+                    "prevention": "Proper curing and moisture control"
+                }
             ]
         }
-        
+
         documents = []
         for category, items in kb.items():
             for item in items:
@@ -87,7 +146,7 @@ class RAGSystem:
                 for key, value in item.items():
                     doc_text += f"{key}: {value}\n"
                 documents.append({"text": doc_text, "metadata": {"category": category}})
-        
+
         return documents
 
     def create_vector_store(self):