Medical GraphRAG Assistant

import pytest from playwright.sync_api import Page, expect import subprocess import time import os import signal # Fixture to start Streamlit app @pytest.fixture(scope="module") def streamlit_app(): # Start Streamlit app process = subprocess.Popen( ["streamlit", "run", "mcp-server/streamlit_app.py", "--server.port", "8501", "--server.headless", "true"], stdout=subprocess.PIPE, stderr=subprocess.PIPE, preexec_fn=os.setsid # To kill the process group later ) # Wait for app to start time.sleep(5) yield process # Teardown: Kill the process group os.killpg(os.getpgid(process.pid), signal.SIGTERM) def test_streamlit_app_loads(page: Page, streamlit_app): """Test that the Streamlit app loads correctly.""" page.goto("http://localhost:8501") # Check title expect(page).to_have_title("Agentic Medical Chat") # Check header expect(page.get_by_role("heading", name="Agentic Medical Chat")).to_be_visible() def test_sidebar_tools(page: Page, streamlit_app): """Test that tools are listed in the sidebar.""" page.goto("http://localhost:8501") # Check for sidebar sidebar = page.locator("[data-testid='stSidebar']") expect(sidebar).to_be_visible() # Check for search_medical_images tool expect(sidebar).to_contain_text("search_medical_images") def test_image_search_demo_mode(page: Page, streamlit_app): """Test image search functionality in demo mode.""" page.goto("http://localhost:8501") # Wait for chat input chat_input = page.get_by_placeholder("Type your message here...") # Streamlit default placeholder? # Actually Streamlit chat input usually has "Your message" or similar. # Let's target by role or generic selector if placeholder is unknown. # Streamlit chat input is usually a textarea. chat_input = page.locator("textarea") expect(chat_input).to_be_visible() # Type query chat_input.fill("Show me chest X-rays of pneumonia") chat_input.press("Enter") # Wait for response # In demo mode, it should say "Found X Images (Demo Mode)" or similar # We added: st.subheader(f"Found {len(images)} Images (Demo Mode)") # Wait for subheader # Note: This might fail if no images are in the DB. # If no images, it returns "No images found matching your query." # We should handle both cases or ensure DB has images. # Assuming DB has images from ingestion (which we haven't run yet in this session, but might persist). # If DB is empty, we expect "No images found". # Let's wait for either result expect(page.locator(".stChatMessage").last).to_be_visible(timeout=10000) # Check content of the last message last_message = page.locator(".stChatMessage").last # It should contain either "Found" or "No images" # We can check for "Images" text # expect(last_message).to_contain_text("Images") # If images are found, check for image elements # st.image renders as <img> tags # We can check if any img tag is present in the chat message # But Streamlit renders images in iframes or specific containers. # Let's just check if the app doesn't crash and returns a response. expect(last_message).not_to_be_empty()