Update chatbot.py

chatbot.py

@@ -66,87 +66,41 @@ async def text_to_speech(text: str, filename: str):
 
 
 def render_chatbot(code: str, output: str, error: str):
-    """Render the chatbot UI in a fixed-height, scrollable panel"""
+    # … your imports, CSS, bot init …
 
-    # Initialize session state
-    if "conversation" not in st.session_state:
-        st.session_state.conversation = []
-    if "audio_count" not in st.session_state:
-        st.session_state.audio_count = 0
+    # 1. Ensure session state keys exist
+    st.session_state.setdefault("conversation", [])
+    st.session_state.setdefault("audio_count", 0)
 
-    # Create bot instance
-    bot = CodeAssistantBot()
-
-    # Apply CSS for scrollable chat window
-    st.markdown("""
-        <style>
-        .chat-container {
-            height: 500px;  /* Fixed height */
-            overflow-y: auto; /* Scrollable */
-            border: 1px solid #ddd;
-            padding: 10px;
-            background-color: #f9f9f9;
-            border-radius: 5px;
-        }
-        .chat-message {
-            padding: 10px;
-            border-radius: 5px;
-            margin-bottom: 10px;
-        }
-        .user-message {
-            background-color: #e3f2fd;
-        }
-        .bot-message {
-            background-color: #f5f5f5;
-        }
-        </style>
-    """, unsafe_allow_html=True)
+    # 2. **Input area first**  
+    col1, col2 = st.columns([4, 1])
+    with col1:
+        # give this a unique key so it doesn't reset on rerun
+        user_q = st.text_input("Ask your Question here", key="chat_input", placeholder="Type your question…")
+    with col2:
+        send = st.button("🚀")
 
-    # Chat history in a scrollable container
-    chat_container = st.container()
-    with chat_container:
-        st.markdown('<div class="chat-container">', unsafe_allow_html=True)
+    # 3. **Handle send**  
+    if send and user_q:
+        bot = CodeAssistantBot()
+        resp = bot.analyze_code(code, output, error, user_q)
+        st.session_state.conversation.append((user_q, resp))
 
-        # Display all chat messages
-        for q, a in st.session_state.conversation:
-            st.markdown(f'<div class="chat-message user-message">You: {q}</div>', unsafe_allow_html=True)
-            st.markdown(f'<div class="chat-message bot-message">Assistant: {a}</div>', unsafe_allow_html=True)
+        # optional: summary+TTS logic…
 
-        # Close the chat container div
-        st.markdown('</div>', unsafe_allow_html=True)
+    # 4. **Now** render the scrollable chat container
+    st.markdown('<div class="chat-container">', unsafe_allow_html=True)
+    for q, a in st.session_state.conversation:
+        st.markdown(f'<div class="chat-message user-message">You: {q}</div>', unsafe_allow_html=True)
+        st.markdown(f'<div class="chat-message bot-message">Assistant: {a}</div>', unsafe_allow_html=True)
+    st.markdown('</div>', unsafe_allow_html=True)
 
-    # Input area at the bottom
-    col1, col2 = st.columns([4, 1])
-    with col1:
-        user_input = st.text_input("Ask your Question here", key="chat_input", placeholder="Type your question here...")
-    with col2:
-        send_clicked = st.button("🚀")  # Button to send the message
-
-    if user_input and send_clicked:
-        # Get response
-        response = bot.analyze_code(code, output, error, user_input)
-        st.session_state.conversation.append((user_input, response))
-
-        # Generate summary and speech if conversation is long enough
-        if len(st.session_state.conversation) > 3:
-            with st.spinner("Generating conversation summary..."):
-                summary = bot.summarize_conversation(st.session_state.conversation)
-                audio_file = f"summary_{st.session_state.audio_count}.wav"
-                asyncio.run(text_to_speech(summary, audio_file))
-                st.session_state.audio_count += 1
-
-                with st.expander("📝 Conversation Summary", expanded=False):
-                    st.markdown(summary)
-                    st.audio(audio_file, format="audio/wav")
-
-    # **Auto-scroll to bottom (forcing UI refresh)**
+    # 5. Auto‑scroll script (as you had it)
     st.markdown("""
-        <script>
-        var chatDiv = window.parent.document.querySelector('.chat-container');
-        if (chatDiv) {
-            chatDiv.scrollTop = chatDiv.scrollHeight;
-        }
-        </script>
+    <script>
+    const el = window.parent.document.querySelector('.chat-container');
+    if(el) el.scrollTop = el.scrollHeight;
+    </script>
     """, unsafe_allow_html=True)