Bug fix

app.py

@@ -1,94 +1,122 @@
 import streamlit as st
-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
+import streamlit_ace as st_ace
+from utils import execute_code
+from chatbot import render_chatbot
 
-GROQ_API_KEY = os.getenv('GROQ_API_KEY')
+# Page configuration (must be first)
+st.set_page_config(
+    page_title="Pro Code Playground",
+    page_icon="💻",
+    layout="wide"
+)
 
-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)
-        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}")
-        ])
-        self.summary_prompt = ChatPromptTemplate.from_messages([
-            ("system", "Summarize key technical points."),
-            ("user", "Conversation: {conversation}")
-        ])
+# Theme toggle
+dm = st.sidebar.checkbox("🌙 Dark mode", value=False)
 
-    def analyze_code(self, code, output, error, question):
-        try:
-            parser = StrOutputParser()
-            chain = self.analysis_prompt | self.model | parser
-            return chain.invoke({
-                'code': code,
-                'output': output,
-                'error': error,
-                'question': question
-            })
-        except Exception as e:
-            return f"Error: {e}"
+# Styling variables
+BG = "#0f1620" if dm else "#f5f5f5"
+PANEL_BG = "#1c2330" if dm else "#ffffff"
+TEXT = "#e3e8f1" if dm else "#1a1a1a"
+ACCENT = "#ff5252"
+BORDER = "#2a3240" if dm else "#dddddd"
+SHADOW = "rgba(0,0,0,0.3)" if dm else "rgba(0,0,0,0.1)"
+ACE_THEME = "monokai" if dm else "chrome"
 
-async def text_to_speech(text, filename):
-    voice = "fr-FR-VivienneMultilingualNeural"
-    await edge_tts.Communicate(text, voice).save(filename)
+# Global CSS
+st.markdown(f"""
+<style>
+  .stApp {{ background-color: {BG}; color: {TEXT}; }}
+  [data-testid="stSidebar"] {{ background-color: {PANEL_BG} !important; }}
+  .ace_editor, .ace_scroller {{
+    background: {PANEL_BG} !important;
+    box-shadow: 0 4px 8px {SHADOW} !important;
+    border-radius: 8px !important;
+  }}
+  textarea, input, .stTextArea textarea {{
+    background: {PANEL_BG} !important;
+    color: {TEXT} !important;
+    border: 1px solid {BORDER} !important;
+    border-radius: 4px !important;
+  }}
+  button, .stDownloadButton > button {{
+    background-color: {ACCENT} !important;
+    color: #fff !important;
+    border-radius: 6px !important;
+  }}
+  .chat-container {{ background: {PANEL_BG} !important; border: 1px solid {BORDER} !important; border-radius: 8px !important; padding: 1rem; max-height: 480px; overflow-y: auto; }}
+  .chat-message {{ margin-bottom: 1rem; padding: 0.75rem 1rem; border-radius: 12px; }}
+  .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; }}
+  pre code {{ display: block; padding: 0.5rem; background: rgba(0,0,0,0.1); border-radius: 4px; overflow-x: auto; }}
+</style>
+""", unsafe_allow_html=True)
 
+# App header
+st.title("Pro Code Playground")
+st.markdown("Write, execute & export Python snippets, with built‑in AI assistance.")
 
-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)
+# Main layout
+col1, col2 = st.columns([2, 1], gap="large")
 
-    # 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("🚀")
+with col1:
+    st.subheader("Editor")
+    code = st_ace.st_ace(
+        placeholder="Start typing Python code…",
+        language="python",
+        theme=ACE_THEME,
+        keybinding="vscode",
+        font_size=14,
+        min_lines=20,
+        show_gutter=True,
+        wrap=True,
+        auto_update=True
+    )
 
-    # Handle send
-    if send and question:
-        bot = CodeAssistantBot()
-        response = bot.analyze_code(code, output, error, question)
-        st.session_state.conversation.append((question, response))
+    # Detect input() calls
+    input_calls = code.count("input(")
+    inputs = []
+    if input_calls:
+        st.info(f"Your code expects {input_calls} input(s). Provide them below:")
+        for i in range(input_calls):
+            val = st.text_input(f"Input #{i+1}", key=f"inp_{i}")
+            inputs.append(val)
+        if not all(inputs):
+            st.warning("Please fill all inputs to run the code.")
+    
+    # Run button
+    if st.button("▶️ Run"):
+        exec_code = code
+        if input_calls:
+            # Prep input override
+            vals = ",".join([f'"{v}"' for v in inputs])
+            stub = (
+                "import builtins\n"
+                "builtins._orig_input = builtins.input\n"
+                f"inputs = iter([{vals}])\n"
+                "builtins.input = lambda prompt='': next(inputs)\n"
+            )
+            exec_code = stub + exec_code
+        out, err, exc = execute_code(exec_code)
+        st.session_state.code_output = out
+        st.session_state.error_output = err or exc
 
-    # Chat container
-    st.markdown('<div class="chat-container">', unsafe_allow_html=True)
-    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)
+    # Show output / errors
+    if st.session_state.get("code_output"):
+        st.text_area("Output", st.session_state.code_output, height=120)
+    if st.session_state.get("error_output"):
+        st.error(st.session_state.error_output)
 
-        # 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:]
-                    code_html = escape("\n".join(lines))
-                    result += f'<pre><code>{code_html}</code></pre>'
-                else:
-                    result += escape(part)
-            return result
+with col2:
+    st.subheader("Code Assistant")
+    render_chatbot(
+        code,
+        st.session_state.get("code_output", ""),
+        st.session_state.get("error_output", "")
+    )
 
-        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
-    st.markdown("""
-    <script>
-    const c = window.parent.document.querySelector('.chat-container');
-    if (c) c.scrollTop = c.scrollHeight;
-    </script>
-    """, unsafe_allow_html=True)
+# Footer
+st.markdown(f"""
+<div style='text-align:center; margin-top:1rem; color:{TEXT}66;'>
+  Built with ❤️ & Streamlit
+</div>
+""", unsafe_allow_html=True)