Update

app.py

@@ -1,14 +1,19 @@
-from langchain import hub
-from langchain.agents import AgentExecutor, create_openai_tools_agent
-from langchain_openai import ChatOpenAI
-from langchain_community.utilities import SQLDatabase
-from langchain_community.agent_toolkits.sql.toolkit import SQLDatabaseToolkit
+# Standard library imports
+import os.path
+
+# Third-party imports
 import gradio as gr
 from dotenv import load_dotenv
-from langchain.schema import HumanMessage, AIMessage
+from langchain.agents import AgentExecutor, create_openai_tools_agent
 from langchain.prompts import ChatPromptTemplate
-import os.path
+from langchain.schema import HumanMessage, AIMessage
+from langchain_community.utilities import SQLDatabase
+from langchain_community.agent_toolkits.sql.toolkit import SQLDatabaseToolkit
+from langchain_openai import ChatOpenAI
+
+# Local imports
 from init_db import db_description, Project
+
 # Load environment variables
 load_dotenv()
 
@@ -82,7 +87,7 @@ def chat_with_sql(message, history):
         yield f"Error: {str(e)}"
 
 # Create the Gradio interface
-count_projects = Project.select().count()
+count_projects = Project.select().where(Project.structure.is_null(False)).count()
 demo = gr.ChatInterface(
     fn=chat_with_sql,
     title="Estate Chat",

estate.db

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8925612016f3177d10cee1e10b3d17e4c6f54eb0b3973f88c13f613405ed2d99
-size 606208
+oid sha256:214270678ea85f7102ea9ce460d33c9ba0fe3226c3c4087bc302cc2e7666b030
+size 2527232

init_db.py

@@ -4,7 +4,7 @@ from playhouse.sqlite_ext import *
 
 # Initialize database
 
-json_structure = "JSON column with hash with keys 'title': str, official title of the project, 'deposit': int, initial deposit (in percentage), 'min_price': int, lowest available property price with VAT. It should not be booked or sold., '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. Some values can be null/unknown"
+json_structure = "JSON column with hash with keys 'title': str, official title of the project, 'deposit': int, initial deposit (in percentage) for apartments, 'min_price': int, lowest available apartment price 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. Some values can be null/unknown"
 
 db = SqliteExtDatabase('estate.db', pragmas=(
     ('cache_size', -1024 * 64),  # 64MB page-cache.