Update app.py

app.py

@@ -47,46 +47,6 @@ class SafetyMonitor:
         img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
         return f"data:image/jpeg;base64,{img_base64}"
 
-    def get_scene_context(self, image):
-        """Analyze the scene context."""
-        try:
-            image_url = self.encode_image(image)
-            completion = self.client.chat.completions.create(
-                model=self.model_name,
-                messages=[
-                    {
-                        "role": "user",
-                        "content": [
-                            {
-                                "type": "text",
-                                "text": """Analyze this workplace image and identify key areas and elements. Include:
-                                1. Worker locations and activities
-                                2. Equipment and machinery
-                                3. Materials and storage
-                                4. Access routes and paths
-                                5. Hazardous areas
-                                
-                                Format each observation as:
-                                - Element: specific location in image"""
-                            },
-                            {
-                                "type": "image_url",
-                                "image_url": {
-                                    "url": image_url
-                                }
-                            }
-                        ]
-                    }
-                ],
-                temperature=0.3,
-                max_tokens=200,
-                stream=False
-            )
-            return completion.choices[0].message.content
-        except Exception as e:
-            print(f"Scene analysis error: {str(e)}")
-            return ""
-
     def analyze_frame(self, frame):
         """Perform safety analysis on the frame."""
         if frame is None:
@@ -113,14 +73,6 @@ 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.7,
@@ -136,12 +88,6 @@ class SafetyMonitor:
         """Convert textual position to coordinates."""
         height, width = image_shape[:2]
         
-        # Parse position for spatial information
-        position = position.lower()
-        
-        # Base coordinates (full image)
-        x1, y1, x2, y2 = 0, 0, width, height
-        
         # Define regions
         regions = {
             'center': (width//3, height//3, 2*width//3, 2*height//3),
@@ -158,18 +104,9 @@ class SafetyMonitor:
             'middle': (0, height//3, width, 2*height//3)
         }
         
-        # Find best matching region
-        best_match = None
-        max_match = 0
-        for region, coords in regions.items():
-            if region in position:
-                words = region.split('-')
-                matches = sum(1 for word in words if word in position)
-                if matches > max_match:
-                    max_match = matches
-                    best_match = coords
-        
-        return best_match if best_match else (x1, y1, x2, y2)
+        # Ensure the region name from the model output matches one of our predefined regions
+        position = position.lower()
+        return regions.get(position, (0, 0, width, height))  # Default to full image if no match
 
     def draw_observations(self, image, observations):
         """Draw bounding boxes and labels for safety observations."""
@@ -184,6 +121,7 @@ class SafetyMonitor:
             
             # Get coordinates for this observation
             x1, y1, x2, y2 = self.get_region_coordinates(obs['location'], image.shape)
+            print(f"Drawing box at coordinates: ({x1}, {y1}, {x2}, {y2}) for {obs['description']}")
             
             # Draw rectangle
             cv2.rectangle(image, (x1, y1), (x2, y2), color, 2)