mlnsio/text2sql

1"""2 3"""4import logging5from django.http import JsonResponse6from core.mfrating.score_calculator import MutualFundScorer7from core.text2sql.handler import QueryDataHandler8from core.text2sql.prompt import get_prompt9 10logger = logging.getLogger(__name__)11 12 13def get_scores(request):14    """15    Retrieves scores for mutual funds based on various factors.16    """17    data = MutualFundScorer().get_scores()18    return JsonResponse({"status": "success", "data": data}, status=200)19 20 21def get_mf_data(request):22    """23    Retrieves mutual fund data based on user query.24    """25    query = request.GET.get("query", "")26    print("Query: ", query)27    prompt = get_prompt(query)28    print("Prompt: ", prompt)29    query, data = QueryDataHandler().get_data_from_query(prompt)30    return JsonResponse({"status": "success", "query": query, "data": data}, status=200)31