Update app.py

app.py

@@ -11,25 +11,6 @@ import base64
 def create_monitor_interface():
     api_key = os.getenv("GROQ_API_KEY")
     
-    if not api_key:
-        with gr.Blocks() as demo:
-            gr.Markdown("""
-            # ⚠️ Groq API Key Required
-            
-            ## Setup Instructions for Hugging Face Space:
-            1. Go to your Space's Settings tab
-            2. Scroll down to "Repository Secrets"
-            3. Click "New Secret"
-            4. Enter:
-               - Secret name: `GROQ_API_KEY`
-               - Secret value: Your Groq API key
-            5. Click "Add secret"
-            6. Rebuild the Space
-            
-            Once configured, the safety monitoring system will be available.
-            """)
-        return demo
-
     class SafetyMonitor:
         def __init__(self, model_name: str = "mixtral-8x7b-vision"):
             self.client = Groq(api_key=api_key)
@@ -40,9 +21,9 @@ def create_monitor_interface():
                 return "No frame received"
                 
             # Convert numpy array to PIL Image
-            if len(frame.shape) == 2:  # If grayscale
+            if len(frame.shape) == 2:
                 frame = cv2.cvtColor(frame, cv2.COLOR_GRAY2RGB)
-            elif len(frame.shape) == 3 and frame.shape[2] == 4:  # If RGBA
+            elif len(frame.shape) == 3 and frame.shape[2] == 4:
                 frame = cv2.cvtColor(frame, cv2.COLOR_RGBA2RGB)
                 
             frame_pil = PILImage.fromarray(frame)
@@ -52,34 +33,24 @@ def create_monitor_interface():
             frame_pil.save(buffered, format="JPEG")
             img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
             
-            # Construct the prompt with proper format
-            system_prompt = """You are a workplace safety expert. Analyze the image for safety concerns, focusing on:
-            1. PPE usage (helmets, vests, etc.)
-            2. Unsafe behaviors
-            3. Equipment safety
-            4. Environmental hazards
-            
-            Provide clear, specific observations."""
-            
             try:
                 completion = self.client.chat.completions.create(
                     messages=[
-                        {
-                            "role": "system",
-                            "content": system_prompt
-                        },
                         {
                             "role": "user",
                             "content": [
                                 {
                                     "type": "text",
-                                    "text": "Please analyze this image for safety concerns."
+                                    "text": """Please analyze this image for workplace safety issues. Focus on:
+                                    1. PPE usage (helmets, safety glasses, vests)
+                                    2. Unsafe behaviors or positions
+                                    3. Equipment and machinery safety
+                                    4. Environmental hazards
+                                    Provide specific observations."""
                                 },
                                 {
-                                    "type": "image",
-                                    "image": {
-                                        "url": f"data:image/jpeg;base64,{img_base64}"
-                                    }
+                                    "type": "image_url",
+                                    "url": f"data:image/jpeg;base64,{img_base64}"
                                 }
                             ]
                         }
@@ -113,7 +84,7 @@ def create_monitor_interface():
                 cv2.putText(display_frame, line[:80], (10, y_position),
                            cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2)
                 y_position += 30
-                if y_position >= 90:  # Prevent text from going outside the overlay
+                if y_position >= 90:
                     break
 
             return display_frame, analysis
@@ -140,6 +111,7 @@ def create_monitor_interface():
                 processed_frame, analysis = monitor.process_frame(image)
                 return processed_frame, analysis
             except Exception as e:
+                print(f"Processing error: {str(e)}")  # For debugging
                 return None, f"Error processing image: {str(e)}"
             
         input_image.change(