Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,149 @@
-import altair as alt
-import numpy as np
-import pandas as pd
+import os
 import streamlit as st
+from openai import OpenAI
 
+# ------------------------------
+# CONFIG
+# ------------------------------
+# Hugging Face: Settings -> Variables -> add OPENAI_API_KEY
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") or st.secrets.get("OPENAI_API_KEY", None)
+
+st.set_page_config(
+    page_title="Procelevate AI Agent",
+    page_icon="🤖",
+    layout="wide"
+)
+
+st.title("🤖 Procelevate Agent Suite")
+st.write("PO Automation + Email Responder (Streamlit, HF Docker)")
+
+if not OPENAI_API_KEY:
+    st.warning("⚠️ OPENAI_API_KEY not found. Set it in the Space secrets.")
+client = OpenAI(api_key=OPENAI_API_KEY) if OPENAI_API_KEY else None
+
+
+def call_llm(system_prompt: str, user_prompt: str) -> str:
+    """Small helper to call OpenAI."""
+    if client is None:
+        return "❗OPENAI_API_KEY not set."
+    resp = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_prompt},
+        ],
+        temperature=0.3,
+    )
+    return resp.choices[0].message.content
+
+
+# ------------------------------
+# PO GENERATOR
+# ------------------------------
+def generate_po(buyer_name, buyer_address, supplier_name, supplier_email, items_text, payment_terms, notes):
+    system_prompt = """You are a Purchase Order (PO) generation assistant for Procelevate.
+You will take supplier details, buyer details, item list, and payment terms, and output a clean, well-structured PO.
+Output in markdown with clear headings, table for line items, and totals.
+If any field is missing, make reasonable placeholders."""
+    user_prompt = f"""
+Create a Purchase Order with the following details:
+
+Buyer:
+- Name: {buyer_name}
+- Address: {buyer_address}
+
+Supplier:
+- Name: {supplier_name}
+- Email: {supplier_email}
+
+Line Items (parse this and make a table with Item, Qty, Unit Price, Line Total):
+{items_text}
+
+Payment Terms: {payment_terms}
+Additional Notes: {notes}
+
+At the end, show Subtotal, Taxes (assume 18% if not given), and Grand Total.
 """
-# Welcome to Streamlit!
+    return call_llm(system_prompt, user_prompt)
+
+
+# ------------------------------
+# EMAIL RESPONDER
+# ------------------------------
+def generate_email_reply(incoming_email, reply_goal, tone):
+    system_prompt = """You are an AI email responder for Procelevate Consulting & Academy.
+You will read an incoming email and write a professional, concise reply matching the selected tone.
+Always add a polite closing with 'Regards, Procelevate Team' unless user says otherwise."""
+    user_prompt = f"""
+Incoming email:
+\"\"\"{incoming_email}\"\"\"
 
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
+Goal of reply: {reply_goal}
+Tone: {tone}
 
-In the meantime, below is an example of what you can do with just a few lines of code:
+Write the reply email body only.
 """
+    return call_llm(system_prompt, user_prompt)
+
+
+# ------------------------------
+# UI
+# ------------------------------
+tab1, tab2 = st.tabs(["📄 PO Automation", "📧 Email Responder"])
+
+with tab1:
+    st.subheader("📄 Generate Purchase Order")
+    col1, col2 = st.columns(2)
+
+    with col1:
+        buyer_name = st.text_input("Buyer Name", value="Procelevate Consulting")
+        buyer_address = st.text_area("Buyer Address", value="Bengaluru, Karnataka, India", height=80)
+        payment_terms = st.text_input("Payment Terms", value="Net 30 days")
+    with col2:
+        supplier_name = st.text_input("Supplier Name", value="")
+        supplier_email = st.text_input("Supplier Email", value="")
+        notes = st.text_area("Additional Notes", value="Please deliver services as per agreed SOW.", height=80)
+
+    items_text = st.text_area(
+        "Line Items (one per line)",
+        value="1. AI/ML Bootcamp Seats, Qty 10, Unit Price 2999\n2. Consulting Hours, Qty 5, Unit Price 1500",
+        height=120
+    )
+
+    if st.button("Generate PO"):
+        po_md = generate_po(
+            buyer_name,
+            buyer_address,
+            supplier_name,
+            supplier_email,
+            items_text,
+            payment_terms,
+            notes
+        )
+        st.markdown("---")
+        st.markdown("### ✅ Generated PO")
+        st.markdown(po_md)
+
+with tab2:
+    st.subheader("📧 Generate Email Reply")
+    incoming_email = st.text_area(
+        "Incoming email",
+        value="Hi Mahesh, we are interested in your Smart Daycare Suite. Can you share pricing and a demo slot?",
+        height=140
+    )
+    reply_goal = st.text_input(
+        "What should the reply do?",
+        value="Acknowledge, share that we have multiple AI/ML modules, and propose 30-min demo."
+    )
+    tone = st.selectbox(
+        "Tone",
+        options=["Professional", "Friendly", "Formal-Client", "Follow-up/Reminder"],
+        index=0
+    )
 
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+    if st.button("Generate Reply"):
+        reply_text = generate_email_reply(incoming_email, reply_goal, tone)
+        st.markdown("---")
+        st.markdown("### ✅ Reply Email")
+        st.markdown(reply_text)
+        st.info("You can copy-paste this into Gmail / Outlook.")