core/views.py

"""

"""
import logging
from django.http import JsonResponse
from core.mfrating.score_calculator import MutualFundScorer
from core.text2sql.handler import QueryDataHandler
from core.text2sql.prompt import get_prompt

logger = logging.getLogger(__name__)


def get_scores(request):
    """
    Retrieves scores for mutual funds based on various factors.
    """
    data = MutualFundScorer().get_scores()
    return JsonResponse({"status": "success", "data": data}, status=200)


def get_mf_data(request):
    """
    Retrieves mutual fund data based on user query.
    """
    query = request.GET.get("query", "")
    print("Query: ", query)
    prompt = get_prompt(query)
    print("Prompt: ", prompt)
    query, data = QueryDataHandler().get_data_from_query(prompt)
    return JsonResponse({"status": "success", "query": query, "data": data}, status=200)