Update app.py

app.py

@@ -4,11 +4,23 @@ import os
 import sys
 import subprocess
 import logging
-import copy # Keep copy import for state initialization if needed directly
+import copy
 
-# --- Step 0: Setup Logging Early ---
-# Setup logging configuration first
-from config import setup_logging, LOG_LEVEL, APP_TITLE, FAVICON, LAYOUT, INITIAL_SIDEBAR_STATE, CUSTOM_CSS, LOGO_PATH, FOOTER_HTML
+# --- Step 0: Setup Logging and Page Config EARLY ---
+from config import (
+    setup_logging, LOG_LEVEL, APP_TITLE, FAVICON, LAYOUT,
+    INITIAL_SIDEBAR_STATE, CUSTOM_CSS, LOGO_PATH, FOOTER_HTML
+)
+
+# !!! IMPORTANT: Set page config as the VERY FIRST Streamlit command !!!
+st.set_page_config(
+    page_title=APP_TITLE,
+    layout=LAYOUT,
+    page_icon=FAVICON,
+    initial_sidebar_state=INITIAL_SIDEBAR_STATE
+)
+
+# Now, setup logging
 logger = setup_logging() # Get the configured logger
 
 logger.info("--- RadVision AI Application Start ---")
@@ -17,11 +29,12 @@ logger.info(f"Logging Level: {LOG_LEVEL}")
 
 # --- Step 1: Handle Core Dependencies and Availability ---
 # Try importing essential libraries and set availability flags
+# (Error/Warning messages using st.* inside these blocks are now OK,
+# because set_page_config has already run)
 
 # Deep Translator (Attempt install if missing)
 try:
     from deep_translator import GoogleTranslator
-    # Also try importing our translation wrapper
     from translation_models import translate, detect_language, LANGUAGE_CODES, AUTO_DETECT_INDICATOR
     TRANSLATION_AVAILABLE = True
     logger.info("Deep Translator and translation_models imported successfully.")
@@ -30,15 +43,14 @@ except ImportError:
     try:
         subprocess.check_call([sys.executable, "-m", "pip", "install", "deep-translator"])
         from deep_translator import GoogleTranslator
-        # Re-try importing our wrapper after install
         from translation_models import translate, detect_language, LANGUAGE_CODES, AUTO_DETECT_INDICATOR
         TRANSLATION_AVAILABLE = True
         logger.info("Successfully installed and imported deep-translator and translation_models.")
     except Exception as e:
         logger.error(f"CRITICAL: Failed to install/import deep-translator or translation_models: {e}", exc_info=True)
-        st.warning("Translation features will be disabled: Could not load 'deep-translator'.")
+        # Display warning in the UI now that set_page_config has run
+        st.warning("Translation features are disabled: Could not load or install 'deep-translator'. See logs for details.")
         TRANSLATION_AVAILABLE = False
-        # Define fallbacks if translation module failed
         translate = None
         detect_language = None
         LANGUAGE_CODES = {"English": "en"}
@@ -51,7 +63,7 @@ try:
     DRAWABLE_CANVAS_AVAILABLE = True
     logger.info(f"Streamlit Drawable Canvas Version: {CANVAS_VERSION} available.")
 except ImportError:
-    st.error("CRITICAL ERROR: streamlit-drawable-canvas is not installed. ROI drawing disabled. Run `pip install streamlit-drawable-canvas`.")
+    st.error("CRITICAL ERROR: `streamlit-drawable-canvas` is not installed. ROI drawing is disabled. Run `pip install streamlit-drawable-canvas`.")
     logger.critical("streamlit-drawable-canvas not found. ROI functionality disabled.")
     DRAWABLE_CANVAS_AVAILABLE = False
 
@@ -62,13 +74,12 @@ try:
     PIL_VERSION = getattr(PIL, '__version__', 'Unknown')
     PIL_AVAILABLE = True
     logger.info(f"Pillow (PIL) Version: {PIL_VERSION} available.")
-    # Make PIL Image available to utils module if needed
     import utils
     utils.Image = Image
 except ImportError:
-    st.error("CRITICAL ERROR: Pillow (PIL) is not installed. Image handling disabled. Run `pip install Pillow`.")
+    # This error is critical, stop the app
     logger.critical("Pillow (PIL) not found. Application cannot function correctly.")
-    PIL_AVAILABLE = False
+    st.error("CRITICAL ERROR: Pillow (PIL) is not installed. Image handling is disabled. Run `pip install Pillow`.")
     st.stop() # Stop execution if PIL is missing
 
 # Pydicom and related libraries
@@ -78,19 +89,16 @@ try:
     PYDICOM_VERSION = getattr(pydicom, '__version__', 'Unknown')
     PYDICOM_AVAILABLE = True
     logger.info(f"Pydicom Version: {PYDICOM_VERSION} available.")
-    # Check for optional DICOM helpers
-    try:
-        import pylibjpeg; logger.info("pylibjpeg found (for extended DICOM decompression).")
-    except ImportError: logger.info("pylibjpeg not found. Some DICOM decompression may fail.")
-    try:
-        import gdcm; logger.info("python-gdcm found (for improved DICOM compatibility).")
-    except ImportError: logger.info("python-gdcm not found. Some DICOM features might be limited.")
+    try: import pylibjpeg; logger.info("pylibjpeg found.")
+    except ImportError: logger.info("pylibjpeg not found.")
+    try: import gdcm; logger.info("python-gdcm found.")
+    except ImportError: logger.info("python-gdcm not found.")
 except ImportError:
     PYDICOM_AVAILABLE = False
     PYDICOM_VERSION = 'Not Installed'
     logger.warning("pydicom not found. DICOM file support will be disabled.")
-    pydicom = None # Ensure pydicom object doesn't exist
-
+    st.sidebar.warning("`pydicom` library not found. DICOM file support is disabled.", icon="⚠️") # Inform user in UI
+    pydicom = None
 
 # Custom Utilities & Backend Modules
 try:
@@ -100,38 +108,43 @@ try:
 except ImportError as e:
     logger.error(f"Failed to import dicom_utils: {e}. DICOM specific features disabled.", exc_info=True)
     DICOM_UTILS_AVAILABLE = False
-    if PYDICOM_AVAILABLE: # Only warn if pydicom *is* installed but utils are missing
-        st.warning("DICOM processing utilities (dicom_utils.py) not found. DICOM handling may be limited.")
-
-try:
-    from llm_interactions import run_initial_analysis, run_multimodal_qa, run_disease_analysis, estimate_ai_confidence
-    LLM_INTERACTIONS_AVAILABLE = True
-    logger.info("llm_interactions imported successfully.")
-except ImportError as e:
-    st.error(f"Core AI module (llm_interactions.py) failed to import: {e}. Analysis functions disabled.")
-    logger.critical(f"Failed to import llm_interactions: {e}", exc_info=True)
-    LLM_INTERACTIONS_AVAILABLE = False
-    # Consider st.stop() if core AI is essential and fails to load
-    # st.stop()
+    if PYDICOM_AVAILABLE:
+        st.warning("DICOM processing utilities (`dicom_utils.py`) not found. DICOM handling may be limited.")
 
 try:
+    # This file needs to exist now, even if it's basic
     from report_utils import generate_pdf_report_bytes
     REPORT_UTILS_AVAILABLE = True
     logger.info("report_utils imported successfully.")
 except ImportError as e:
     logger.error(f"Failed to import report_utils: {e}. PDF reporting disabled.", exc_info=True)
     REPORT_UTILS_AVAILABLE = False
-    st.warning("Report generation utility (report_utils.py) not found. PDF reporting disabled.")
+    # Display warning in the UI
+    st.warning("Report generation utility (`report_utils.py`) not found or failed to import. PDF reporting disabled.")
+    # Define a placeholder function to avoid errors in actions.py if it's called
+    def generate_pdf_report_bytes(*args, **kwargs):
+        logger.warning("generate_pdf_report_bytes called, but report_utils is unavailable. Returning None.")
+        return None
+
+
+try:
+    from llm_interactions import run_initial_analysis, run_multimodal_qa, run_disease_analysis, estimate_ai_confidence
+    LLM_INTERACTIONS_AVAILABLE = True
+    logger.info("llm_interactions imported successfully.")
+except ImportError as e:
+    st.error(f"Core AI module (`llm_interactions.py`) failed to import: {e}. AI analysis functions are disabled.")
+    logger.critical(f"Failed to import llm_interactions: {e}", exc_info=True)
+    LLM_INTERACTIONS_AVAILABLE = False
+    # Define placeholders if needed, or let actions.py handle the disabled state
 
 try:
     from ui_components import display_dicom_metadata, dicom_wl_sliders
     UI_COMPONENTS_AVAILABLE = True
     logger.info("ui_components imported successfully.")
 except ImportError as e:
-    logger.warning(f"Failed to import ui_components: {e}. Custom UI elements (like sliders/metadata display) might use fallbacks.", exc_info=True)
+    logger.warning(f"Failed to import ui_components: {e}. Custom UI elements might use fallbacks.", exc_info=True)
     UI_COMPONENTS_AVAILABLE = False
 
-
 # HF Fallback Models (Optional)
 try:
     from hf_models import query_hf_vqa_inference_api, HF_VQA_MODEL_ID
@@ -142,34 +155,24 @@ except ImportError:
     HF_MODELS_AVAILABLE = False
     logger.warning("hf_models not found or failed to import. Fallback VQA disabled.")
 
-# --- Update config with availability flags (optional, if config needs them) ---
+
+# --- Update config with availability flags (optional) ---
 import config
+# (Keep this section as is)
 config.DRAWABLE_CANVAS_AVAILABLE = DRAWABLE_CANVAS_AVAILABLE
-config.PIL_AVAILABLE = PIL_AVAILABLE
-config.PYDICOM_AVAILABLE = PYDICOM_AVAILABLE
-config.DICOM_UTILS_AVAILABLE = DICOM_UTILS_AVAILABLE
-config.LLM_INTERACTIONS_AVAILABLE = LLM_INTERACTIONS_AVAILABLE
-config.REPORT_UTILS_AVAILABLE = REPORT_UTILS_AVAILABLE
-config.UI_COMPONENTS_AVAILABLE = UI_COMPONENTS_AVAILABLE
-config.HF_MODELS_AVAILABLE = HF_MODELS_AVAILABLE
-config.TRANSLATION_AVAILABLE = TRANSLATION_AVAILABLE
-
+# ... etc ...
+config.REPORT_UTILS_AVAILABLE = REPORT_UTILS_AVAILABLE # Update based on import success
 
 # --- Step 2: Configure Streamlit Page ---
-st.set_page_config(
-    page_title=APP_TITLE,
-    layout=LAYOUT,
-    page_icon=FAVICON,
-    initial_sidebar_state=INITIAL_SIDEBAR_STATE
-)
+# Already done at the top!
 
 # Apply Custom CSS
 st.markdown(CUSTOM_CSS, unsafe_allow_html=True)
 
 # --- Step 3: Apply Monkey Patches (Conditional) ---
 import streamlit.elements.image as st_image
-import utils # Import our utils module
-utils.apply_image_to_url_monkey_patch(st_image) # Call the patch function
+import utils
+utils.apply_image_to_url_monkey_patch(st_image)
 
 # --- Step 4: Initialize Session State ---
 from state import initialize_session_state
@@ -178,15 +181,12 @@ logger.debug(f"Session state verified/initialized for session ID: {st.session_st
 
 # --- Step 5: Display Logo ---
 if os.path.exists(LOGO_PATH):
-    # Use columns to control logo width/position if needed
-    logo_col, _ = st.columns([1, 3]) # Adjust ratio as needed
+    logo_col, _ = st.columns([1, 3])
     with logo_col:
-        st.image(LOGO_PATH, width=300) # Adjust width
+        st.image(LOGO_PATH, width=300)
 else:
     logger.warning(f"Hero logo not found at: {LOGO_PATH}")
-    # Optionally display a placeholder text if logo is missing
-    # st.warning("RadVision AI Logo missing.")
-
+    # st.warning("RadVision AI Logo missing.") # Optional UI warning
 
 # --- Step 6: Render Sidebar and Get Uploaded File ---
 from ui import render_sidebar
@@ -194,20 +194,17 @@ uploaded_file = render_sidebar(
     IS_DICOM=st.session_state.get("is_dicom", False),
     UI_COMPONENTS_AVAILABLE=UI_COMPONENTS_AVAILABLE,
     DICOM_UTILS_AVAILABLE=DICOM_UTILS_AVAILABLE,
-    REPORT_UTILS_AVAILABLE=REPORT_UTILS_AVAILABLE
+    REPORT_UTILS_AVAILABLE=REPORT_UTILS_AVAILABLE # Pass the actual availability status
 )
 
-
 # --- Step 7: Handle File Upload Processing ---
 from file_handler import process_uploaded_file
-# Pass availability flags needed by the handler
 process_uploaded_file(uploaded_file, PYDICOM_AVAILABLE, DICOM_UTILS_AVAILABLE, PIL_AVAILABLE)
 
 # --- Step 8: Render Main Page Content ---
 st.markdown("---")
 st.title(f"⚕️ {APP_TITLE}: AI-Assisted Image Analysis")
 
-# User Guide/Disclaimer Expander
 with st.expander("User Guide & Disclaimer", expanded=False):
     st.warning("⚠️ **Disclaimer**: This tool is for research/educational purposes only and is **NOT** a substitute for professional medical advice or diagnosis.")
     st.markdown("""
@@ -223,13 +220,12 @@ with st.expander("User Guide & Disclaimer", expanded=False):
     6.  **Review Results**: Check the tabs ('Initial Analysis', 'Q&A History', 'Condition Focus') for AI output.
     7.  **(Optional) Translation**: Use the 'Translation' tab to translate AI-generated text.
     8.  **(Optional) Confidence**: Use the 'Estimate AI Confidence' button (sidebar) after analysis.
-    9.  **(Optional) Report**: Click 'Generate PDF Report Data' (sidebar) and then 'Download PDF Report' to save a summary.
-    """)
+    9.  **(Optional) Report**: Click 'Generate PDF Report Data' (sidebar) and then 'Download PDF Report' to save a summary. **(Note: PDF generation may be disabled if `report_utils.py` is missing or faulty)**
+    """) # Added note about disabled reporting
 
 st.markdown("---")
 
-# Main layout columns
-col1, col2 = st.columns([2, 3]) # Image column wider slightly
+col1, col2 = st.columns([2, 3])
 
 with col1:
     from ui import render_image_viewer
@@ -243,7 +239,6 @@ with col1:
 
 with col2:
     from ui import render_results_tabs
-    # Pass the actual translation function if available
     translate_func = translate if TRANSLATION_AVAILABLE else None
     render_results_tabs(
         TRANSLATION_SERVICE_AVAILABLE=TRANSLATION_AVAILABLE,
@@ -251,11 +246,11 @@ with col2:
         )
 
 # --- Step 9: Handle Button Actions ---
-# This logic is now encapsulated in actions.py
 from actions import handle_actions
+# Check REPORT_UTILS_AVAILABLE before passing to handle_actions,
+# or handle_actions should already check the flag from session state/config
 handle_actions(LLM_INTERACTIONS_AVAILABLE, REPORT_UTILS_AVAILABLE)
 
-
 # --- Step 10: Render Footer ---
 from ui import render_footer
 render_footer(st.session_state.get("session_id"))