Update app.py

app.py

@@ -6,8 +6,6 @@ import time
 from PIL import Image as PILImage
 import io
 import os
-from tempfile import NamedTemporaryFile
-from pathlib import Path
 import base64
 
 def create_monitor_interface():
@@ -15,7 +13,7 @@ def create_monitor_interface():
     
     class SafetyMonitor:
         def __init__(self):
-            self.client = Groq(api_key=api_key)
+            self.client = Groq()  # API key handled by environment variable
             self.model_name = "llama-3.2-90b-vision-preview"
             self.max_image_size = (128, 128)
             
@@ -52,35 +50,37 @@ def create_monitor_interface():
                          quality=20, 
                          optimize=True)
             img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
+            image_url = f"data:image/jpeg;base64,{img_base64}"
             
             try:
                 completion = self.client.chat.completions.create(
+                    model=self.model_name,
                     messages=[
-                        {
-                            "role": "user",
-                            "content": "You are a workplace safety expert. Analyze the following image for safety concerns."
-                        },
-                        {
-                            "role": "assistant",
-                            "content": "I'll analyze the image for workplace safety concerns and provide specific observations."
-                        },
                         {
                             "role": "user",
                             "content": [
                                 {
                                     "type": "text",
-                                    "text": "What safety issues do you see?"
+                                    "text": "Analyze this workplace for safety concerns. List any safety issues, missing PPE, or hazards you observe."
                                 },
                                 {
                                     "type": "image_url",
-                                    "url": f"data:image/jpeg;base64,{img_base64}"
+                                    "image_url": {
+                                        "url": image_url
+                                    }
                                 }
                             ]
+                        },
+                        {
+                            "role": "assistant",
+                            "content": ""
                         }
                     ],
-                    model=self.model_name,
+                    temperature=0.1,
                     max_tokens=100,
-                    temperature=0.1
+                    top_p=1,
+                    stream=False,
+                    stop=None
                 )
                 return completion.choices[0].message.content
             except Exception as e: