Update app.py

app.py

@@ -5,6 +5,13 @@ from PIL import Image
 import io # Needed for handling image bytes
 from typing import Optional, Tuple, Any # For type hinting
 
+# --- Page Configuration (MUST BE THE FIRST STREAMLIT COMMAND) ---
+st.set_page_config(
+    page_title="AI Clinical Support Demonstrator",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
 # --- Configuration and Initialization ---
 
 # Securely load API key
@@ -18,6 +25,8 @@ if GEMINI_API_KEY:
         genai.configure(api_key=GEMINI_API_KEY)
         genai_client_configured = True
     except Exception as e:
+        # Use st.exception() for a more detailed error in the app if needed during debugging
+        # st.exception(e)
         st.error(f"Fatal Error: Failed to configure Google Generative AI. Check API Key. Details: {e}")
         st.stop() # Stop execution if configuration fails
 else:
@@ -25,7 +34,6 @@ else:
     st.stop() # Stop execution if no API key
 
 # Initialize models (Consider more powerful model for agentic reasoning if needed)
-# Using 1.5 Pro for text (agentic) and 1.5 Flash for vision might be a good balance
 TEXT_MODEL_NAME = 'gemini-1.5-pro-latest' # Model for agentic text reasoning
 VISION_MODEL_NAME = 'gemini-1.5-flash' # Model for image analysis
 
@@ -33,11 +41,14 @@ if genai_client_configured:
     try:
         text_model = genai.GenerativeModel(TEXT_MODEL_NAME)
         vision_model = genai.GenerativeModel(VISION_MODEL_NAME)
-        st.success(f"Successfully initialized models: Text ({TEXT_MODEL_NAME}), Vision ({VISION_MODEL_NAME})", icon="✅")
+        # Optional: Display success message (won't violate the rule as set_page_config was first)
+        # st.sidebar.success(f"Models Initialized ({TEXT_MODEL_NAME}, {VISION_MODEL_NAME})", icon="✅")
     except Exception as e:
         st.error(f"Fatal Error: Failed to initialize Gemini models. Text: {TEXT_MODEL_NAME}, Vision: {VISION_MODEL_NAME}. Details: {e}")
         st.stop()
 else:
+    # This state should not be reached due to earlier st.stop() calls,
+    # but included for completeness.
     st.error("AI Models could not be initialized due to configuration issues.")
     st.stop()
 
@@ -132,8 +143,10 @@ def run_agentic_text_analysis(text_input: str) -> Tuple[Optional[str], Optional[
                  return None, "Received an empty or unexpected response from the AI model."
 
     except Exception as e:
-        st.error(f"Error during agentic text analysis: {e}", icon="🚨")
-        return None, f"Error communicating with the AI model for text analysis. Details: {e}"
+        # Log error for debugging purposes server-side if possible
+        print(f"ERROR in run_agentic_text_analysis: {e}") # Simple print for demo
+        st.error(f"Error during agentic text analysis. Check logs if possible.", icon="🚨") # Keep UI error generic
+        return None, f"An internal error occurred during text analysis." # Generic message to user
 
 def analyze_medical_image(image_file: Any, user_prompt: str = "") -> Tuple[Optional[str], Optional[str]]:
     """
@@ -173,18 +186,16 @@ def analyze_medical_image(image_file: Any, user_prompt: str = "") -> Tuple[Optio
                  return None, "Received an empty or unexpected response from the AI model for image analysis."
 
     except Exception as e:
-        st.error(f"Error during image analysis: {e}", icon="🖼️")
-        return None, f"Error communicating with the AI model for image analysis. Details: {e}"
+        # Log error for debugging purposes server-side if possible
+        print(f"ERROR in analyze_medical_image: {e}") # Simple print for demo
+        st.error(f"Error during image analysis. Check logs if possible.", icon="🖼️") # Keep UI error generic
+        return None, f"An internal error occurred during image analysis." # Generic message to user
 
 
 # --- Streamlit User Interface ---
 
 def main():
-    st.set_page_config(
-        page_title="AI Clinical Support Demonstrator",
-        layout="wide",
-        initial_sidebar_state="expanded"
-        )
+    # REMOVED st.set_page_config() FROM HERE
 
     st.title("🤖 AI Clinical Support Demonstrator")
     st.caption(f"Agentic Text Analysis ({TEXT_MODEL_NAME}) & Image Analysis ({VISION_MODEL_NAME})")
@@ -259,6 +270,7 @@ def main():
 
             if analyze_button:
                 if image_file:
+                    # Using st.image within the column context
                     st.image(image_file, caption="Uploaded Image Preview", use_column_width=True)
                     with st.spinner("👁️ Analyzing image..."):
                         analysis_result, error_message = analyze_medical_image(image_file, user_image_prompt)
@@ -269,17 +281,24 @@ def main():
     # --- Output Column ---
     with col2:
         st.header(output_header)
-        # Check if a button was pressed in this run (using session state keys is more robust for complex apps,
-        # but checking the result variables works here as they are reset on each run unless persisted).
-        button_pressed = st.session_state.get('analyze_text_button', False) or st.session_state.get('analyze_image_button', False)
-
-        if button_pressed: # Only show results if a button was pressed in this run cycle
+        # Determine if analysis was attempted in this run
+        # Use session state keys which are set when buttons are clicked
+        button_pressed = st.session_state.get('analyze_text_button', False) or \
+                         st.session_state.get('analyze_image_button', False)
+
+        if button_pressed:
+            # Check if the button was clicked AND we have results or an error from that action
+            # analysis_result and error_message are set within the button's if block in col1
             if error_message:
                  st.error(f"Analysis Failed: {error_message}", icon="❌")
             elif analysis_result:
                  st.markdown(analysis_result) # Display the successful result
-            # Removed the "unknown error" case here, as the functions should return either result or error message
-        else:
+            # If button was pressed but neither error nor result is set, maybe input was invalid
+            # The warnings for empty input are handled in col1, so this case might not be needed
+            # else:
+            #    st.info("Provide valid input and click Analyze.") # Fallback?
+
+        else: # No button pressed in this run
             st.info("Analysis results will appear here after providing input and clicking the corresponding analysis button.")
 
 
@@ -298,5 +317,9 @@ def main():
         "**Ethical Use Reminder:** AI in medicine requires extreme caution. This tool is for demonstration and education, not clinical practice. Verify all information and rely on professional expertise."
         )
 
+# --- Main Execution Guard ---
 if __name__ == "__main__":
+    # Ensure configuration happens before main UI logic runs
+    # NOTE: The config/init code now runs automatically when the script is imported/run,
+    # BEFORE the main() function is called here. This is correct.
     main()