adding to get input too

chatbot.py

@@ -21,7 +21,7 @@ class CodeAssistantBot:
              " Refer to previous summary and recent interactions to make answers accurate."
              " Keep your response short, relevant, and conversational."),
             ("user",
-             "Code: {code}\nOutput: {output}\nError: {error}\n"
+             "Code: {code}\nInput: {input}\nOutput: {output}\nError: {error}\n"
              "Summary: {summary}\nRecent: {recent}\nQuestion: {question}")
         ])
         self.summary_prompt = ChatPromptTemplate.from_messages([
@@ -34,16 +34,17 @@ class CodeAssistantBot:
              " explain it aloud like you're helping someone understand the topic clearly and confidently."
              " Keep your response conversational and short not too long, but not over short."),
             ("user",
-             "Code: {code}\nOutput: {output}\nError: {error}\n"
+             "Code: {code}\nInput: {input}\nOutput: {output}\nError: {error}\n"
              "Conversation so far: {summary}\nAnswer to explain: {answer}")
         ])
 
-    def analyze_code(self, code, output, error, question, summary="", history=None):
+    def analyze_code(self, code, input, output, error, question, summary="", history=None):
         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,
+            'input': input,
             'output': output,
             'error': error,
             'summary': summary,
@@ -51,11 +52,12 @@ class CodeAssistantBot:
             'question': question
         })
 
-    def narrate_response(self, code, output, error, answer, summary=""):
+    def narrate_response(self, code, input, output, error, answer, summary=""):
         parser = StrOutputParser()
         narration_chain = self.voice_prompt | self.model | parser
         return narration_chain.invoke({
             'code': code,
+            'input': input,
             'output': output,
             'error': error,
             'summary': summary,
@@ -67,7 +69,7 @@ async def text_to_speech(text, filename):
     communicate = edge_tts.Communicate(text, voice)
     await communicate.save(filename)
 
-def render_chatbot(code, output, error):
+def render_chatbot(code, input, output, error):
     st.markdown("""
     <style>
     .chat-container {
@@ -112,7 +114,7 @@ def render_chatbot(code, output, error):
         bot = CodeAssistantBot()
         history = st.session_state.conversation[-4:]
         summary = st.session_state.chat_summary
-        response = bot.analyze_code(code, output, error, question, summary, history)
+        response = bot.analyze_code(code, input, output, error, question, summary, history)
         st.session_state.conversation.append((question, response))
         st.session_state.chat_display_count = 5
         if len(st.session_state.conversation) >= 3:
@@ -155,7 +157,7 @@ def render_chatbot(code, output, error):
                 status_placeholder = st.empty()
                 status_placeholder.info("🧠 Generating narration...")
                 bot = CodeAssistantBot()
-                narration = bot.narrate_response(code, output, error, a, st.session_state.chat_summary)
+                narration = bot.narrate_response(code, input, output, error, a, st.session_state.chat_summary)
                 status_placeholder.info("🎙️ Converting to audio...")
                 audio_file = f"audio_{uuid.uuid4().hex}.mp3"
                 asyncio.run(text_to_speech(narration, audio_file))