Error Fix

app.py

@@ -36,28 +36,59 @@ os.makedirs(os.path.dirname(DB_PATH), exist_ok=True)
 
 # Define the prompt with examples
 query_prompt = ChatPromptTemplate.from_messages([
-    ("system", """You are an SQL expert. Generate an appropriate SQL query using SQLite syntax for the question provided. The query should be executable and return exactly what was asked for.
+    ("system", """
+            You are an SQL and data analysis expert. Generate an appropriate SQL query using SQLite syntax for the question provided, without any explanations or code comments.
+            Follow SQLite-specific conventions, as shown in the examples below:
+            
+            Example 1:
+            Question: "What is the average fare for trips over 10 miles?"
+            SQL Query: SELECT AVG(fare_amount) FROM taxi_data WHERE trip_distance > 10;
+
+            Example 2:
+            Question: "How many trips were taken in each month?"
+            SQL Query: SELECT strftime('%m', pickup_datetime) AS month, COUNT(*) AS trip_count FROM taxi_data GROUP BY month;
+
+            Example 3:
+            Question: "What is the total fare amount for each driver (medallion) per day?"
+            SQL Query: SELECT DATE(pickup_datetime) AS date, medallion, SUM(fare_amount) AS total_fare FROM taxi_data GROUP BY date, medallion;
+            
+            SQLite-Specific Conventions:
+            
+            1. Date and Time Extraction:
+               - Instead of `EXTRACT(YEAR FROM column)`, use `strftime('%Y', column)` to extract the year.
+               - Example: `SELECT strftime('%Y', pickup_datetime) FROM taxi_data;`
 
-For questions about maximum/highest values, use MAX().
-For minimum/lowest values, use MIN().
-For averages, use AVG().
-For counts, use COUNT().
-For sums, use SUM().
+            2. String Length:
+               - Instead of `CHAR_LENGTH(column)`, use `LENGTH(column)`.
+               - Example: `SELECT LENGTH(passenger_name) FROM taxi_data;`
 
-Examples:
-1. Question: "What is the highest tip_amount in the dataset?"
-   SQL: SELECT MAX(tip_amount) as highest_tip FROM data_tab;
+            3. Regular Expressions:
+               - SQLite does not support `REGEXP`. Use `LIKE` for simple patterns or avoid regular expressions.
+               - Example: `SELECT * FROM taxi_data WHERE passenger_name LIKE 'A%';`
 
-2. Question: "What is the average fare amount?"
-   SQL: SELECT AVG(fare_amount) as average_fare FROM data_tab;
+            4. Window Functions:
+               - For row numbering, use `ROW_NUMBER()` if supported, or simulate with joins.
+               - Example: `SELECT id, ROW_NUMBER() OVER (ORDER BY pickup_datetime) AS row_num FROM taxi_data;`
 
-3. Question: "How many trips are there?"
-   SQL: SELECT COUNT(*) as trip_count FROM data_tab;
+            5. Data Type Casting:
+               - Use `CAST(column AS TYPE)`, but note that SQLite supports limited types.
+               - Example: `SELECT CAST(fare_amount AS INTEGER) FROM taxi_data;`
 
-4. Question: "What are the top 5 highest tip amounts?"
-   SQL: SELECT * FROM data_tab ORDER BY tip_amount DESC LIMIT 5;
+            6. Full Outer Join Workaround:
+               - SQLite doesn’t support `FULL OUTER JOIN`. Combine `LEFT JOIN` and `UNION` for a similar effect.
+               - Example:
+                 ```
+                 SELECT a.*, b.*
+                 FROM table_a a
+                 LEFT JOIN table_b b ON a.id = b.id
+                 UNION
+                 SELECT a.*, b.*
+                 FROM table_a a
+                 RIGHT JOIN table_b b ON a.id = b.id;
+                 ```
 
-Generate only the SQL query, nothing else. Make sure to use the correct table name from the context provided."""),
+            Use these examples and guidelines to generate an SQL query compatible with SQLite syntax for the question provided.
+        """),
     ("human", "{question}")
 ])