Update app.py
Browse files
app.py
CHANGED
@@ -79,7 +79,7 @@ def analyze_construction_image(images=None, video=None):
|
|
79 |
|
80 |
if images:
|
81 |
for i, image_file in enumerate(images):
|
82 |
-
image = Image.open(image_file.name)
|
83 |
resized_image = resize_image(image) # Resize image before processing
|
84 |
image_data_url = f"data:image/png;base64,{encode_image(resized_image)}"
|
85 |
messages = [
|
@@ -112,7 +112,7 @@ def analyze_construction_image(images=None, video=None):
|
|
112 |
results.append((f"Image {i+1} analysis", result))
|
113 |
|
114 |
if video:
|
115 |
-
frames = extract_frames_from_video(video
|
116 |
for i, frame in enumerate(frames):
|
117 |
image_data_url = f"data:image/png;base64,{encode_image(frame)}"
|
118 |
messages = [
|
|
|
79 |
|
80 |
if images:
|
81 |
for i, image_file in enumerate(images):
|
82 |
+
image = Image.open(image_file.name) # For image uploads, we use image_file.name
|
83 |
resized_image = resize_image(image) # Resize image before processing
|
84 |
image_data_url = f"data:image/png;base64,{encode_image(resized_image)}"
|
85 |
messages = [
|
|
|
112 |
results.append((f"Image {i+1} analysis", result))
|
113 |
|
114 |
if video:
|
115 |
+
frames = extract_frames_from_video(video) # Use video directly, as it's a file path
|
116 |
for i, frame in enumerate(frames):
|
117 |
image_data_url = f"data:image/png;base64,{encode_image(frame)}"
|
118 |
messages = [
|