Update app.py

app.py

@@ -50,7 +50,7 @@ def analyze_construction_image(image):
                 "content": [
                     {
                         "type": "text",
-                        "text": "Analyze this construction site image. Identify any issues or snags, categorize them, provide a detailed description, and suggest steps to resolve them. Format your response as a JSON object with keys 'snag_category', 'snag_description', and 'desnag_steps' (as an array)."
+                        "text": "Analyze this construction site image. Identify any safety issues or hazards, categorize them, provide a detailed description, and suggest steps to resolve them. Format your response as a JSON object with a key 'issues' containing an array of issue objects. Each issue object should have 'category', 'description', and 'resolution' keys."
                     },
                     {
                         "type": "image_url",
@@ -78,33 +78,26 @@ def analyze_construction_image(image):
         result = completion.choices[0].message.content
         logger.debug(f"Raw API response: {result}")
 
-        # Try to parse the result as JSON
+        # Parse the result as JSON
         try:
             parsed_result = json.loads(result)
+            issues = parsed_result.get('issues', [])
         except json.JSONDecodeError:
             logger.error("Failed to parse API response as JSON")
             return [(None, "Error: Invalid response format")]
 
-        snag_category = str(parsed_result.get('snag_category', 'N/A'))
-        snag_description = str(parsed_result.get('snag_description', 'N/A'))
-        
-        # Ensure desnag_steps is a list of strings
-        desnag_steps = parsed_result.get('desnag_steps', ['N/A'])
-        if not isinstance(desnag_steps, list):
-            desnag_steps = [str(desnag_steps)]
-        else:
-            desnag_steps = [str(step) for step in desnag_steps]
-        
-        desnag_steps_str = '\n'.join(desnag_steps)
+        # Format the analysis results
+        analysis_text = "Image Analysis Results:\n\n"
+        for issue in issues:
+            analysis_text += f"Category: {issue.get('category', 'N/A')}\n"
+            analysis_text += f"Description: {issue.get('description', 'N/A')}\n"
+            analysis_text += "Resolution Steps:\n"
+            for step in issue.get('resolution', ['N/A']):
+                analysis_text += f"- {step}\n"
+            analysis_text += "\n"
 
         logger.info("Analysis completed successfully")
-        
-        # Initialize chat history with analysis results
-        chat_history = [
-            (None, f"Image Analysis Results:\n\nSnag Category: {snag_category}\n\nSnag Description: {snag_description}\n\nSteps to Desnag:\n{desnag_steps_str}")
-        ]
-        
-        return chat_history
+        return [(None, analysis_text)]
     except Exception as e:
         logger.error(f"Error during image analysis: {str(e)}")
         return [(None, f"Error: {str(e)}")]