Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,33 +3,33 @@ import json
 import streamlit as st
 from transformers import pipeline
 
-# ✅ Set custom cache directory to avoid permission errors
-os.environ['HF_HOME'] = './hf_cache'
-os.environ['TRANSFORMERS_CACHE'] = './hf_cache'
+# Set environment variables for Hugging Face cache
+os.environ['HF_HOME'] = os.path.expanduser('~/.cache/huggingface')
+os.environ['TRANSFORMERS_CACHE'] = os.environ['HF_HOME']
 
-# ✅ Streamlit config
+# Streamlit configuration
 st.set_page_config(page_title="SmartProcureAI", layout="wide")
 
-# ✅ Load local JSON data
-with open("src/sap_data.json") as f:
+# Load data
+with open("sap_data.json") as f:
     sap_data = json.load(f)
 
-# ✅ Load lightweight GPT-2 model using the safe cache
+# Initialize the text generation pipeline
 summarizer = pipeline("text-generation", model="gpt2", max_new_tokens=150)
 
-# ✅ App title
+# App title
 st.title("📦 SmartProcureAI: GenAI-Powered SAP Procurement Assistant")
 
-# ✅ Sidebar input form
+# Sidebar inputs
 st.sidebar.header("📄 RFQ Generator")
 product_type = st.sidebar.selectbox("Select Product", [p["name"] for p in sap_data["products"]])
 budget = st.sidebar.number_input("Enter Budget (INR)", value=50000)
 
-# ✅ Fetch selected product and vendor
+# Retrieve selected product and vendor
 product = next(p for p in sap_data["products"] if p["name"] == product_type)
 vendor = next(v for v in sap_data["vendors"] if v["id"] == product["vendor_id"])
 
-# ✅ Prompt template
+# Generate prompt
 prompt = f"""
 Write a professional RFQ email for the following procurement need:
 - Product: {product['name']}
@@ -40,14 +40,14 @@ Write a professional RFQ email for the following procurement need:
 Keep it concise and formal.
 """
 
-# ✅ Generate RFQ
+# Generate RFQ
 if st.sidebar.button("📝 Generate RFQ"):
     with st.spinner("Generating RFQ..."):
         output = summarizer(prompt)[0]["generated_text"]
         st.subheader("📄 Generated RFQ:")
         st.code(output)
 
-# ✅ Vendor info
+# Display vendor information
 st.sidebar.markdown("---")
 st.sidebar.subheader("📊 Vendor Summary")
 if st.sidebar.button("🔍 Show Vendor Info"):