Update app.py

app.py

@@ -2,13 +2,16 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import psycopg2
 import os
-import re
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
 
 # Hugging Face Zephyr model
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 # TimescaleDB config (via Hugging Face Space secrets)
-#connected
 DB_CONFIG = {
     "host": os.getenv("localhost"),
     "port": os.getenv("DB_PORT", 5434),
@@ -17,14 +20,19 @@ DB_CONFIG = {
     "password": os.getenv("password"),
 }
 
-# Query TimescaleDB
+# Query TimescaleDB with improved error handling
 def query_timescaledb(sql_query):
     try:
+        logger.debug("Attempting to connect to the database...")
         with psycopg2.connect(**DB_CONFIG) as conn:
+            logger.debug("Connected to the database successfully.")
             with conn.cursor() as cur:
                 cur.execute(sql_query)
-                return cur.fetchall()
+                result = cur.fetchall()
+                logger.debug("Query executed successfully.")
+                return result
     except Exception as e:
+        logger.error(f"DB Error: {e}")
         return f"DB Error: {e}"
 
 # Basic pattern matching to route question types
@@ -45,7 +53,6 @@ def get_sql_for_question(message):
             AND created_at >= NOW() - INTERVAL '1 day';
         """, "Here is the total number of machines that are not faulty today:"
 
-    
     elif "total current" in message:
         return """
             SELECT created_at, total_current FROM machine_current_log