Update app.py

app.py

@@ -84,27 +84,18 @@ def analyze_construction_image(image):
 
         # Parse the result
         analysis_text = "Safety and Hazard Analysis of the Construction Site Image:\n\n"
-        categories = result.split("Category")[1:]  # Split by categories
         
-        if not categories:
-            logger.warning("No categories found in the response")
-            return [(None, "Error: Unable to parse the response. No categories found.")]
-
-        for category in categories:
-            lines = category.split("\n")
-            category_name = lines[0].split(":")[1].strip() if ":" in lines[0] else lines[0].strip()
-            analysis_text += f"Category: {category_name}\n"
-
-            description = next((line.split("Description:")[1].strip() for line in lines if "Description:" in line), "N/A")
-            hazard = next((line.split("Hazard:")[1].strip() for line in lines if "Hazard:" in line), "N/A")
-            resolution_steps = [line.strip() for line in lines if line.strip().startswith("*")]
-
-            analysis_text += f"Description: {description}\n"
-            analysis_text += f"Hazard: {hazard}\n"
-            analysis_text += "Resolution Steps:\n"
-            for step in resolution_steps:
-                analysis_text += f"{step}\n"
-            analysis_text += "\n"
+        sections = ["Hazard Identification", "Categorization", "Detailed Description", "Resolution Steps"]
+        for section in sections:
+            start = result.find(section)
+            if start != -1:
+                end = result.find(next((s for s in sections if s != section and result.find(s) > start), None))
+                content = result[start:end if end != -1 else None].strip()
+                analysis_text += f"{content}\n\n"
+
+        if analysis_text == "Safety and Hazard Analysis of the Construction Site Image:\n\n":
+            logger.warning("No sections found in the response")
+            return [(None, "Error: Unable to parse the response. No sections found.")]
 
         logger.info("Analysis completed successfully")
         return [(None, analysis_text)]