File size: 857 Bytes
38171fa
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8b9a027
38171fa
8b9a027
38171fa
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
"""

"""
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)