Adding summery to chatbot

chatbot.py

@@ -15,22 +15,30 @@ 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", "You are an expert code assistant. Keep responses concise."),
-            ("user", "Code: {code}\nOutput: {output}\nError: {error}\nQuestion: {question}")
+            ("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", 
+             "Code: {code}\nOutput: {output}\nError: {error}\n"
+             "Summary: {summary}\nRecent: {recent}\nQuestion: {question}")
         ])
         self.summary_prompt = ChatPromptTemplate.from_messages([
-            ("system", "Summarize key technical points."),
+            ("system", "Summarize key technical points from the conversation so far."),
             ("user", "Conversation: {conversation}")
         ])
 
-    def analyze_code(self, code, output, error, question):
+    def analyze_code(self, code, output, error, question, summary="", history=None):
         try:
             parser = StrOutputParser()
+            recent = "\n".join([f"User: {q}\nBot: {a}" for q, a in (history or [])[-4:]])
             chain = self.analysis_prompt | self.model | parser
             return chain.invoke({
                 'code': code,
                 'output': output,
                 'error': error,
+                'summary': summary,
+                'recent': recent,
                 'question': question
             })
         except Exception as e:
@@ -40,11 +48,9 @@ async def text_to_speech(text, filename):
     voice = "fr-FR-VivienneMultilingualNeural"
     await edge_tts.Communicate(text, voice).save(filename)
 
-
 def render_chatbot(code, output, error):
     """Render the chatbot UI with code-block support and a scrollable chat container."""
-    
-    # Inject CSS to make chat container scrollable
+
     st.markdown("""
     <style>
     .chat-container {
@@ -74,35 +80,43 @@ def render_chatbot(code, output, error):
     </style>
     """, unsafe_allow_html=True)
 
-    # Setup session state
     st.session_state.setdefault('conversation', [])
     st.session_state.setdefault('audio_count', 0)
+    st.session_state.setdefault('chat_summary', "")
 
-    # Input row
     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("🚀")
 
-    # Handle send
     if send and question:
         bot = CodeAssistantBot()
-        response = bot.analyze_code(code, output, error, question)
+        history = st.session_state.conversation[-4:]
+        summary = st.session_state.chat_summary
+
+        response = bot.analyze_code(code, output, error, question, summary, history)
+
         st.session_state.conversation.append((question, response))
 
-    # Chat container
+        # Optionally update running summary
+        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:
+                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
+
     for q, a in st.session_state.conversation:
-        # User message
         st.markdown(f'<div class="chat-message user-message">{escape(q)}</div>', unsafe_allow_html=True)
 
-        # Bot message with code formatting
         def format_response(text):
             parts = text.split('```')
             result = ''
             for i, part in enumerate(parts):
                 if i % 2 == 1:
-                    # Remove optional language tag
                     lines = part.splitlines()
                     if lines and lines[0].isalpha():
                         lines = lines[1:]
@@ -114,9 +128,9 @@ 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 inside chat container
     st.markdown("""
     <script>
     const c = window.parent.document.querySelector('.chat-container');