Update app.py

app.py

@@ -100,6 +100,155 @@ 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.
+
+The dataset includes three fields:
+
+Defect
+
+Sub-Defect Type
+
+Recommended Solution
+
+Your task is to do the following:
+
+Accept a defect name as input (e.g., "Trowel Mark", "Peel off/Debonding").
+
+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:  
+1. Surface preparation on existing resinious flooring by light grinding.  
+2. Application of Primer.  
+3. Application of thin film coating with Apcoflor TC 510  
+
+Defect: Spike Roller Mark  
+Sub-Defect Type: PU  
+Recommended Solution:  
+1. Surface preparation on existing resinious flooring by light grinding.  
+2. Application of Primer.  
+3. Application of thin film coating with Apcoflor PUV  
+
+Defect: Trowel Mark  
+Sub-Defect Type: Epoxy  
+Recommended Solution:  
+1. Surface preparation on existing resinious flooring by light grinding.  
+2. Application of Primer.  
+3. Application of epoxy topcoat with 1mm thickness using Apcoflor SL1TC  
+
+Defect: Trowel Mark  
+Sub-Defect Type: PU  
+Recommended Solution:  
+1. Surface preparation on existing resinious flooring by light grinding  
+2. Groove cutting  
+3. Application of PU scratch coat  
+4. Application of Topcoat with 1 to 2 mm thickness  
+
+Defect: Bubble Formation  
+Sub-Defect Type: Epoxy  
+Recommended Solution:  
+1. Surface preparation on existing resinious flooring by light grinding.  
+2. Application of Primer.  
+3. Application of thin film coating with Apcoflor TC 510  
+
+Defect: Bubble Formation  
+Sub-Defect Type: PU  
+Recommended Solution:  
+1. Surface preparation on existing resinious flooring by light grinding  
+2. Groove cutting  
+3. Application of PU scratch coat  
+4. Application of Topcoat with 1 to 5 mm thickness  
+
+Defect: Peel off/Debonding  
+Sub-Defect Type: Epoxy  
+Recommended Solution:  
+1. Surface preparation on existing resinious flooring by light grinding  
+2. Removing all loose resinious flooring  
+3. Application of Primer  
+4. Application of thin film coating with Apcoflor TC 510/Apcoflor SL1TC  
+
+Defect: Peel off/Debonding  
+Sub-Defect Type: PU  
+Recommended Solution:  
+1. Surface preparation on existing resinious flooring by light grinding  
+2. Groove cutting  
+3. Application of PU scratch coat  
+4. Application of Topcoat with 1 to 5 mm thickness  
+
+Defect: Shade Variation  
+Sub-Defect Type: Epoxy  
+Recommended Solution:  
+1. Surface preparation on existing resinious flooring by light grinding  
+2. Application of Primer  
+3. Application of thin film coating with Apcoflor TC 510/Apcoflor SL1 TC  
+
+Defect: Shade Variation  
+Sub-Defect Type: PU  
+Recommended Solution:  
+1. Surface preparation on existing resinious flooring by light grinding  
+2. Application of Primer  
+3. Application of thin film coating with Apcoflor PUV  
+
+Defect: Cracks on Floors  
+Sub-Defect Type: Epoxy and PU  
+Recommended Solution:  
+1. Surface preparation - opening the cracks in "V" groove  
+2. Application of Primer  
+3. Application of epoxy putty using Apcoflor LSC3 XL  
+
+Defect: Expansion Joint  
+Sub-Defect Type: Cracking & Spalling of Joint Edge  
+Recommended Solution:  
+1. Crack Treatment - Open cracks in "V" groove  
+   - Apply Epoxy Primer  
+   - Apply epoxy putty using Apcoflor LSC3 XL  
+2. Joint Edge Treatment  
+   - Remove loose concrete  
+   - Apply Epoxy mortar using Apcoflor HFP 120 + special sand at 1:3.5 mix ratio  
+   - Apply at damaged joint edges  
+
+Defect: Expansion Joint  
+Sub-Defect Type: Sealant Failure  
+Recommended Solution:  
+1. Apply PU-based backer rod  
+2. Apply PU/Polysulphide Sealant over the backer rod up to the top of the joint edges  
+3. Provide protection using self-adhesive tapes or aluminum stripping  
+
+Defect: Construction Joint  
+Sub-Defect Type: -  
+Recommended Solution:  
+1. Remove dust/impurities from joints  
+2. Apply Primer  
+3. Apply Epoxy SL Screed or PU Screed  
+
+Defect: Wall & Floor Junction Coving  
+Sub-Defect Type: -  
+Recommended Solution:  
+1. Surface preparation  
+2. Apply Epoxy Primer or scratch coat  
+3. Apply Epoxy mortar using Apcoflor HFP 120 + special graded sand at 1:3 mixing ratio  
+
+Defect: Flooring - POT Holes  
+Sub-Defect Type: -  
+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  """
             ai_prompt = (
                  f"Our trained model predicts the following defect: {class_name}. "
                
@@ -115,14 +264,15 @@ if category_choice == "Flooring":
                         model="gpt-4o",
                         messages=[
                             {
-                                "role": "user",
-                                "content": [
-                                    {"type": "text", "text": ai_prompt},
-                                    {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{compressed_base64}"}}
-                                ]
+                                "role": "system",
+                                "content": ai_content
+                            },
+                            {
+                                "role":"user",
+                                "content":"Generate recommended solutions for this defect '{class_name}'"
                             }
                         ],
-                        max_tokens=300,
+                        max_tokens=600,
                     )
                     ai_description = response.choices[0].message.content
                     st.subheader("📝 AI-Generated Defect Description")