Update

app.py

@@ -12,7 +12,7 @@ from langchain_community.agent_toolkits.sql.toolkit import SQLDatabaseToolkit
 from langchain_openai import ChatOpenAI
 
 # Local imports
-from init_db import db_description, Project
+from init_db import Project, db_description, json_structure
 
 # Load environment variables
 load_dotenv()
@@ -24,7 +24,7 @@ if not os.path.exists('estate.db'):
     )
 
 # Initialize model and database
-model = ChatOpenAI(model="gpt-4o-2024-08-06", streaming=True)
+model = ChatOpenAI(model="o3-mini-2025-01-31", streaming=True)
 db = SQLDatabase.from_uri("sqlite:///estate.db")
 
 # Set up SQL toolkit and tools
@@ -51,7 +51,7 @@ LIMIT 5;
 
 # Get the OpenAI tools agent prompt
 prompt = ChatPromptTemplate.from_messages([
-  ("system", "You are a helpful assistant. You speak Czech. You can answer questions about real estate projects (novostavby) in Czech Republic. You have access to a database of real estate projects (translate output to Czech too). Some info about the structure of the database: " + db_description + "Note that some json values can be null so you must sometimes so make sure to add IS NOT NULL when appropriate. Always query projects with structure->'$.ignore'=false or NULL. For calculating you can adapt this query: " + sql_distance_query),
+  ("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}."),
   ("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"Plánuješ koupit novostavbu? Zeptej se mě na {count_projects} novostaveb v ČR 🇨🇿",
+    description=f"Planning to buy a new apartment? Ask me about {count_projects} new buildings (novostavby) in Prague 🇨🇿",
     examples=[
-        "Projekt s nejnižším vkladem",
-        "Nejlevnější byt v Praze",
-        "Nejmenší m2 v Praze"
+        "Project with the lowest deposit before completion",
+        "Cheapest apartment in Prague",
+        "Best price per sqm in Prague"
     ],
     type="messages"
 )

estate.db

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

init_db.py

@@ -4,17 +4,14 @@ 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) for apartments or null if unknown, '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. If they sell house and not apartment, calculate only from the apartment size, not size of garden (land), '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 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"
 
-db = SqliteExtDatabase('estate.db', pragmas=(
-    ('cache_size', -1024 * 64),  # 64MB page-cache.
-    ('journal_mode', 'wal'),  # Use WAL-mode (you should always use this!).
-    ('foreign_keys', 1)))  # Enforce foreign-key constraints.
+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: contents of the website, information about the project.
+    content: unstructured additional information about the project.
     created_at: date and time of creation.
 """