Update app.py

app.py

@@ -99,12 +99,8 @@ def analyze_construction_media(media):
                 if file_type in ['jpg', 'jpeg', 'png', 'gif']:
                     # Handle image
                     try:
-                        with Image.open(file_path) as image:
-                            logger.info(f"Successfully opened image: {file_path}")
-                            resized_image = resize_image(image)
-                            image_data = encode_image(resized_image)
-                            logger.info(f"Image {i+1} encoded, size: {len(image_data)} bytes")
-                            image_data_url = f"data:image/png;base64,{image_data}"
+                        image_base64 = encode_image_to_base64(file_path)
+                        logger.info(f"Image {i+1} encoded, size: {len(image_base64)} bytes")
                         
                         messages = [
                             {
@@ -117,7 +113,7 @@ def analyze_construction_media(media):
                                     {
                                         "type": "image_url",
                                         "image_url": {
-                                            "url": image_data_url
+                                            "url": f"data:image/{file_type};base64,{image_base64}"
                                         }
                                     }
                                 ]
@@ -144,9 +140,48 @@ def analyze_construction_media(media):
                         results.append((f"Image {i+1} analysis", f"Error processing image: {str(img_error)}"))
                 
                 elif file_type in ['mp4', 'avi', 'mov', 'wmv']:
-                    # Handle video (keep the existing video processing code)
-                    ...
-
+                    # Handle video
+                    try:
+                        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)
+                            logger.info(f"Video {i+1}, Frame {j+1} encoded, size: {len(frame_base64)} bytes")
+                            
+                            messages = [
+                                {
+                                    "role": "user",
+                                    "content": [
+                                        {
+                                            "type": "text",
+                                            "text": f"{instruction}\n\nAnalyze this frame from a video (File {i+1}/{len(media)}, Frame {j+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",
+                                            "image_url": {
+                                                "url": f"data:image/png;base64,{frame_base64}"
+                                            }
+                                        }
+                                    ]
+                                }
+                            ]
+                            completion = client.chat.completions.create(
+                                model="llama-3.2-90b-vision-preview",
+                                messages=messages,
+                                temperature=0.7,
+                                max_tokens=1000,
+                                top_p=1,
+                                stream=False,
+                                stop=None
+                            )
+                            result = completion.choices[0].message.content
+                            results.append((f"Video {i+1}, Frame {j+1} analysis", result))
+                        logger.info(f"Successfully analyzed video {i+1}")
+                    except Exception as vid_error:
+                        logger.error(f"Error processing video {i+1}: {str(vid_error)}")
+                        logger.error(traceback.format_exc())
+                        results.append((f"Video {i+1} analysis", f"Error processing video: {str(vid_error)}"))
+                
                 else:
                     logger.warning(f"Unsupported file type: {file_type}")
                     results.append((f"File {i+1} analysis", f"Unsupported file type: {file_type}"))