Update app.py

app.py

@@ -3,6 +3,7 @@ from huggingface_hub import InferenceClient
 import psycopg2
 import os
 import logging
+from datetime import datetime, timezone
 
 # Set up logging
 logging.basicConfig(level=logging.DEBUG)
@@ -11,17 +12,15 @@ logger = logging.getLogger(__name__)
 # Hugging Face Zephyr model
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# TimescaleDB config (via Hugging Face Space secrets)
+# Use your existing database connection settings
 DB_CONFIG = {
-    "host": os.getenv("DB_HOST", "127.0.0.1"),  # Use the actual DB host
-    "port": os.getenv("DB_PORT", 5434),
-    "database": os.getenv("DB_NAME", "postgres"),
-    "user": os.getenv("DB_USER", "postgres"),
-    "password": os.getenv("DB_PASSWORD", "password"),
+    "host": "127.0.0.1",
+    "port": 5434,
+    "database": "postgres",
+    "user": "postgres",
+    "password": "password"
 }
 
-
-
 # Query TimescaleDB with improved error handling
 def query_timescaledb(sql_query):
     try:
@@ -37,47 +36,51 @@ def query_timescaledb(sql_query):
         logger.error(f"DB Error: {e}")
         return f"DB Error: {e}"
 
-# Basic pattern matching to route question types
+# Modified to match your table structure
 def get_sql_for_question(message):
     message = message.lower()
     
     if "average current" in message:
         return """
-            SELECT AVG(CT_Avg) as avg_current FROM machine_current_log
+            SELECT AVG("CT_Avg") as avg_current 
+            FROM machine_current_log
             WHERE created_at >= NOW() - INTERVAL '1 day';
         """, "Here's the average current over the past 24 hours:"
 
-    elif "total machines not faulty" in message:
+    elif "machine status" in message:
         return """
-            SELECT COUNT(DISTINCT machine_id) 
+            SELECT mac, state, state_duration, fault_status
             FROM machine_current_log
-            WHERE fault_status = 'NOT_FAULTY' 
-            AND created_at >= NOW() - INTERVAL '1 day';
-        """, "Here is the total number of machines that are not faulty today:"
-
-    elif "total current" in message:
+            WHERE created_at = (
+                SELECT MAX(created_at) 
+                FROM machine_current_log
+            )
+            LIMIT 5;
+        """, "Here are the latest machine statuses:"
+
+    elif "current readings" in message:
         return """
-            SELECT created_at, total_current FROM machine_current_log
-            WHERE created_at >= NOW() - INTERVAL '1 day'
-            ORDER BY created_at DESC LIMIT 10;
-        """, "Here are the latest 10 total current readings:"
+            SELECT mac, created_at, "CT1", "CT2", "CT3", "CT_Avg"
+            FROM machine_current_log
+            ORDER BY created_at DESC
+            LIMIT 5;
+        """, "Here are the latest current readings:"
     
-    elif "state duration" in message or "longest running state" in message:
+    elif "fault status" in message:
         return """
-            SELECT state, MAX(state_duration) FROM machine_current_log
-            WHERE created_at >= NOW() - INTERVAL '1 week'
-            GROUP BY state
-            ORDER BY MAX(state_duration) DESC LIMIT 1;
-        """, "Here's the longest running machine state this week:"
+            SELECT fault_status, COUNT(*) 
+            FROM machine_current_log
+            WHERE created_at >= NOW() - INTERVAL '1 day'
+            GROUP BY fault_status;
+        """, "Here's the distribution of fault statuses in the last 24 hours:"
     
-    elif "fault" in message:
+    elif "firmware versions" in message:
         return """
-            SELECT fault_status, COUNT(*) FROM machine_current_log
-            WHERE fault_status IS NOT NULL
-            GROUP BY fault_status
-            ORDER BY COUNT(*) DESC;
-        """, "Here is the frequency of different fault statuses:"
-    
+            SELECT DISTINCT fw_version, COUNT(*) 
+            FROM machine_current_log
+            GROUP BY fw_version;
+        """, "Here are the firmware versions in use:"
+
     return None, None
 
 # Respond using LLM + data if relevant
@@ -118,10 +121,14 @@ def respond(message, history: list[tuple[str, str]], system_message, max_tokens,
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 Wiser AI Assistant")
+    gr.Markdown("## 🤖 Machine Monitoring Assistant")
     gr.Markdown(
         """
-        Welcome to **Wiser's AI Assistant**, your smart companion for all things manufacturing.
+        Welcome to the **Machine Monitoring Assistant**. You can ask questions about:
+        - Current readings (CT1, CT2, CT3, CT_Avg)
+        - Machine status and state duration
+        - Fault status
+        - Firmware versions
         """
     )
     
@@ -129,7 +136,7 @@ with gr.Blocks() as demo:
         respond,
         additional_inputs=[
             gr.Textbox(
-                value="You are Wiser, an expert AI assistant in smart manufacturing. Help users understand machine metrics using the latest database values.",
+                value="You are an expert AI assistant for machine monitoring. Help users understand machine metrics and status using the latest database values.",
                 label="System message"
             ),
             gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
@@ -140,4 +147,4 @@ with gr.Blocks() as demo:
 
 # Run
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()