Spaces:
Running
Running
File size: 5,734 Bytes
352a4b6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
# Copyright 2025 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
from pathlib import Path
import csv # Import the csv module
import logging
# --- Configuration ---
# Configure basic logging (optional, adjust as needed)
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)
# API Configuration
HF_TOKEN = os.environ.get("HF_TOKEN", None)
MEDGEMMA_ENDPOINT_URL = os.environ.get("MEDGEMMA_ENDPOINT_URL", None)
# --- Paths using pathlib ---
# Determine the base directory of the application
BASE_DIR = Path(__file__).parent.resolve() # Use resolve() for absolute path
STATIC_DIR = BASE_DIR / 'static'
# --- Load available report/image pairs from CSV ---
AVAILABLE_REPORTS = []
MANIFEST_CSV_PATH = STATIC_DIR / 'reports_manifest.csv'
if MANIFEST_CSV_PATH.is_file():
try:
with open(MANIFEST_CSV_PATH, mode='r', encoding='utf-8') as csvfile:
reader = csv.DictReader(csvfile)
# Expected CSV headers: 'image_type', 'case_display_name', 'image_path', 'report_path'
required_headers = {'case_display_name', 'image_path', 'report_path'}
if not required_headers.issubset(reader.fieldnames):
logger.error(
f"CSV file {MANIFEST_CSV_PATH} is missing one or more required headers: {required_headers - set(reader.fieldnames)}"
)
else:
for row in reader:
case_name = row['case_display_name']
image_path_from_csv = row['image_path'] # e.g., static/images/report1.jpg
report_path_from_csv = row['report_path'] # e.g., static/reports/report1.txt or empty
# Validate image_path_from_csv (must not be empty)
if not image_path_from_csv:
logger.warning(f"Empty image_path in CSV for case '{case_name}'. Skipping this entry.")
continue
# Construct absolute path for image file validation (paths from CSV are relative to BASE_DIR)
abs_image_path_to_check = BASE_DIR / image_path_from_csv
if not abs_image_path_to_check.is_file():
logger.warning(f"Image file not found for case '{case_name}' at '{abs_image_path_to_check}'. Skipping this entry.")
continue
image_file_for_config = image_path_from_csv
report_file_for_config = "" # Default to empty if no report or error
if report_path_from_csv: # Report path is optional
# Construct absolute path for report file validation (paths from CSV are relative to BASE_DIR)
abs_report_path_to_check = BASE_DIR / report_path_from_csv
if not abs_report_path_to_check.is_file():
logger.warning(
f"Report file specified for case '{case_name}' at '{abs_report_path_to_check}' not found. "
f"Proceeding without report file for this entry."
)
# report_file_for_config remains ""
else:
# The file (BASE_DIR / report_path_from_csv) exists.
# Now, ensure report_path_from_csv string itself starts with "static/"
# as per the assumption about CSV content.
if report_path_from_csv.startswith('static/') or report_path_from_csv.startswith('static\\'):
# Path is well-formed (starts with static/) and file exists.
# Store the path as is (e.g., "static/reports/report1.txt").
report_file_for_config = report_path_from_csv
else:
logger.warning(
f"Report path '{report_path_from_csv}' for case '{case_name}' in CSV "
f"is malformed (does not start with 'static/'). Treating as if no report path was specified."
)
# report_file_for_config remains ""
AVAILABLE_REPORTS.append({
"name": case_name,
"image_file": image_file_for_config, # static/images/report1.jpg
"report_file": report_file_for_config, # static/reports/report1.txt or ""
"image_type": row['image_type']
})
AVAILABLE_REPORTS.sort(key=lambda x: x['name'])
logger.info(f"Loaded {len(AVAILABLE_REPORTS)} report/image pairs from CSV.")
except Exception as e:
logger.error(f"Error reading or processing CSV file {MANIFEST_CSV_PATH}: {e}", exc_info=True)
else:
logger.warning(f"Manifest CSV file not found at {MANIFEST_CSV_PATH}. AVAILABLE_REPORTS will be empty.")
# --- Optional: Define a default report if needed ---
DEFAULT_REPORT_INFO = AVAILABLE_REPORTS[0] if AVAILABLE_REPORTS else None
|