Update

app.py

@@ -51,7 +51,7 @@ LIMIT 5;
 
 # Get the OpenAI tools agent prompt
 prompt = ChatPromptTemplate.from_messages([
-  ("system", f"You are a helpful assistant that presents information about real estate projects in database for investors. Before executing queries make surey youhave schema of dabatase and first rows. This JSON structure info can help you: {json_structure}."),
+  ("system", f"You are a helpful assistant that presents information about real estate projects in database for investors. Use 'IS NOT NULL' when necessary. Do not present SQL or JSON. Use simple language. Do not make up information or hallucinate. If final result rows are empty respond that you could not find records matching criteria. Table structure: {db_description}. This JSON structure info can help you: {json_structure}."),
   ("placeholder", "{chat_history}"),
   ("human", "{input}"),
   ("placeholder", "{agent_scratchpad}"),
@@ -91,11 +91,11 @@ count_projects = Project.select().where(Project.structure.is_null(False)).count(
 demo = gr.ChatInterface(
     fn=chat_with_sql,
     title="Estate Chat",
-    description=f"Planning to buy a new apartment? Ask me about {count_projects} new buildings (novostavby) in Prague 🇨🇿",
+    description=f"Planning to buy a new apartment? Ask me about {count_projects} new constructions (novostavby) in Prague 🇨🇿",
     examples=[
-        "Project with the lowest deposit before completion",
         "Cheapest apartment in Prague",
-        "Best price per sqm in Prague"
+        "The lowest deposit before completion",
+        "All 2+kk below 8000000 CZK",
     ],
     type="messages"
 )

estate.db

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eb20aad2ca84e0866daaf3888ddcc304f123dc9671afdd45adc2fa2e0a2036c9
+oid sha256:25a3f76ad7a2856560c2de8067c3e677d86fd6bcaa20f9f7c1353def16289b2d
 size 2560000

init_db.py

@@ -4,19 +4,52 @@ from playhouse.sqlite_ext import *
 
 # Initialize database
 
-json_structure = "JSON column with hash with keys 'title': str, official title of the project, 'deposit': int (in percentage), sum of all deposit payments before the apartment is complete. 'min_price': int, lowest available apartment price in CZK with VAT. It should not be booked or sold., 'average_price_per_sqm': int, average price per square meter of the lowest available apartment. 'status': str, status of the project (preparation, selling, sold out), 'city': str, city of the project, 'lat': float, gps coordinates of the project, 'lng': float, gps coordinates of the project, 'start_year': int, year of construction start, 'end_year': int, estimated year of construction end, 'developer': str, name of the contruction company/developer, 'ignore': bool, if True, the project does not have any apartments for sale. Some values can be null/unknown"
+json_structure = """JSON column with hash with keys
 
+'title': str, official title of the project
+'deposit': int (in percentage), sum of all payments before the apartment is complete ("koladace" requested)
+'cheapest_available_apartments' : hash with keys:
+    '1+kk': int, lowest price for 1+kk with VAT in CZK. It should not be booked or sold
+    '2+kk': int
+    '3+kk': int
+    '4+kk': int
+    '5+kk': int
+    '7+kk': int
+    '8+kk': int
+    '9+kk': int
+    '10+kk': int
+    '1+1': int
+    '2+1': int
+    '3+1': int
+    '4+1': int
+    '5+1': int
+    '6+1': int
+    '7+1': int
+    '8+1': int
+    '9+1': int
+    '10+1': int
+ 'min_price': int, lowest available apartment price in CZK with VAT. It should not be booked or sold.
+ 'status': str, status of the project (preparation, selling, sold out)
+ 'city': str, city of the project
+ 'lat': float, gps coordinates of the project
+ 'lng': float, gps coordinates of the project
+ 'start_year': int, year of construction start
+ 'end_year': int, estimated year of construction end
+ 'developer': str, name of the contruction company/"developer"
+ 'ignore': bool, if True, the project does not have any apartments for sale. Some values can be null/unknown"""
 db = SqliteExtDatabase('estate.db')
 
 db_description = f"""Table "project" - list of real estate projects (novostavby) in Czech Republic
     url: url of the project.
     structure: {json_structure}.
     content: unstructured additional information about the project.
-    created_at: date and time of creation.
+    created_at: date and time of creation
+    title: title of the project
 """
 
 class Project(Model):
     url = CharField(unique=True)
+    title = CharField(null=True)
     structure = JSONField(null=True)
     content = TextField(null=True)
     created_at = DateTimeField(default=datetime.now)