Update app.py

app.py

@@ -100,35 +100,28 @@ if category_choice == "Flooring":
             st.warning("⚠️ The confidence score is below 59%. Please manually verify this result.")
         else:
             compressed_base64 = process_image_for_openai(file_bytes)
-            ai_content =  """You are an expert in identifying and resolving defects in wall and flooring systems. Use the provided dataset to analyze and generate all applicable recommended solutions for a given defect or combination of defects.
+            ai_content =  """
+            You are an expert in identifying and resolving wall and flooring defects. Use the dataset below to generate all applicable recommended solutions for any provided defect or combination of defects.
 
-The dataset includes three fields:
+Instructions:
+Accept a defect name as input.
 
-Defect
+Search the dataset for matches and return all recommended solutions, grouped by Sub-Defect Type.
 
-Sub-Defect Type
+If the defect appears under multiple subtypes, combine and list all applicable solutions.
 
-Recommended Solution
+If the defect is not found in the dataset, use semantic analysis and your domain expertise to generate a logical and professional solution.
 
-Your task is to do the following:
+Do not mention whether the defect is listed in the dataset or not — simply return the recommended solution.
 
-Accept a defect name as input (e.g., "Trowel Mark", "Peel off/Debonding").
+Use clear steps such as: surface preparation, grinding, priming, coating, sealing, repairing, joint treatment, or finishing, depending on the nature of the defect
 
-Search the dataset and return all matching recommended solutions.
 
-Clearly label the results by Sub-Defect Type (e.g., Epoxy, PU).
 
-If a defect includes multiple sub-defect types (e.g., both Epoxy and PU), return all corresponding solutions.
-
-If the defect is a combined case (e.g., "Cracks on floors Epoxy as well PU"), provide the full solution as a combined recommendation.
-
-Use the following dataset for reference:
 
 Dataset:
 
-vbnet
-Copy
-Edit
+
 Defect: Spike Roller Mark  
 Sub-Defect Type: Epoxy  
 Recommended Solution:  
@@ -248,7 +241,54 @@ Recommended Solution:
 1. Surface preparation - Cut the pothole in square/rectangle shape, remove dust/oil  
 2. Apply Epoxy Primer  
 3. Apply Apcoflor HFP 120 + special graded sand at 1:3 mix ratio  
-4. Finish/level the surface  """
+4. Finish/level the surface  
+
+Fallback Logic for Unknown Defects:
+Use these patterns to infer the solution without stating the defect is not listed.
+
+Structural Defect:
+
+Cut/clean damaged area
+
+Surface preparation (light grinding or cutting)
+
+Apply epoxy/PU primer
+
+Apply screed or mortar (e.g., Apcoflor HFP 120 + sand)
+
+Finish and level as needed
+
+Aesthetic Defect (e.g., fading, discoloration):
+
+Surface preparation by light grinding
+
+Application of appropriate primer
+
+Application of thin film coating (Epoxy: Apcoflor TC 510, PU: Apcoflor PUV, Hybrid as needed)
+
+Final finish for uniform appearance
+
+Cracks & Breaks:
+
+Open in "V" groove
+
+Apply primer
+
+Fill with epoxy putty (e.g., Apcoflor LSC3 XL)
+
+Joint Issues:
+
+Remove loose particles
+
+Apply epoxy primer
+
+Joint treatment with sealants or epoxy mortars
+
+Protect with tape or metal strip if needed
+
+
+            """
+
             ai_prompt = (
                  f"Our trained model predicts the following defect: {class_name}. "