use duckdb

app.py

@@ -2,6 +2,7 @@ from datetime import datetime, timedelta
 import gradio as gr
 import pandas as pd
 import duckdb
+import logging
 from tabs.trades import (
     prepare_trades, 
     get_overall_trades, 
@@ -26,6 +27,19 @@ from tabs.error import (
 from tabs.about import about_olas_predict
 
 
+def get_logger():
+    logger = logging.getLogger(__name__)
+    logger.setLevel(logging.DEBUG)
+    # stream handler and formatter
+    stream_handler = logging.StreamHandler()
+    stream_handler.setLevel(logging.DEBUG)
+    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+    stream_handler.setFormatter(formatter)
+    logger.addHandler(stream_handler)
+    return logger
+
+logger = get_logger()
+
 def get_last_two_months_data():
     con = duckdb.connect(':memory:')
     two_months_ago = (datetime.now() - timedelta(days=60)).strftime('%Y-%m-%d')
@@ -40,11 +54,12 @@ def get_last_two_months_data():
     FROM read_parquet('./data/all_trades_profitability.parquet')
     WHERE creation_timestamp >= '{two_months_ago}'
     """
-    print("here")
+    logger.info("Query 1: " + query1)
+    logger.info("Query 2: " + query2)
     df2 = con.execute(query2).fetchdf()
-    print("here2")
+    logger.info("here2")
     df1 = con.execute(query1).fetchdf()
-    print("here3")
+    logger.info("here3")
     con.close()
     return df1, df2