Update app.py

app.py

@@ -16,7 +16,17 @@ from reportlab.lib.styles import getSampleStyleSheet
 from reportlab.lib.pagesizes import A4
 
 # =========================================================
-# HRMS INTENT CLASSIFICATION (ADVANCED)
+# CHATGPT-STYLE MEMORY (RECENT CONTEXT ONLY)
+# =========================================================
+def build_chat_history(messages, max_turns=6):
+    history = []
+    for msg in messages[-max_turns:]:
+        role = "User" if msg["role"] == "user" else "Assistant"
+        history.append(f"{role}: {msg['content']}")
+    return "\n".join(history)
+
+# =========================================================
+# HRMS INTENT CLASSIFICATION
 # =========================================================
 INTENTS = {
     "LEAVE": ["leave", "holiday", "pto", "sick"],
@@ -44,7 +54,7 @@ st.set_page_config(
 )
 
 # =========================================================
-# PHARMA UI
+# CHATGPT-LIKE PHARMA UI
 # =========================================================
 st.markdown("""
 <style>
@@ -52,10 +62,10 @@ st.markdown("""
 section[data-testid="stSidebar"] { background-color: #ffffff; border-right: 1px solid #e5e7eb; }
 .header-card { background: white; padding: 18px; border-radius: 18px; box-shadow: 0 2px 10px rgba(0,0,0,0.06); margin-bottom: 16px; }
 .role-badge { padding: 6px 14px; border-radius: 999px; font-size: 12px; font-weight: 600; background-color: #e0e7ff; color: #3730a3; }
-section[data-testid="stChatMessage"] { border-radius: 18px; padding: 14px; margin-bottom: 10px; font-size: 15px; }
+section[data-testid="stChatMessage"] { border-radius: 18px; padding: 14px; margin-bottom: 10px; font-size: 15px; line-height: 1.6; }
 div[data-testid="stChatMessage"][aria-label="user"] { background-color: #eef2ff; border-left: 5px solid #4f46e5; }
 div[data-testid="stChatMessage"][aria-label="assistant"] { background-color: #ffffff; border-left: 5px solid #16a34a; }
-textarea { border-radius: 16px !important; }
+textarea { border-radius: 16px !important; font-size: 15px !important; }
 button { border-radius: 12px !important; }
 </style>
 """, unsafe_allow_html=True)
@@ -65,16 +75,17 @@ button { border-radius: 12px !important; }
 # =========================================================
 if "messages" not in st.session_state:
     st.session_state.messages = []
+
 if "vector_ready" not in st.session_state:
     st.session_state.vector_ready = False
 
 # =========================================================
-# SIDEBAR
+# SIDEBAR (MINIMAL – LIKE CHATGPT)
 # =========================================================
 with st.sidebar:
     st.markdown("### 💼 HRMS Chatbot")
 
-    role = st.selectbox("Select Role", ["Employee", "HR", "Manager"])
+    role = st.selectbox("Active Role", ["Employee", "HR", "Manager"])
     st.markdown("---")
 
     uploaded = st.file_uploader("Upload HR Document", type=["pdf", "docx", "txt"])
@@ -92,7 +103,6 @@ with st.sidebar:
         st.session_state.vector_ready = False
         st.rerun()
 
-    # PDF EXPORT
     if st.session_state.messages:
         def generate_pdf():
             buffer = BytesIO()
@@ -111,7 +121,9 @@ with st.sidebar:
 
             for msg in st.session_state.messages:
                 prefix = "Q:" if msg["role"] == "user" else "A:"
-                story.append(Paragraph(f"<b>{prefix}</b> {msg['content']}", styles["Normal"]))
+                story.append(
+                    Paragraph(f"<b>{prefix}</b> {msg['content']}", styles["Normal"])
+                )
                 story.append(Spacer(1, 8))
 
             doc.build(story)
@@ -137,47 +149,51 @@ st.markdown("""
 </div>
 """, unsafe_allow_html=True)
 
-st.markdown(f"**Active Role:** <span class='role-badge'>{role}</span>", unsafe_allow_html=True)
+st.markdown(
+    f"**Active Role:** <span class='role-badge'>{role}</span>",
+    unsafe_allow_html=True
+)
 
 # =========================================================
-# CHAT HISTORY
+# CHAT HISTORY (CHATGPT STYLE)
 # =========================================================
 for msg in st.session_state.messages:
     with st.chat_message(msg["role"]):
         st.markdown(msg["content"])
 
 # =========================================================
-# CHAT INPUT (ADVANCED + STREAMING)
+# CHAT INPUT (ENTER = SEND | SHIFT+ENTER = NEW LINE)
 # =========================================================
-question = st.chat_input("Ask an HRMS-related question...")
+question = st.chat_input(
+    "Message HRMS Chatbot…  (Enter to send • Shift+Enter for new line)"
+)
 
 if question:
+    # ---------------- USER ----------------
     st.session_state.messages.append({"role": "user", "content": question})
     with st.chat_message("user"):
         st.markdown(question)
 
+    # ---------------- ASSISTANT (CHATGPT STYLE) ----------------
     with st.chat_message("assistant"):
-        container = st.empty()
-
-        # Typing dots
-        typing = st.empty()
-        for dots in ["⏳", "⏳.", "⏳..", "⏳..."]:
-            typing.markdown(dots)
-            time.sleep(0.15)
-        typing.empty()
+        response_container = st.empty()
+        response_container.markdown("_Thinking…_")
 
         if not is_hrms_query(question) or not role_allowed(role, question):
             answer = hrms_refusal()
-            container.markdown(answer)
+            response_container.markdown(answer)
         else:
             intent = detect_intent(question)
             retriever = get_retriever()
+
             context_docs = (
                 retriever.get_relevant_documents(intent)
                 if retriever and st.session_state.vector_ready
                 else []
             )
 
+            chat_history = build_chat_history(st.session_state.messages)
+
             final_prompt = f"""
 {SYSTEM_PROMPT}
 
@@ -187,24 +203,29 @@ ROLE CONTEXT:
 INTENT:
 {intent}
 
-INSTRUCTIONS:
-- Start with a short overview
-- Explain key rules
-- End with role responsibility
-- Do not make decisions
+CONVERSATION HISTORY:
+{chat_history}
 
 HR DOCUMENT CONTEXT:
 {context_docs}
 
-QUESTION:
+USER QUESTION:
 {question}
+
+RESPONSE GUIDELINES:
+- Use clear Markdown formatting
+- Use bullet points and headings when helpful
+- Be concise but complete
+- Stay strictly within HRMS scope
 """
 
-            stream_handler = StreamHandler(container)
+            stream_handler = StreamHandler(response_container)
             llm = get_llm(callbacks=[stream_handler])
             llm.invoke(final_prompt)
-            answer = stream_handler.text
 
-        st.markdown("<script>window.scrollTo(0, document.body.scrollHeight);</script>", unsafe_allow_html=True)
+            answer = stream_handler.text
 
-    st.session_state.messages.append({"role": "assistant", "content": answer})
+    # ---------------- SAVE ASSISTANT MESSAGE ----------------
+    st.session_state.messages.append(
+        {"role": "assistant", "content": answer}
+    )