Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,6 +14,7 @@ from typing import List, Dict, Any, Optional
|
|
| 14 |
from ultralytics import YOLO
|
| 15 |
import piexif
|
| 16 |
import zipfile
|
|
|
|
| 17 |
|
| 18 |
os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
|
| 19 |
logging.basicConfig(filename="app.log", level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
|
|
@@ -299,14 +300,21 @@ def generate_report(
|
|
| 299 |
f"- DAMS Dashboard: `/project_xyz/dams/{datetime.now().strftime('%Y%m%d')}`",
|
| 300 |
"",
|
| 301 |
"## 14. Captured Images",
|
| 302 |
-
"Below are the images from the captured frames directory showing detected issues:",
|
| 303 |
""
|
| 304 |
])
|
| 305 |
|
| 306 |
for image_path in detected_issues:
|
| 307 |
if os.path.exists(image_path):
|
| 308 |
image_name = os.path.basename(image_path)
|
| 309 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 310 |
|
| 311 |
try:
|
| 312 |
with open(report_path, 'w') as f:
|
|
|
|
| 14 |
from ultralytics import YOLO
|
| 15 |
import piexif
|
| 16 |
import zipfile
|
| 17 |
+
import base64
|
| 18 |
|
| 19 |
os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
|
| 20 |
logging.basicConfig(filename="app.log", level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
|
|
|
|
| 300 |
f"- DAMS Dashboard: `/project_xyz/dams/{datetime.now().strftime('%Y%m%d')}`",
|
| 301 |
"",
|
| 302 |
"## 14. Captured Images",
|
| 303 |
+
"Below are the embedded images from the captured frames directory showing detected issues:",
|
| 304 |
""
|
| 305 |
])
|
| 306 |
|
| 307 |
for image_path in detected_issues:
|
| 308 |
if os.path.exists(image_path):
|
| 309 |
image_name = os.path.basename(image_path)
|
| 310 |
+
try:
|
| 311 |
+
with open(image_path, "rb") as image_file:
|
| 312 |
+
base64_string = base64.b64encode(image_file.read()).decode('utf-8')
|
| 313 |
+
report_content.append(f"")
|
| 314 |
+
report_content.append(f"**Image: {image_name}**")
|
| 315 |
+
report_content.append("")
|
| 316 |
+
except Exception as e:
|
| 317 |
+
log_entries.append(f"Error: Failed to encode image {image_name} to base64: {str(e)}")
|
| 318 |
|
| 319 |
try:
|
| 320 |
with open(report_path, 'w') as f:
|