Limiting chatbot

chatbot.py

@@ -15,11 +15,11 @@ class CodeAssistantBot:
         self.client = Groq(api_key=GROQ_API_KEY)
         self.model = ChatGroq(model="llama-3.3-70b-versatile", temperature=0.6)
         self.analysis_prompt = ChatPromptTemplate.from_messages([
-            ("system", 
+            ("system",
              "You are a skilled coding assistant. Use the following context and user input to help."
-             " Refer to previous summary and recent interactions to make answers accurate. "
-             "Keep your response short, relevant, and conversational — smaller is better if it covers what's needed."),
-            ("user", 
+             " Refer to previous summary and recent interactions to make answers accurate."
+             " Keep your response short, relevant, and conversational — smaller is better if it covers what's needed."),
+            ("user",
              "Code: {code}\nOutput: {output}\nError: {error}\n"
              "Summary: {summary}\nRecent: {recent}\nQuestion: {question}")
         ])
@@ -51,6 +51,7 @@ async def text_to_speech(text, filename):
 def render_chatbot(code, output, error):
     """Render the chatbot UI with code-block support and a scrollable chat container."""
 
+    # Style
     st.markdown("""
     <style>
     .chat-container {
@@ -80,16 +81,20 @@ def render_chatbot(code, output, error):
     </style>
     """, unsafe_allow_html=True)
 
+    # Session state setup
     st.session_state.setdefault('conversation', [])
     st.session_state.setdefault('audio_count', 0)
     st.session_state.setdefault('chat_summary', "")
+    st.session_state.setdefault('chat_display_count', 5)
 
+    # Input UI
     c1, c2 = st.columns([4, 1], gap='small')
     with c1:
         question = st.text_input("Ask your question…", key="chat_input")
     with c2:
         send = st.button("🚀")
 
+    # Process question
     if send and question:
         bot = CodeAssistantBot()
         history = st.session_state.conversation[-4:]
@@ -98,18 +103,30 @@ def render_chatbot(code, output, error):
         response = bot.analyze_code(code, output, error, question, summary, history)
 
         st.session_state.conversation.append((question, response))
+        st.session_state.chat_display_count = 5  # Reset visible count for new message
 
-        # Optionally update running summary
+        # Update summary (based on last 10 entries)
         if len(st.session_state.conversation) >= 3:
-            full_chat = "\n".join([f"User: {q}\nBot: {a}" for q, a in st.session_state.conversation[-10:]])
             try:
+                full_chat = "\n".join([f"User: {q}\nBot: {a}" for q, a in st.session_state.conversation[-10:]])
                 parser = StrOutputParser()
                 summarizer = bot.summary_prompt | bot.model | parser
                 st.session_state.chat_summary = summarizer.invoke({'conversation': full_chat})
             except:
-                pass  # Summary failure shouldn't break the UI
+                pass  # Don't fail on summary issues
 
-    for q, a in reversed(st.session_state.conversation):
+    # Display conversation (newest at top, paginated)
+    total_messages = len(st.session_state.conversation)
+    start_idx = max(0, total_messages - st.session_state.chat_display_count)
+    visible_conversation = st.session_state.conversation[start_idx:]
+
+    if start_idx > 0:
+        if st.button("🔽 Show more"):
+            st.session_state.chat_display_count += 5
+            st.experimental_rerun()
+
+    # Newest first
+    for q, a in reversed(visible_conversation):
         st.markdown(f'<div class="chat-message user-message">{escape(q)}</div>', unsafe_allow_html=True)
 
         def format_response(text):
@@ -129,8 +146,7 @@ def render_chatbot(code, output, error):
         formatted = format_response(a)
         st.markdown(f'<div class="chat-message bot-message">{formatted}</div>', unsafe_allow_html=True)
 
-    st.markdown('</div>', unsafe_allow_html=True)
-
+    # Auto-scroll (optional)
     st.markdown("""
     <script>
     const c = window.parent.document.querySelector('.chat-container');