main.py

import json
from utils import describe_json, generate_response, execute_api_code, colored, install_missing_libraries

with open('sample.json', 'r') as f:
    data = json.load(f)
    
descr = describe_json(data)



user_query = input("ask your query: ")


prompt = f'''
So we have a json whose structure's can be described with the following struct:
{descr}
 
You are an AI developer who is trying to write a program that will generate code for the user based on their intent. only write valid code. Do not include code fences in your response, for example
Bad response:
```python
print("hello world")


Good response:
print("hello world")

    
Our Goal is to find the proper answer to the user's query in the format the user wants (markdown and use tables where it will be better)
Use json's description to write the python code to find the answer to the user query.
Have proper print messages. 
Assume that you already have a `data` variable which has the json in string



Only If there is String Matching required: never do absolute string matches, use fuzzywuzzy python lib  if no string matching required then ignore this rule
the code is for: {user_query}


Remember these rules that you have to obey:
1) Only give code, no explainations or anything, Only Code
2) the purpose for the code is to print the answer for {user_query}
3) Write the code only in python

Begin generating the code now.

'''



code = f"data = \'\'\'{json.dumps(data)}\'\'\'\n"+generate_response(prompt=prompt)
install_missing_libraries(code)
output = execute_api_code(code=code)
print(colored(output, "green"))