import os
import chainlit as cl
from dotenv import load_dotenv
from rag_dspy import MedicalAnswer, rerank_with_colbert, MedicalRAG
import dspy
from dspy_qdrant import QdrantRM
from qdrant_client import QdrantClient

# Configure Chainlit
rag_chain = MedicalRAG()

@cl.on_message
async def main(message: cl.Message):
    user_question = message.content
    result = rag_chain.forward(user_question)
    result = result.final_answer
    await cl.Message(content=result).send()