Update app.py

app.py

@@ -18,10 +18,10 @@ def create_monitor_interface():
             self.max_image_size = (800, 800)
             self.colors = [(0, 0, 255), (255, 0, 0), (0, 255, 0), (255, 255, 0), (255, 0, 255)]
             
-        def analyze_frame(self, frame: np.ndarray) -> str:
+         def analyze_frame(self, frame: np.ndarray) -> str:
             if frame is None:
                 return ""
-                
+                    
             # Convert image
             if len(frame.shape) == 2:
                 frame = cv2.cvtColor(frame, cv2.COLOR_GRAY2RGB)
@@ -33,9 +33,9 @@ def create_monitor_interface():
             
             buffered = io.BytesIO()
             frame_pil.save(buffered, 
-                         format="JPEG", 
-                         quality=85,
-                         optimize=True)
+                          format="JPEG", 
+                          quality=95,  # Increased quality
+                          optimize=True)
             img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
             image_url = f"data:image/jpeg;base64,{img_base64}"
             
@@ -45,27 +45,34 @@ def create_monitor_interface():
                     messages=[
                         {
                             "role": "system",
-                            "content": "You are a safety analysis expert. Analyze images for safety concerns and provide detailed observations."
+                            "content": """You are a construction site safety expert specializing in ergonomics and workplace safety. 
+                            Analyze images for:
+                            1. Worker posture and ergonomic risks
+                            2. PPE usage and compliance
+                            3. Tool and equipment safety
+                            4. Environmental hazards
+                            5. Working position and technique"""
                         },
                         {
                             "role": "user",
                             "content": [
                                 {
                                     "type": "text",
-                                    "text": """Analyze this image for safety concerns and risks. For each issue you identify:
-
-1. Specify the exact location in the image where the issue is visible
-2. Describe what the safety concern is
-3. Include any relevant details about PPE, posture, equipment, or environmental hazards
-
-Format EACH observation exactly like this:
-- <location>position:detailed description of the concern</location>
-
-Example format:
-- <location>center:Worker bending incorrectly while lifting heavy materials</location>
-- <location>top-right:Missing safety guardrail near elevated platform</location>
-
-Provide multiple observations if you see multiple issues."""
+                                    "text": """Carefully analyze this construction worker's position and environment. Look for:
+        
+        1. Ergonomic issues (kneeling position, back posture, repetitive motions)
+        2. PPE compliance (knee pads, gloves, appropriate footwear)
+        3. Working technique and body mechanics
+        4. Surrounding hazards or risks
+        
+        For each issue identified, format your response as:
+        - <location>position:specific safety concern and recommendation</location>
+        
+        For example:
+        - <location>center:Worker kneeling without knee protection, risking joint injury. Recommend knee pads.</location>
+        - <location>bottom:Improper back posture while working, potential for strain. Should maintain straight back.</location>
+        
+        Be specific about each safety concern you observe."""
                                 },
                                 {
                                     "type": "image_url",
@@ -76,7 +83,7 @@ Provide multiple observations if you see multiple issues."""
                             ]
                         }
                     ],
-                    temperature=0.5,  # Increased for more varied observations
+                    temperature=0.3,  # Lowered temperature for more focused analysis
                     max_tokens=500,
                     stream=False
                 )