Update app.py

app.py

@@ -77,22 +77,24 @@ def analyze_construction_image(images=None, video=None):
         logger.info("Starting analysis")
         results = []
 
+        instruction = ("You are an AI assistant specialized in analyzing images for safety issues. "
+                       "Your task is to first determine if the image shows a construction site. "
+                       "If it does, identify any safety issues or hazards, categorize them, provide a detailed description, "
+                       "and suggest steps to resolve them. If it's not a construction site, simply state that and "
+                       "briefly describe what you see in the image.")
+
         if images:
             for i, image_file in enumerate(images):
                 image = Image.open(image_file.name)
                 resized_image = resize_image(image)
                 image_data_url = f"data:image/png;base64,{encode_image(resized_image)}"
                 messages = [
-                    {
-                        "role": "system",
-                        "content": "You are an AI assistant specialized in analyzing images for safety issues. Your task is to first determine if the image shows a construction site. If it does, identify any safety issues or hazards, categorize them, provide a detailed description, and suggest steps to resolve them. If it's not a construction site, simply state that and briefly describe what you see in the image."
-                    },
                     {
                         "role": "user",
                         "content": [
                             {
                                 "type": "text",
-                                "text": f"Analyze this image (Image {i+1}/{len(images)}). First, determine if it's a construction site. If it is, explain the image in detail, focusing on safety aspects. If it's not, briefly describe what you see."
+                                "text": f"{instruction}\n\nAnalyze this image (Image {i+1}/{len(images)}). First, determine if it's a construction site. If it is, explain the image in detail, focusing on safety aspects. If it's not, briefly describe what you see."
                             },
                             {
                                 "type": "image_url",
@@ -120,16 +122,12 @@ def analyze_construction_image(images=None, video=None):
             for i, frame in enumerate(frames):
                 image_data_url = f"data:image/png;base64,{encode_image(frame)}"
                 messages = [
-                    {
-                        "role": "system",
-                        "content": "You are an AI assistant specialized in analyzing video frames for safety issues. Your task is to first determine if the frame shows a construction site. If it does, identify any safety issues or hazards, categorize them, provide a detailed description, and suggest steps to resolve them. If it's not a construction site, simply state that and briefly describe what you see in the frame."
-                    },
                     {
                         "role": "user",
                         "content": [
                             {
                                 "type": "text",
-                                "text": f"Analyze this frame from a video (Frame {i+1}/{len(frames)}). First, determine if it's a construction site. If it is, explain what you observe, focusing on safety aspects. If it's not, briefly describe what you see."
+                                "text": f"{instruction}\n\nAnalyze this frame from a video (Frame {i+1}/{len(frames)}). First, determine if it's a construction site. If it is, explain what you observe, focusing on safety aspects. If it's not, briefly describe what you see."
                             },
                             {
                                 "type": "image_url",
@@ -158,6 +156,7 @@ def analyze_construction_image(images=None, video=None):
         logger.error(f"Error during analysis: {str(e)}")
         logger.error(traceback.format_exc())
         return [("Analysis error", f"Error during analysis: {str(e)}")]
+
         
 def chat_about_image(message, chat_history):
     try: