wrongmove/vqa/questions.py

37 lines
1.3 KiB
Python

import json
def load_questions(bigquestion=False):
with open("questions.json") as f:
qs: dict = json.load(f)
if bigquestion:
s = """
Tell me from this floor plan information in a valid json format. The json format should have the following format. Only fill out the information if it is visible by the floor plan. Do not try to convert footage or meter into the each other or multiply width with height if area is not given. Instead use null whenever the data is not explicitly displayed.
```
{
"balcony": "boolean",
"terrace": "boolean",
"total_number_of_rooms": "number",
"number_of_bathrooms": "number",
"number_of_bedrooms": "number",
"total_square_meter": "number", // if described
"total_square_footage": "number", // if described
"kitchen_separate": "boolean",
"bedroom_faces_south_or_east": "boolean",
// Per room I also want following information
"rooms": {
"title": "e.g. bedroom 1"
"type": "bedroom|livingroom|bathroom",
"dimension_width_metric": "3", // in meters if exist
"dimension_height_metric": "4", // in meters if exist
"sqm" : "number", // if exists
"dimension_width_foot": "5.5", // in foot if exist
"dimension_height_foot": "7", // in foot if exist
}
}
```
""".strip()
qs[s] = "<not set>"
return qs