Fixing error

chatbot.py

@@ -3,38 +3,27 @@ from groq import Groq
 from langchain_groq import ChatGroq
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
+from html import escape
 import edge_tts
 import asyncio
 import os
-from html import escape
 
 GROQ_API_KEY = os.getenv('GROQ_API_KEY')
 
-
 class CodeAssistantBot:
     def __init__(self):
         self.client = Groq(api_key=GROQ_API_KEY)
         self.model = ChatGroq(model="llama-3.3-70b-versatile", temperature=0.6)
-
-        # Initialize prompts
         self.analysis_prompt = ChatPromptTemplate.from_messages([
-            ("system",
-             """You are an expert code assistant. Analyze the code and context provided,
-            then give clear, helpful responses. Keep responses concise and focused on the code."""),
-            ("user", """Code: {code}
-            Output: {output}
-            Error: {error}
-            Question: {question}""")
+            ("system", "You are an expert code assistant. Keep responses concise."),
+            ("user", "Code: {code}\nOutput: {output}\nError: {error}\nQuestion: {question}")
         ])
-
         self.summary_prompt = ChatPromptTemplate.from_messages([
-            ("system",
-             """Summarize the conversation focusing on key technical points and insights.
-            Keep it brief and clear."""),
+            ("system", "Summarize key technical points."),
             ("user", "Conversation: {conversation}")
         ])
 
-    def analyze_code(self, code: str, output: str, error: str, question: str) -> str:
+    def analyze_code(self, code, output, error, question):
         try:
             parser = StrOutputParser()
             chain = self.analysis_prompt | self.model | parser
@@ -45,115 +34,61 @@ class CodeAssistantBot:
                 'question': question
             })
         except Exception as e:
-            return f"Sorry, I encountered an error: {str(e)}"
-
-    def summarize_conversation(self, conversation: list) -> str:
-        try:
-            parser = StrOutputParser()
-            chain = self.summary_prompt | self.model | parser
-            formatted_conv = "\n".join([f"Q: {q}\nA: {a}" for q, a in conversation])
-            return chain.invoke({'conversation': formatted_conv})
-        except Exception as e:
-            return f"Could not generate summary: {str(e)}"
+            return f"Error: {e}"
 
-
-async def text_to_speech(text: str, filename: str):
+async def text_to_speech(text, filename):
     voice = "fr-FR-VivienneMultilingualNeural"
-    communicate = edge_tts.Communicate(text, voice)
-    await communicate.save(filename)
+    await edge_tts.Communicate(text, voice).save(filename)
 
 
-def render_chatbot(code: str, output: str, error: str):
-    """Render the chatbot UI in a polished, scrollable panel with support for code blocks."""
-    # --- session state ---
+def render_chatbot(code, output, error):
+    """Render the chatbot UI with code-block support and no deprecated APIs."""
     st.session_state.setdefault('conversation', [])
     st.session_state.setdefault('audio_count', 0)
 
-    # --- CSS styling ---
-    st.markdown("""
-    <style>
-    .chat-container {
-        display: flex;
-        flex-direction: column;
-        gap: 0.75rem;
-        max-height: 480px;
-        overflow-y: auto;
-        padding: 1rem;
-        border-radius: 8px;
-        background-color: inherit;
-        border: 1px solid;
-        animation: fadeIn 0.5s ease;
-    }
-    .chat-message {
-        max-width: 90%;
-        padding: 0.75rem 1rem;
-        border-radius: 12px;
-        position: relative;
-        animation: popIn 0.3s ease;
-    }
-    .user-message {
-        background: rgba(100, 149, 237, 0.2);
-        align-self: flex-end;
-    }
-    .bot-message {
-        background: rgba(200, 200, 200, 0.2);
-        align-self: flex-start;
-    }
-    .chat-message pre {
-        background: rgba(0,0,0,0.1);
-        padding: 0.5rem;
-        border-radius: 4px;
-        overflow-x: auto;
-    }
-    @keyframes fadeIn {
-        from { opacity: 0; } to { opacity: 1; }
-    }
-    @keyframes popIn {
-        from { transform: scale(0.95); opacity: 0; }
-        to { transform: scale(1);   opacity: 1; }
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
-    # --- input area ---
-    cols = st.columns([4, 1], gap='small')
-    with cols[0]:
+    # Input row
+    c1, c2 = st.columns([4,1], gap='small')
+    with c1:
         question = st.text_input("Ask your question…", key="chat_input")
-    with cols[1]:
+    with c2:
         send = st.button("🚀")
 
-    # --- handle send ---
+    # Handle send
     if send and question:
         bot = CodeAssistantBot()
-        answer = bot.analyze_code(code, output, error, question)
-        st.session_state.conversation.append((question, answer))
-        # summary & tts can go here
+        response = bot.analyze_code(code, output, error, question)
+        st.session_state.conversation.append((question, response))
 
-    # --- render chat ---
+    # Chat container
     st.markdown('<div class="chat-container">', unsafe_allow_html=True)
     for q, a in st.session_state.conversation:
-        # user bubble
-        html_q = f'<div class="chat-message user-message">{escape(q)}</div>'
-        st.markdown(html_q, unsafe_allow_html=True)
-        # bot bubble: convert code fences to <pre>
-        def render_bot(text):
+        # 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('```')
-            html = ''
+            result = ''
             for i, part in enumerate(parts):
-                if i % 2 == 0:
-                    html += escape(part)
+                if i % 2 == 1:
+                    # Remove optional language tag
+                    lines = part.splitlines()
+                    if lines and lines[0].isalpha():
+                        lines = lines[1:]
+                    code_html = escape("\n".join(lines))
+                    result += f'<pre><code>{code_html}</code></pre>'
                 else:
-                    code_block = escape(part)
-                    html += f'<pre><code>{code_block}</code></pre>'
-            return html
-        html_a = f'<div class="chat-message bot-message">{render_bot(a)}</div>'
-        st.markdown(html_a, unsafe_allow_html=True)
+                    result += escape(part)
+            return result
+
+        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 ---
+    # Auto-scroll
     st.markdown("""
     <script>
     const c = window.parent.document.querySelector('.chat-container');
-    if(c) { c.scrollTop = c.scrollHeight; }
+    if (c) c.scrollTop = c.scrollHeight;
     </script>
-    """, unsafe_allow_html=True)
+    """, unsafe_allow_html=True)