Update app.py

app.py

@@ -99,7 +99,9 @@ def analyze_construction_media(media):
                 if file_type in ['jpg', 'jpeg', 'png', 'gif']:
                     # Handle image
                     try:
-                        image_base64 = encode_image_to_base64(file_path)
+                        with Image.open(file_path) as img:
+                            img = img.convert('RGB')  # Convert to RGB to ensure compatibility
+                            image_base64 = encode_image(img)
                         logger.info(f"Image {i+1} encoded, size: {len(image_base64)} bytes")
                         
                         messages = [
@@ -113,7 +115,7 @@ def analyze_construction_media(media):
                                     {
                                         "type": "image_url",
                                         "image_url": {
-                                            "url": f"data:image/{file_type};base64,{image_base64}"
+                                            "url": f"data:image/png;base64,{image_base64}"
                                         }
                                     }
                                 ]
@@ -145,7 +147,7 @@ def analyze_construction_media(media):
                         frames = extract_frames_from_video(file_path)
                         logger.info(f"Extracted {len(frames)} frames from video: {file_path}")
                         for j, frame in enumerate(frames):
-                            frame_base64 = encode_image_to_base64(frame)
+                            frame_base64 = encode_image(frame)
                             logger.info(f"Video {i+1}, Frame {j+1} encoded, size: {len(frame_base64)} bytes")
                             
                             messages = [