| from rich import print as rprint |
| from rich.panel import Panel |
| from models import Question, FinalAnswer |
| from query_service import QueryService |
| from typing import List |
|
|
| def run_test(): |
| TEST_DOCUMENT_URL = "https://hackrx.blob.core.windows.net/assets/policy.pdf?sv=2023-01-03&st=2025-07-04T09%3A11%3A24Z&se=2027-07-05T09%3A11%3A00Z&sr=b&sp=r&sig=N4a9OU0w0QXO6AOIBiu4bpl7AXvEZogeT%2FjUHNO7HzQ%3D" |
| |
| questions = [ |
| "What is the grace period for premium payment under the National Parivar Mediclaim Plus Policy?", |
| "What is the waiting period for pre-existing diseases (PED) to be covered?", |
| "Does this policy cover maternity expenses, and what are the conditions?", |
| "What is the waiting period for cataract surgery?", |
| "Are the medical expenses for an organ donor covered under this policy?", |
| "What is the No Claim Discount (NCD) offered in this policy?", |
| "Is there a benefit for preventive health check-ups?", |
| "How does the policy define a 'Hospital'?", |
| "What is the extent of coverage for AYUSH treatments?", |
| "Are there any sub-limits on room rent and ICU charges for Plan A?" |
| ] |
|
|
| TEST_QUESTIONS = [Question(question=question) for question in questions] |
|
|
| rprint(Panel( |
| f"Document URL: [blue]{TEST_DOCUMENT_URL}[/blue]\n" |
| f"Questions: [yellow]{len(TEST_QUESTIONS)}[/yellow]", |
| title="[bold green]Starting Insurance Policy Test[/bold green]", |
| border_style="green" |
| )) |
|
|
| query_service = QueryService() |
|
|
| results: List[FinalAnswer] = query_service.process_queries( |
| document_url=TEST_DOCUMENT_URL, |
| questions=TEST_QUESTIONS |
| ) |
| |
| rprint(Panel("[bold green]Processing Complete. Displaying Final Results...[/bold green]")) |
|
|
| for i, response in enumerate(results): |
| question_panel = Panel( |
| f"[bold]Answer:[/] {response.answer}", |
| title=f"[bold magenta]Result for Question #{i+1}[/bold magenta]: {TEST_QUESTIONS[i].question}", |
| border_style="magenta", |
| expand=True |
| ) |
| rprint(question_panel) |
|
|
| if __name__ == "__main__": |
| run_test() |
