|
|
| from retriever.bm25_retriever import BM25Retriever |
| from qa.biobert_qa import BioBERTAnswerExtractor |
|
|
| def main(): |
| |
| retriever = BM25Retriever("data/medquad_cleaned.json") |
| qa = BioBERTAnswerExtractor() |
|
|
| print("\n🩺 MedBot is ready! Type your question or 'exit' to quit.") |
|
|
| while True: |
| question = input("\nAsk a medical question: ").strip() |
| if question.lower() in {"exit", "quit"}: |
| print("👋 Goodbye!") |
| break |
|
|
| |
| results = retriever.retrieve(question, top_k=3) |
|
|
| |
| print("\n🔍 Best answers:") |
| for idx, item in enumerate(results, 1): |
| context = item["context"] |
| answer = qa.extract_answer(question, context) |
| print(f"\nResult {idx}") |
| print(f"Q: {item['title']}") |
| print(f"A: {answer}") |
| print(f"Source: {item['source']} (BM25 Score: {item['score']:.2f})") |
|
|
| if __name__ == "__main__": |
| main() |
|
|
