Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,7 +4,6 @@ import logging
 from datetime import datetime, timedelta
 import streamlit as st
 from openai import OpenAI
-import os
 
 # ==================================================
 # LOGGING
@@ -15,11 +14,6 @@ logging.basicConfig(
 )
 logger = logging.getLogger("AI-Agent")
 
-# ==================================================
-# OPENAI CLIENT
-# ==================================================
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-
 # ==================================================
 # RAW DATA (600+ ROWS WITH DATE)
 # ==================================================
@@ -53,13 +47,28 @@ def generate_data(rows: int = 600):
 DATA = generate_data()
 
 # ==================================================
-# TOOLS (FUNCTION CALLING)
+# UTILS: FORMAT RESPONSE AS TEXT
+# ==================================================
+def format_rows_as_text(rows):
+    lines = []
+    for r in rows:
+        lines.append(
+            f"🧾 Order #{r['order_id']} | {r['order_date']}\n"
+            f"   Product: {r['product']}\n"
+            f"   Supplier: {r['supplier']}\n"
+            f"   Qty: {r['quantity']} | Price: ${r['price']}\n"
+            f"   Delivery: {r['delivery_days']} days | Status: {r['status']}\n"
+        )
+    return "\n".join(lines)
+
+# ==================================================
+# TOOLS
 # ==================================================
 def tool_get_stats():
     prices = [x["price"] for x in DATA]
     days = [x["delivery_days"] for x in DATA]
-    logger.info("Tool → get_stats")
 
+    logger.info("Tool → get_stats")
     return {
         "rows": len(DATA),
         "avg_price": round(sum(prices) / len(prices), 2),
@@ -73,17 +82,12 @@ def tool_query_product(product: str):
 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]
+    return [x for x in DATA if x["order_date"] >= cutoff][:5]
 
 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)",
-        "status": "Created",
-        "description": text,
-    }
+    return ticket_id
 
 # ==================================================
 # SAFETY
@@ -95,7 +99,7 @@ def is_dangerous(text: str):
 # ==================================================
 # AGENT
 # ==================================================
-def agent(user_input: str):
+def agent(user_input: str, openai_client: OpenAI):
     logger.info("User input: %s", user_input)
 
     if is_dangerous(user_input):
@@ -107,10 +111,10 @@ def agent(user_input: str):
     if "stats" in text or "summary" in text:
         s = tool_get_stats()
         return (
-            f"📊 **Business Overview**\n"
-            f"- Rows: {s['rows']}\n"
-            f"- Avg Price: ${s['avg_price']}\n"
-            f"- Avg Delivery Days: {s['avg_delivery_days']}"
+            "📊 Business Overview\n"
+            f"- Total rows: {s['rows']}\n"
+            f"- Average price: ${s['avg_price']}\n"
+            f"- Average delivery days: {s['avg_delivery_days']}"
         )
 
     # ---- LAST N DAYS ----
@@ -119,38 +123,43 @@ def agent(user_input: str):
             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."
+        if not rows:
+            return f"No data found for last {days} days."
+
+        return f"📦 Orders from last {days} days:\n\n" + format_rows_as_text(rows)
 
     # ---- PRODUCT ----
     if text.startswith("show"):
         product = user_input.replace("show", "").strip()
         rows = tool_query_product(product)
+
         if not rows:
-            return "No data found. Would you like to create a support ticket?"
-        return rows
+            return "No records found. Would you like me to create a support ticket?"
+
+        return f"📦 Showing orders for **{product}**:\n\n" + format_rows_as_text(rows)
 
     # ---- SUPPORT ----
     if "support" in text or "ticket" in text:
-        t = tool_create_support_ticket(user_input)
+        ticket_id = tool_create_support_ticket(user_input)
         return (
-            f"🎫 **Support Ticket Created**\n"
-            f"ID: {t['ticket_id']}\n"
-            f"System: {t['system']}"
+            "🎫 Support Ticket Created\n"
+            f"- Ticket ID: {ticket_id}\n"
+            "- System: GitHub Issues (mock)"
         )
 
     # ---- OPENAI FALLBACK ----
     logger.info("LLM fallback via OpenAI")
 
-    response = client.chat.completions.create(
+    response = openai_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."
+                    "You do NOT have access to raw data."
                 )
             },
             {"role": "user", "content": user_input}
@@ -165,9 +174,16 @@ def agent(user_input: str):
 # STREAMLIT UI
 # ==================================================
 st.set_page_config(page_title="AI Procurement Agent", layout="wide")
-st.title("🤖 AI Procurement Agent (OpenAI powered)")
+st.title("🤖 AI Procurement Agent (OpenAI)")
+
+# ---- SIDEBAR ----
+st.sidebar.header("🔑 OpenAI Configuration")
+api_key = st.sidebar.text_input(
+    "OpenAI API Key",
+    type="password",
+    help="Paste your OpenAI API key here"
+)
 
-# Sidebar
 stats = tool_get_stats()
 st.sidebar.header("📊 Business Information")
 st.sidebar.metric("Rows", stats["rows"])
@@ -182,7 +198,13 @@ st.sidebar.code(
     "create support ticket"
 )
 
-# Chat
+if not api_key:
+    st.warning("Please enter your OpenAI API key to continue.")
+    st.stop()
+
+openai_client = OpenAI(api_key=api_key)
+
+# ---- CHAT ----
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
@@ -190,7 +212,7 @@ user_input = st.chat_input("Ask the procurement agent...")
 
 if user_input:
     st.session_state.messages.append(("user", user_input))
-    reply = agent(user_input)
+    reply = agent(user_input, openai_client)
     st.session_state.messages.append(("assistant", reply))
 
 for role, message in st.session_state.messages: