Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,10 @@
 import random
 import uuid
 import logging
+from datetime import datetime, timedelta
 import streamlit as st
+from openai import OpenAI
+import os
 
 # ==================================================
 # LOGGING
@@ -13,16 +16,23 @@ logging.basicConfig(
 logger = logging.getLogger("AI-Agent")
 
 # ==================================================
-# RAW DATA (600+ ROWS, IN-MEMORY)
+# OPENAI CLIENT
+# ==================================================
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+
+# ==================================================
+# RAW DATA (600+ ROWS WITH DATE)
 # ==================================================
 @st.cache_data
 def generate_data(rows: int = 600):
     random.seed(42)
+    today = datetime.today().date()
     data = []
 
     for i in range(1, rows + 1):
         data.append({
             "order_id": i,
+            "order_date": today - timedelta(days=random.randint(0, 30)),
             "supplier": random.choice(
                 ["Supplier A", "Supplier B", "Supplier C", "Supplier D"]
             ),
@@ -37,7 +47,7 @@ def generate_data(rows: int = 600):
             )
         })
 
-    logger.info("Generated %d rows of raw data", len(data))
+    logger.info("Generated %d rows of data", len(data))
     return data
 
 DATA = generate_data()
@@ -46,10 +56,9 @@ DATA = generate_data()
 # TOOLS (FUNCTION CALLING)
 # ==================================================
 def tool_get_stats():
-    logger.info("Tool → get_stats")
-
     prices = [x["price"] for x in DATA]
     days = [x["delivery_days"] for x in DATA]
+    logger.info("Tool → get_stats")
 
     return {
         "rows": len(DATA),
@@ -59,24 +68,16 @@ def tool_get_stats():
 
 def tool_query_product(product: str):
     logger.info("Tool → query_product(%s)", product)
+    return [x for x in DATA if x["product"].lower() == product.lower()][:5]
 
-    return [
-        {
-            "order_id": x["order_id"],
-            "supplier": x["supplier"],
-            "quantity": x["quantity"],
-            "price": x["price"],
-            "delivery_days": x["delivery_days"],
-            "status": x["status"],
-        }
-        for x in DATA
-        if x["product"].lower() == product.lower()
-    ][:5]
+def tool_query_last_days(days: int):
+    logger.info("Tool → query_last_days(%d)", days)
+    cutoff = datetime.today().date() - timedelta(days=days)
+    return [x for x in DATA if x["order_date"] >= cutoff][:10]
 
 def tool_create_support_ticket(text: str):
     ticket_id = str(uuid.uuid4())[:8]
     logger.info("Tool → create_support_ticket (%s)", ticket_id)
-
     return {
         "ticket_id": ticket_id,
         "system": "GitHub Issues (mock)",
@@ -87,72 +88,87 @@ def tool_create_support_ticket(text: str):
 # ==================================================
 # SAFETY
 # ==================================================
-def is_dangerous(text: str) -> bool:
+def is_dangerous(text: str):
     blocked = ["delete", "drop", "truncate", "remove"]
     return any(b in text.lower() for b in blocked)
 
 # ==================================================
-# AGENT (DETERMINISTIC, TOOL-BASED)
+# AGENT
 # ==================================================
 def agent(user_input: str):
     logger.info("User input: %s", user_input)
 
     if is_dangerous(user_input):
-        logger.warning("Blocked dangerous operation")
         return "❌ Dangerous operations are not allowed."
 
     text = user_input.lower()
 
-    # -------- FUNCTION CALL ROUTING --------
+    # ---- STATS ----
     if "stats" in text or "summary" in text:
-        stats = tool_get_stats()
+        s = tool_get_stats()
         return (
             f"📊 **Business Overview**\n"
-            f"- Rows: {stats['rows']}\n"
-            f"- Avg Price: ${stats['avg_price']}\n"
-            f"- Avg Delivery Days: {stats['avg_delivery_days']}"
+            f"- Rows: {s['rows']}\n"
+            f"- Avg Price: ${s['avg_price']}\n"
+            f"- Avg Delivery Days: {s['avg_delivery_days']}"
         )
 
+    # ---- LAST N DAYS ----
+    if "last" in text and "day" in text:
+        try:
+            days = int([x for x in text.split() if x.isdigit()][0])
+        except:
+            days = 3
+        rows = tool_query_last_days(days)
+        return rows if rows else f"No data for last {days} days."
+
+    # ---- PRODUCT ----
     if text.startswith("show"):
         product = user_input.replace("show", "").strip()
         rows = tool_query_product(product)
-
         if not rows:
-            return (
-                "No records found for this product.\n\n"
-                "Would you like me to create a support ticket?"
-            )
+            return "No data found. Would you like to create a support ticket?"
         return rows
 
+    # ---- SUPPORT ----
     if "support" in text or "ticket" in text:
-        ticket = tool_create_support_ticket(user_input)
+        t = tool_create_support_ticket(user_input)
         return (
             f"🎫 **Support Ticket Created**\n"
-            f"- ID: {ticket['ticket_id']}\n"
-            f"- System: {ticket['system']}\n"
-            f"- Status: {ticket['status']}"
+            f"ID: {t['ticket_id']}\n"
+            f"System: {t['system']}"
         )
 
-    # -------- FALLBACK (NO LLM) --------
-    logger.info("Fallback response (LLM disabled)")
-
-    return (
-        "🤖 I can help you with:\n"
-        "- database statistics (`stats`)\n"
-        "- product queries (`show tomato`)\n"
-        "- creating support tickets\n\n"
-        "Please try one of the supported actions."
+    # ---- OPENAI FALLBACK ----
+    logger.info("LLM fallback via OpenAI")
+
+    response = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[
+            {
+                "role": "system",
+                "content": (
+                    "You are an AI procurement assistant. "
+                    "You do NOT have direct access to raw data. "
+                    "Suggest support if needed."
+                )
+            },
+            {"role": "user", "content": user_input}
+        ],
+        max_tokens=150,
+        temperature=0.3
     )
 
+    return response.choices[0].message.content
+
 # ==================================================
 # STREAMLIT UI
 # ==================================================
 st.set_page_config(page_title="AI Procurement Agent", layout="wide")
-st.title("🤖 AI Procurement Agent — Single File")
+st.title("🤖 AI Procurement Agent (OpenAI powered)")
 
-# -------- Sidebar: Business Info --------
+# Sidebar
 stats = tool_get_stats()
-
 st.sidebar.header("📊 Business Information")
 st.sidebar.metric("Rows", stats["rows"])
 st.sidebar.metric("Avg Price", f"${stats['avg_price']}")
@@ -161,11 +177,12 @@ st.sidebar.metric("Avg Delivery Days", stats["avg_delivery_days"])
 st.sidebar.markdown("### Sample queries")
 st.sidebar.code(
     "show tomato\n"
+    "last 3 days\n"
     "database stats\n"
     "create support ticket"
 )
 
-# -------- Chat --------
+# Chat
 if "messages" not in st.session_state:
     st.session_state.messages = []