text2sql / core /views.py
ns-devel
Text2SQL app
38171fa
raw
history blame
871 Bytes
"""
"""
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", "")
logger.info(f"Query: {query}")
prompt = get_prompt(query)
logger.info(f"Prompt: {prompt}")
query, data = QueryDataHandler().get_data_from_query(prompt)
return JsonResponse({"status": "success", "query": query, "data": data}, status=200)