Update src/streamlit_app.py

src/streamlit_app.py

@@ -12,10 +12,6 @@ from sentence_transformers import SentenceTransformer
 from pinecone import Pinecone
 
 
-# os.environ.setdefault("HF_HOME", "/home/user/huggingface_cache")
-# os.environ.setdefault("TRANSFORMERS_CACHE", "/home/user/huggingface_cache/transformers")
-# os.environ.setdefault("SENTENCE_TRANSFORMERS_HOME", "/home/user/huggingface_cache/sentence_transformers")
-
 # Setup (exact hardcoded keys you provided)
 client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
@@ -70,7 +66,9 @@ def generate_response(user_query, docs):
          "If multiple sources are used, separate them with commas. The user should be able to clearly understand where the information came from."}
     ]
     messages.extend(st.session_state.history)
-    messages.append({"role": "user", "content": f"Context:\n{context}\n\nQuestion:\n{user_query}"})
+    messages.append({"role": "user", "content": f"Context:\n{context}\n\n"
+                   f"Sources:\n{', '.join(sources)}\n\n"
+                   f"Question:\n{user_query}"})
 
     try:
         resp = client.chat.completions.create(