from fastapi import FastAPI, UploadFile,File
from src import pipeline
from src.pipeline import QAPipeline
from langchain_community.document_loaders import CSVLoader
import shutil
import os 
pipeline= QAPipeline()

app= FastAPI()

@app.post("/")
async def upload_documents(file:UploadFile=File(...)):
    """
    Create a new collection in Qdrant.
    """
    try:
        if not file.filename.endswith('.csv'):
            return {"error": "The uploaded file is not a CSV file."}
        temp_path = f"/tmp/{file.filename}"
        with open(temp_path, "wb") as buffer:
            shutil.copyfileobj(file.file, buffer)
        loader = CSVLoader(file_path=temp_path)
        data = loader.load()
        await pipeline.upload_documents(data)
        os.remove(temp_path)
        return {"message": "Documents uploaded successfully."}
        
    except Exception as e:
        raise e

@app.get("/answer")
async def answer_query(query:str):
    """
    Answer a query using the Groq model.
    """
    try:
        response = await pipeline.answer_query_(query)
        return {"response": response}
    except Exception as e:
        raise e