Update app.py

app.py

@@ -91,7 +91,7 @@ class SafetyMonitor:
         """Perform safety analysis on the frame."""
         if frame is None:
             return "No frame received", {}
-
+    
         frame = self.preprocess_image(frame)
         image_url = self.encode_image(frame)
         
@@ -104,23 +104,7 @@ class SafetyMonitor:
                         "content": [
                             {
                                 "type": "text",
-                                "text": """Analyze this image for safety hazards. For each hazard:
-                                1. Specify the precise location in the image
-                                2. Describe the safety concern or violation
-                                3. Indicate the potential risk
-
-                                Format each finding as:
-                                - <location>position:detailed safety concern</location>
-
-                                Look for all types of safety issues:
-                                - PPE compliance
-                                - Ergonomic risks
-                                - Equipment safety
-                                - Environmental hazards
-                                - Material handling
-                                - Work procedures
-                                - Access and egress
-                                - Housekeeping"""
+                                "text": "Identify and list safety concerns in this workplace image. For each issue found, include its location and specific safety concern. Look for hazards related to PPE, ergonomics, equipment, environment, and work procedures."
                             },
                             {
                                 "type": "image_url",
@@ -129,9 +113,17 @@ class SafetyMonitor:
                                 }
                             }
                         ]
+                    },
+                    {
+                        "role": "assistant",
+                        "content": "I'll analyze the image for safety hazards and provide the location and description of each concern."
+                    },
+                    {
+                        "role": "user",
+                        "content": "Please format each observation as: - <location>position:detailed safety concern</location>"
                     }
                 ],
-                temperature=0.5,
+                temperature=0.7,
                 max_tokens=500,
                 stream=False
             )
@@ -186,7 +178,7 @@ class SafetyMonitor:
         font_scale = 0.5
         thickness = 2
         padding = 10
-
+    
         for idx, obs in enumerate(observations):
             color = self.colors[idx % len(self.colors)]
             
@@ -225,6 +217,7 @@ class SafetyMonitor:
         try:
             # Get analysis
             analysis, _ = self.analyze_frame(frame)
+            print(f"Raw analysis: {analysis}")  # Debug print
             display_frame = frame.copy()
             
             # Parse observations
@@ -243,6 +236,8 @@ class SafetyMonitor:
                             'description': description.strip()
                         })
             
+            print(f"Parsed observations: {observations}")  # Debug print
+            
             # Draw observations
             if observations:
                 annotated_frame = self.draw_observations(display_frame, observations)