Update app.py

app.py

@@ -67,16 +67,25 @@ def text_to_sql(query: str, schema: str) -> str:
         return f"Error: {e}"
 
 
-# Function: Execute SQL on SQLite database
+def preprocess_sql_for_sqlite(sql_query: str) -> str:
+    """
+    Replace non-SQLite functions with SQLite-compatible equivalents.
+    """
+    sql_query = re.sub(r"\bMONTH\s*\(\s*([\w.]+)\s*\)", r"strftime('%m', \1)", sql_query)
+    sql_query = re.sub(r"\bYEAR\s*\(\s*([\w.]+)\s*\)", r"strftime('%Y', \1)", sql_query)
+    return sql_query
+
 def execute_sql(sql_query: str) -> Tuple[Optional[pd.DataFrame], Optional[str]]:
     try:
         conn = sqlite3.connect(DB_PATH)
+        sql_query = preprocess_sql_for_sqlite(sql_query)  # Convert to SQLite-compatible SQL
         df = pd.read_sql_query(sql_query, conn)
         conn.close()
         return df, None
     except Exception as e:
         return None, f"SQL Execution Error: {e}"
 
+
 # Function: Generate Dynamic Visualization
 def visualize_data(df: pd.DataFrame) -> Optional[str]:
     if df.empty or df.shape[1] < 2: