Function get_census_data_get

api/app/api/chat_gpt_plugin.py:7–41

Get census data based on the question

()

@plugin.route('get_census_data', methods=['GET'])
def get_census_data_get():
  """
  Get census data based on the question
  """

  question = request.args.get('question')
  if not question:
    return jsonify({
      "error": "question is missing from the request args"
    })

  raw_table_json = requests.post('https://text-sql-be.onrender.com/api/get_tables', json={"natural_language_query": question})

  print('raw table: ', raw_table_json.json())

  newJson = {
      **raw_table_json.json(),
      "natural_language_query": question,
  }

  print('new json: ', newJson)

  final_res = requests.post('https://text-sql-be.onrender.com/api/text_to_sql', json=newJson)

  print('final res: ', final_res.json())

  parsed = final_res.json()
  
  resultsData = parsed['result']['results']
  sqlData = parsed['sql_query']

  return jsonify({
    "answer": resultsData,
    "sql_query": sqlData,
  })


plugin_config = Blueprint('plugin_config', __name__)