Refactor UI to Code Studio IDE layout

app.py

@@ -14,7 +14,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 # Basic Setup
-st.set_page_config(page_title="Code Chatbot", page_icon="💻", layout="wide")
+st.set_page_config(page_title="Code Chatbot", page_icon="💻", layout="wide", initial_sidebar_state="collapsed")
 logging.basicConfig(level=logging.INFO)
 
 # --- Custom CSS for Premium Slate UI ---
@@ -486,7 +486,7 @@ with st.sidebar:
                     st.session_state.indexed_files = repo_files  # For file tree
                     st.session_state.workspace_root = workspace_root  # For relative paths
                     time.sleep(0.5)  # Brief pause to show success
-                    st.rerun()
+                    st.switch_page("pages/1_⚡_Code_Studio.py")
 
     if st.session_state.processed_files:
         st.success(f"✅ Codebase Ready ({provider}) + AST 🧠")
@@ -549,50 +549,5 @@ if not st.session_state.processed_files:
     4. **Explore** your code with the file explorer and chat interface
     """)
 else:
-    # Home page - show navigation to other pages
-    st.markdown("""
-    ### 🎉 Codebase Ready!
-    
-    Your codebase has been indexed and is ready to explore. Use the pages below:
-    """)
-    
-    # Navigation cards
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        st.markdown("""
-        <div style="background: rgba(30, 41, 59, 0.6); border: 1px solid rgba(56, 189, 248, 0.2); border-radius: 12px; padding: 20px; margin: 10px 0;">
-            <h3>📁 Explorer</h3>
-            <p style="color: #94a3b8;">Browse files and view code with syntax highlighting</p>
-        </div>
-        """, unsafe_allow_html=True)
-        
-        st.markdown("""
-        <div style="background: rgba(30, 41, 59, 0.6); border: 1px solid rgba(139, 92, 246, 0.2); border-radius: 12px; padding: 20px; margin: 10px 0;">
-            <h3>🔍 Search</h3>
-            <p style="color: #94a3b8;">Search across all indexed files with regex support</p>
-        </div>
-        """, unsafe_allow_html=True)
-    
-    with col2:
-        st.markdown("""
-        <div style="background: rgba(30, 41, 59, 0.6); border: 1px solid rgba(34, 197, 94, 0.2); border-radius: 12px; padding: 20px; margin: 10px 0;">
-            <h3>💬 Chat</h3>
-            <p style="color: #94a3b8;">Ask questions about your code and get AI-powered answers</p>
-        </div>
-        """, unsafe_allow_html=True)
-        
-        st.markdown("""
-        <div style="background: rgba(30, 41, 59, 0.6); border: 1px solid rgba(251, 191, 36, 0.2); border-radius: 12px; padding: 20px; margin: 10px 0;">
-            <h3>✨ Generate</h3>
-            <p style="color: #94a3b8;">Generate new code and modify existing files</p>
-        </div>
-        """, unsafe_allow_html=True)
-    
-    st.info("👈 Use the sidebar to navigate between pages")
-    
-    # Quick stats
-    indexed_files = st.session_state.get("indexed_files", [])
-    if indexed_files:
-        st.markdown("---")
-        st.markdown(f"**📊 Stats:** {len(indexed_files)} files indexed")
+    # Codebase Ready! Redirect to Code Studio
+    st.switch_page("pages/1_⚡_Code_Studio.py")

components/panels.py

@@ -0,0 +1,268 @@
+import streamlit as st
+import os
+import re
+from pathlib import Path
+import logging
+
+def render_chat_panel(chat_engine):
+    """
+    Renders the Chat interface within the side panel.
+    """
+    st.markdown("### 💬 Chat")
+    
+    # Initialize messages for this specific panel usage if needed
+    # But we usually share global st.session_state.messages
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+
+    # Suggestion buttons (only if no messages)
+    if not st.session_state.messages:
+        st.markdown("#### 💡 Try asking:")
+        
+        col1, col2 = st.columns(2)
+        with col1:
+            if st.button("🔍 Explain structure", use_container_width=True, key="btn_explain"):
+                st.session_state.pending_prompt = "Explain the project structure and main components"
+                st.rerun()
+            if st.button("⚡ Generate utility", use_container_width=True, key="btn_util"):
+                st.session_state.pending_prompt = "Generate a new utility function for this project"
+                st.rerun()
+        with col2:
+            if st.button("📝 List functions", use_container_width=True, key="btn_list"):
+                st.session_state.pending_prompt = "List all the main functions and their purpose"
+                st.rerun()
+            if st.button("🔧 Improvements", use_container_width=True, key="btn_imp"):
+                st.session_state.pending_prompt = "What improvements would you suggest for this code?"
+                st.rerun()
+    
+    # Message Container
+    # We use a container with fixed height to allow scrolling independent of the editor
+    # But standard Streamlit scrolling works best if we just let it flow in the column.
+    
+    # Display chat history
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            # Render Sources if available
+            if "sources" in message and message["sources"]:
+                _render_sources(message["sources"])
+            
+            st.markdown(message["content"])
+
+    # Handle pending prompt
+    prompt = st.session_state.pop("pending_prompt", None)
+
+    # Chat input
+    # key needs to be unique if we have multiple inputs, but usually only one chat input active
+    if user_input := st.chat_input("Ask about your code...", key="chat_panel_input"):
+        prompt = user_input
+
+    if prompt:
+        # Add user message
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.markdown(prompt)
+        
+        # Generate response
+        with st.chat_message("assistant"):
+            with st.spinner("Thinking..."):
+                try:
+                    # Blocking call
+                    answer_payload = chat_engine.chat(prompt)
+                    
+                    if isinstance(answer_payload, tuple):
+                        response, sources = answer_payload
+                    else:
+                        response = answer_payload
+                        sources = []
+                    
+                    if sources:
+                        _render_sources(sources)
+                    
+                    st.markdown(response)
+                    
+                    st.session_state.messages.append({
+                        "role": "assistant", 
+                        "content": response,
+                        "sources": sources
+                    })
+                    
+                except Exception as e:
+                    error_msg = f"Error: {str(e)}"
+                    st.error(error_msg)
+                    st.session_state.messages.append({"role": "assistant", "content": error_msg})
+
+def _render_sources(sources):
+    unique_sources = {}
+    for s in sources:
+        if isinstance(s, dict):
+            fp = s.get('file_path', 'Unknown')
+        else:
+            fp = str(s)
+        if fp not in unique_sources:
+            unique_sources[fp] = s
+
+    chips_html = '<div style="display: flex; gap: 8px; flex-wrap: wrap; margin-bottom: 10px;">'
+    for fp in unique_sources:
+        basename = os.path.basename(fp) if "/" in fp else fp
+        chips_html += f"""
+        <div class="source-chip">
+            📄 {basename}
+        </div>
+        """
+    chips_html += '</div>'
+    st.markdown(chips_html, unsafe_allow_html=True)
+
+
+def render_search_panel(indexed_files):
+    """
+    Renders the Search interface.
+    """
+    st.markdown("### 🔍 Search")
+    
+    query = st.text_input("Search pattern", placeholder="Enter search term or regex...", key="search_query")
+    use_regex = st.checkbox("Use regex", value=False, key="search_regex")
+    
+    file_types = st.multiselect(
+        "Filter by file type",
+        options=[".py", ".js", ".ts", ".jsx", ".tsx", ".html", ".css", ".json", ".md"],
+        default=[],
+        key="search_types"
+    )
+    
+    if query and st.button("Go", key="search_go", type="primary"):
+        results = []
+        try:
+            pattern = re.compile(query, re.IGNORECASE) if use_regex else None
+        except re.error as e:
+            st.error(f"Invalid regex: {e}")
+            return
+
+        with st.spinner("Searching..."):
+            for file_path in indexed_files:
+                if file_types:
+                    ext = Path(file_path).suffix.lower()
+                    if ext not in file_types:
+                        continue
+                
+                try:
+                    with open(file_path, "r", errors="ignore") as f:
+                        lines = f.readlines()
+                    
+                    for i, line in enumerate(lines, 1):
+                        if use_regex:
+                            if pattern.search(line):
+                                results.append({
+                                    "file": file_path,
+                                    "line_num": i,
+                                    "content": line.strip(),
+                                    "match": pattern.search(line).group()
+                                })
+                        else:
+                            if query.lower() in line.lower():
+                                results.append({
+                                    "file": file_path,
+                                    "line_num": i,
+                                    "content": line.strip(),
+                                    "match": query
+                                })
+                except Exception:
+                    continue
+        
+        st.markdown(f"**Found {len(results)} matches**")
+        
+        if results:
+            # Group by file
+            by_file = {}
+            for r in results:
+                f = r["file"]
+                if f not in by_file:
+                    by_file[f] = []
+                by_file[f].append(r)
+            
+            for file_path, matches in by_file.items():
+                filename = os.path.basename(file_path)
+                with st.expander(f"📄 {filename} ({len(matches)})", expanded=False):
+                    for m in matches[:5]:
+                        # Make clickable logic? Streamlit doesn't easily support clickable text to trigger state change without rerun
+                        # We use buttons
+                        if st.button(f"L{m['line_num']}: {m['content'][:40]}...", key=f"nav_{file_path}_{m['line_num']}"):
+                            st.session_state.selected_file = file_path
+                            # st.rerun() # Might not be needed if this triggers a rerun naturally
+        else:
+            st.info("No matches.")
+
+
+def render_generate_panel(chat_engine, indexed_files):
+    """
+    Renders the Generate/Modify interface.
+    """
+    st.markdown("### ✨ Generate")
+    
+    mode = st.radio(
+        "Mode",
+        ["New Code", "Modify", "New File"],
+        horizontal=True,
+        label_visibility="collapsed"
+    )
+    
+    if mode == "New Code":
+        st.caption("Generate new code snippet")
+        description = st.text_area("Request", placeholder="e.g. Email validator function", height=100, key="gen_desc")
+        context = st.text_input("Context", placeholder="e.g. style of utils.py", key="gen_ctx")
+        
+        if st.button("Generate", type="primary", key="btn_gen_new"):
+            with st.spinner("Working..."):
+                prompt = f"Generate code: {description}\nContext: {context}"
+                try:
+                    resp = chat_engine.chat(prompt)
+                     # Unwrap if tuple
+                    if isinstance(resp, tuple):
+                        resp = resp[0]
+                    st.code(resp)
+                except Exception as e:
+                    st.error(str(e))
+
+    elif mode == "Modify":
+        st.caption("Modify existing file")
+        # Ensure we have files
+        if not indexed_files:
+            st.error("No files indexed.")
+            return
+
+        # Use session state to remember selection if possible, or just default
+        # We need a unique key
+        selected_file = st.selectbox(
+            "File", 
+            indexed_files, 
+            format_func=lambda x: os.path.basename(x),
+            key="mod_file_select"
+        )
+        
+        modification = st.text_area("Instructions", placeholder="Add error handling...", height=100, key="mod_instr")
+        
+        if st.button("Modify", type="primary", key="btn_mod"):
+            with st.spinner("Modifying..."):
+                prompt = f"Modify {selected_file}: {modification}"
+                try:
+                    resp = chat_engine.chat(prompt)
+                    if isinstance(resp, tuple):
+                        resp = resp[0]
+                    st.code(resp)
+                except Exception as e:
+                    st.error(str(e))
+
+    elif mode == "New File":
+        st.caption("Create new file")
+        fname = st.text_input("Filename", placeholder="utils/helper.py", key="new_fname")
+        desc = st.text_area("Content Description", placeholder="Functions for...", height=100, key="new_fdesc")
+        
+        if st.button("Create", type="primary", key="btn_create_file"):
+            with st.spinner("Creating..."):
+                prompt = f"Create file {fname}: {desc}"
+                try:
+                    resp = chat_engine.chat(prompt)
+                    if isinstance(resp, tuple):
+                        resp = resp[0]
+                    st.code(resp)
+                except Exception as e:
+                    st.error(str(e))

pages/1_⚡_Code_Studio.py

@@ -0,0 +1,127 @@
+"""
+⚡ Code Studio - The Main IDE Interface
+"""
+import streamlit as st
+import os
+from components.style import apply_custom_css
+from components.file_explorer import render_file_tree
+from components.code_viewer import render_code_viewer_simple
+from components.panels import render_chat_panel, render_search_panel, render_generate_panel
+
+st.set_page_config(
+    page_title="Code Studio", 
+    page_icon="⚡", 
+    layout="wide", 
+    initial_sidebar_state="collapsed" # Hide standard sidebar
+)
+
+apply_custom_css()
+
+# --- State Management ---
+if "active_tab" not in st.session_state:
+    st.session_state.active_tab = "explorer"
+    
+if "processed_files" not in st.session_state or not st.session_state.processed_files:
+    # If accessed directly without processing, redirect home
+    st.warning("⚠️ Please index a codebase first.")
+    if st.button("Go Home"):
+        st.switch_page("app.py")
+    st.stop()
+
+# --- Layout ---
+# We use a custom 3-column layout to mimic IDE
+# Col 1: Activity Bar (Narrow, just icons)
+# Col 2: Side Panel (Resizable-ish via column ratio, contains the active tool)
+# Col 3: Main Editor (Wide, contains code)
+
+# Define column ratios
+# Activity bar needs to be very narrow. Streamlit cols are proportional.
+# Ratio: 0.5 : 3 : 7
+col_activity, col_panel, col_editor = st.columns([0.5, 3, 7])
+
+# --- 1. Activity Bar ---
+with col_activity:
+    # We use a vertical layout of buttons. 
+    # To make them look like tabs, we can use a custom component or just buttons that update state.
+    # Current limitation: Buttons rerun app. That's fine for Streamlit.
+    
+    st.markdown("<div style='margin-top: 10px;'></div>", unsafe_allow_html=True)
+    
+    # Explorer Tab
+    if st.button("📁", key="tab_explorer", help="Explorer", use_container_width=True):
+        st.session_state.active_tab = "explorer"
+    
+    # Search Tab
+    if st.button("🔍", key="tab_search", help="Search", use_container_width=True):
+        st.session_state.active_tab = "search"
+        
+    # Chat Tab
+    if st.button("💬", key="tab_chat", help="Chat", use_container_width=True):
+        st.session_state.active_tab = "chat"
+        
+    # Generate Tab
+    if st.button("✨", key="tab_generate", help="Generate", use_container_width=True):
+        st.session_state.active_tab = "generate"
+
+    # Settings / Home (Exit)
+    st.markdown("<div style='margin-top: 50vh;'></div>", unsafe_allow_html=True)
+    if st.button("🏠", key="tab_home", help="Back to Home", use_container_width=True):
+        st.switch_page("app.py")
+
+
+# --- 2. Side Panel ---
+with col_panel:
+    active_tab = st.session_state.active_tab
+    
+    if active_tab == "explorer":
+        st.markdown("### 📁 Explorer")
+        render_file_tree(
+            st.session_state.get("indexed_files", []),
+            st.session_state.get("workspace_root", "")
+        )
+        
+    elif active_tab == "search":
+        render_search_panel(st.session_state.get("indexed_files", []))
+        
+    elif active_tab == "chat":
+        chat_engine = st.session_state.get("chat_engine")
+        if chat_engine:
+            render_chat_panel(chat_engine)
+        else:
+            st.error("Chat engine unavailable.")
+            
+    elif active_tab == "generate":
+        chat_engine = st.session_state.get("chat_engine")
+        if chat_engine:
+            render_generate_panel(chat_engine, st.session_state.get("indexed_files", []))
+        else:
+            st.error("Chat engine unavailable.")
+
+
+# --- 3. Main Editor ---
+with col_editor:
+    # If a file is selected, show it. Otherwise show welcome/empty state.
+    selected_file = st.session_state.get("selected_file")
+    
+    if selected_file:
+        # We use a container to ensure height consistency
+        with st.container():
+            # Breadcrumbs / File Header
+            filename = os.path.basename(selected_file)
+            st.markdown(f"**{filename}**")
+            
+            # Code Viewer
+            render_code_viewer_simple(selected_file)
+            
+    else:
+        # Empty State
+        st.markdown(
+            """
+            <div style="display: flex; flex-direction: column; align-items: center; justify-content: center; height: 60vh; opacity: 0.5;">
+                <h1>⚡ Code Studio</h1>
+                <p>Select a file from the explorer to view context.</p>
+                <p>Use the activity bar on the left to toggle tools.</p>
+            </div>
+            """, 
+            unsafe_allow_html=True
+        )

pages/1_📁_Explorer.py

@@ -1,42 +0,0 @@
-"""
-📁 Explorer Page - Browse files and view code
-"""
-import streamlit as st
-import os
-from pathlib import Path
-
-from components.style import apply_custom_css
-
-st.set_page_config(page_title="Explorer | Code Crawler", page_icon="📁", layout="wide")
-apply_custom_css()
-
-# Check if codebase is indexed
-if not st.session_state.get("processed_files"):
-    st.warning("⚠️ No codebase indexed yet. Go to **Home** to upload and index a codebase.")
-    st.stop()
-
-# Get indexed files
-indexed_files = st.session_state.get("indexed_files", [])
-workspace_root = st.session_state.get("workspace_root", "")
-
-st.title("📁 Code Explorer")
-st.caption(f"{len(indexed_files)} files indexed")
-
-# Two-column layout: File tree (25%) | Code viewer (75%)
-# CSS is now handled by apply_custom_css
-
-
-col1, col2 = st.columns([1, 3])
-
-with col1:
-    from components.file_explorer import render_file_tree
-    
-    render_file_tree(
-        st.session_state.get("indexed_files", []),
-        st.session_state.get("workspace_root", "")
-    )
-
-with col2:
-    from components.code_viewer import render_code_viewer_simple
-    
-    render_code_viewer_simple(st.session_state.get("selected_file"))

pages/2_💬_Chat.py

@@ -1,133 +0,0 @@
-"""
-💬 Chat Page - Chat with your codebase
-"""
-import streamlit as st
-import os
-from components.style import apply_custom_css
-
-st.set_page_config(page_title="Chat | Code Crawler", page_icon="💬", layout="wide")
-apply_custom_css()
-
-# Check if codebase is indexed
-if not st.session_state.get("processed_files"):
-    st.warning("⚠️ No codebase indexed yet. Go to **Home** to upload and index a codebase.")
-    st.stop()
-
-chat_engine = st.session_state.get("chat_engine")
-if not chat_engine:
-    st.error("Chat engine not initialized. Please re-index your codebase.")
-    st.stop()
-
-st.title("💬 Chat with Your Codebase")
-
-# Initialize messages
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-
-# Suggestion buttons (only if no messages)
-if not st.session_state.messages:
-    st.markdown("### 💡 Try asking:")
-    
-    col1, col2 = st.columns(2)
-    with col1:
-        if st.button("🔍 Explain project structure", use_container_width=True):
-            st.session_state.pending_prompt = "Explain the project structure and main components"
-            st.rerun()
-        if st.button("⚡ Generate utility function", use_container_width=True):
-            st.session_state.pending_prompt = "Generate a new utility function for this project"
-            st.rerun()
-    with col2:
-        if st.button("📝 List main functions", use_container_width=True):
-            st.session_state.pending_prompt = "List all the main functions and their purpose"
-            st.rerun()
-        if st.button("🔧 Suggest improvements", use_container_width=True):
-            st.session_state.pending_prompt = "What improvements would you suggest for this code?"
-            st.rerun()
-
-# Display chat history
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        # Render Sources if available
-        if "sources" in message and message["sources"]:
-            unique_sources = {}
-            for s in message["sources"]:
-                if isinstance(s, dict):
-                    fp = s.get('file_path', 'Unknown')
-                else:
-                    fp = str(s)
-                if fp not in unique_sources:
-                    unique_sources[fp] = s
-
-            chips_html = '<div style="display: flex; gap: 8px; flex-wrap: wrap; margin-bottom: 10px;">'
-            for fp in unique_sources:
-                basename = os.path.basename(fp) if "/" in fp else fp
-                chips_html += f"""
-                <div class="source-chip">
-                    📄 {basename}
-                </div>
-                """
-            chips_html += '</div>'
-            st.markdown(chips_html, unsafe_allow_html=True)
-        
-        st.markdown(message["content"])
-
-# Handle pending prompt from suggestion buttons
-prompt = st.session_state.pop("pending_prompt", None)
-
-# Chat input
-if user_input := st.chat_input("Ask about your code..."):
-    prompt = user_input
-
-if prompt:
-    # Add user message
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    with st.chat_message("user"):
-        st.markdown(prompt)
-    
-    # Generate response
-    with st.chat_message("assistant"):
-        with st.spinner("Thinking..."):
-            try:
-                # Revert to blocking chat for stability
-                answer_payload = chat_engine.chat(prompt)
-                
-                # Handle response format
-                if isinstance(answer_payload, tuple):
-                    response, sources = answer_payload
-                else:
-                    response = answer_payload
-                    sources = []
-                
-                # Render sources
-                if sources:
-                    unique_sources = {}
-                    for s in sources:
-                        fp = s.get('file_path', 'Unknown')
-                        if fp not in unique_sources:
-                            unique_sources[fp] = s
-                    
-                    chips_html = '<div style="display: flex; gap: 8px; flex-wrap: wrap; margin-bottom: 10px;">'
-                    for fp in unique_sources:
-                        basename = os.path.basename(fp) if "/" in fp else fp
-                        chips_html += f"""
-                        <div class="source-chip">
-                            📄 {basename}
-                        </div>
-                        """
-                    chips_html += '</div>'
-                    st.markdown(chips_html, unsafe_allow_html=True)
-                
-                st.markdown(response)
-                
-                # Save to history
-                st.session_state.messages.append({
-                    "role": "assistant", 
-                    "content": response,
-                    "sources": sources
-                })
-                
-            except Exception as e:
-                error_msg = f"Error: {str(e)}"
-                st.error(error_msg)
-                st.session_state.messages.append({"role": "assistant", "content": error_msg})
-

pages/3_🔍_Search.py

@@ -1,106 +0,0 @@
-"""
-🔍 Search Page - Search across your codebase
-"""
-import streamlit as st
-import os
-import re
-import re
-from pathlib import Path
-from components.style import apply_custom_css
-
-st.set_page_config(page_title="Search | Code Crawler", page_icon="🔍", layout="wide")
-apply_custom_css()
-
-# Check if codebase is indexed
-if not st.session_state.get("processed_files"):
-    st.warning("⚠️ No codebase indexed yet. Go to **Home** to upload and index a codebase.")
-    st.stop()
-
-indexed_files = st.session_state.get("indexed_files", [])
-
-st.title("🔍 Search Codebase")
-st.caption(f"Search across {len(indexed_files)} indexed files")
-
-# Search inputs
-col1, col2 = st.columns([3, 1])
-with col1:
-    query = st.text_input("Search pattern", placeholder="Enter search term or regex...")
-with col2:
-    use_regex = st.checkbox("Use regex", value=False)
-
-# File type filter
-file_types = st.multiselect(
-    "Filter by file type",
-    options=[".py", ".js", ".ts", ".jsx", ".tsx", ".html", ".css", ".json", ".md"],
-    default=[]
-)
-
-if query and st.button("🔍 Search", type="primary"):
-    results = []
-    
-    try:
-        pattern = re.compile(query, re.IGNORECASE) if use_regex else None
-    except re.error as e:
-        st.error(f"Invalid regex: {e}")
-        st.stop()
-    
-    with st.spinner("Searching..."):
-        for file_path in indexed_files:
-            # Filter by file type
-            if file_types:
-                ext = Path(file_path).suffix.lower()
-                if ext not in file_types:
-                    continue
-            
-            try:
-                with open(file_path, "r", errors="ignore") as f:
-                    lines = f.readlines()
-                
-                for i, line in enumerate(lines, 1):
-                    if use_regex:
-                        if pattern.search(line):
-                            results.append({
-                                "file": file_path,
-                                "line_num": i,
-                                "content": line.strip(),
-                                "match": pattern.search(line).group()
-                            })
-                    else:
-                        if query.lower() in line.lower():
-                            results.append({
-                                "file": file_path,
-                                "line_num": i,
-                                "content": line.strip(),
-                                "match": query
-                            })
-            except Exception:
-                continue
-    
-    # Display results
-    st.markdown(f"### Found {len(results)} matches")
-    
-    if results:
-        # Group by file
-        by_file = {}
-        for r in results:
-            f = r["file"]
-            if f not in by_file:
-                by_file[f] = []
-            by_file[f].append(r)
-        
-        for file_path, matches in by_file.items():
-            filename = os.path.basename(file_path)
-            with st.expander(f"📄 **{filename}** ({len(matches)} matches)", expanded=True):
-                st.caption(file_path)
-                for m in matches[:10]:  # Limit to 10 per file
-                    st.markdown(f"**Line {m['line_num']}:** `{m['content'][:100]}...`" if len(m['content']) > 100 else f"**Line {m['line_num']}:** `{m['content']}`")
-                
-                if len(matches) > 10:
-                    st.caption(f"... and {len(matches) - 10} more matches")
-                
-                # Button to view file
-                if st.button(f"View {filename}", key=f"view_{file_path}"):
-                    st.session_state.selected_file = file_path
-                    st.switch_page("pages/1_📁_Explorer.py")
-    else:
-        st.info("No matches found. Try a different search term.")

pages/4_✨_Generate.py

@@ -1,149 +0,0 @@
-"""
-✨ Generate Page - Generate and modify code
-"""
-import streamlit as st
-from components.style import apply_custom_css
-
-st.set_page_config(page_title="Generate | Code Crawler", page_icon="✨", layout="wide")
-apply_custom_css()
-
-# Check if codebase is indexed
-if not st.session_state.get("processed_files"):
-    st.warning("⚠️ No codebase indexed yet. Go to **Home** to upload and index a codebase.")
-    st.stop()
-
-chat_engine = st.session_state.get("chat_engine")
-if not chat_engine:
-    st.error("Chat engine not initialized. Please re-index your codebase.")
-    st.stop()
-
-st.title("✨ Code Generation")
-st.caption("Generate new code based on your codebase patterns")
-
-# Generation mode
-mode = st.radio(
-    "What would you like to do?",
-    ["Generate new code", "Modify existing code", "Create a new file"],
-    horizontal=True
-)
-
-if mode == "Generate new code":
-    st.markdown("### Generate New Code")
-    
-    description = st.text_area(
-        "Describe what you want to generate:",
-        placeholder="e.g., Create a utility function to validate email addresses",
-        height=100
-    )
-    
-    context = st.text_input(
-        "Additional context (optional):",
-        placeholder="e.g., Should follow the style used in utils.py"
-    )
-    
-    if st.button("✨ Generate", type="primary", disabled=not description):
-        with st.spinner("Generating code..."):
-            prompt = f"""Generate new code based on this request:
-
-**Request:** {description}
-
-**Additional Context:** {context if context else "None"}
-
-Please generate the code following the patterns and style used in this codebase.
-Include comments explaining the code."""
-            
-            try:
-                response_payload = chat_engine.chat(prompt)
-                if isinstance(response_payload, tuple):
-                    response, _ = response_payload
-                else:
-                    response = response_payload
-                    
-                st.markdown("### Generated Code")
-                st.markdown(response)
-                
-                # Copy button
-                st.download_button(
-                    "📋 Download as file",
-                    response,
-                    file_name="generated_code.txt",
-                    mime="text/plain"
-                )
-            except Exception as e:
-                st.error(f"Error: {str(e)}")
-
-elif mode == "Modify existing code":
-    st.markdown("### Modify Existing Code")
-    
-    # File selector
-    indexed_files = st.session_state.get("indexed_files", [])
-    selected_file = st.selectbox(
-        "Select file to modify:",
-        options=indexed_files,
-        format_func=lambda x: x.split("/")[-1] if "/" in x else x
-    )
-    
-    modification = st.text_area(
-        "Describe the modification:",
-        placeholder="e.g., Add error handling to the main function",
-        height=100
-    )
-    
-    if st.button("🔧 Modify", type="primary", disabled=not modification):
-        with st.spinner("Analyzing and modifying..."):
-            prompt = f"""Modify the code in the file '{selected_file}' based on this request:
-
-**Modification Request:** {modification}
-
-Show the modified code with explanations of what changed."""
-            
-            try:
-                response_payload = chat_engine.chat(prompt)
-                if isinstance(response_payload, tuple):
-                    response, _ = response_payload
-                else:
-                    response = response_payload
-                    
-                st.markdown("### Modified Code")
-                st.markdown(response)
-            except Exception as e:
-                st.error(f"Error: {str(e)}")
-
-else:  # Create a new file
-    st.markdown("### Create New File")
-    
-    file_name = st.text_input("File name:", placeholder="e.g., utils/helpers.py")
-    
-    description = st.text_area(
-        "Describe the file's purpose:",
-        placeholder="e.g., Utility functions for data validation and formatting",
-        height=100
-    )
-    
-    if st.button("📄 Create", type="primary", disabled=not (file_name and description)):
-        with st.spinner("Creating file..."):
-            prompt = f"""Create a new file named '{file_name}' with the following purpose:
-
-**Purpose:** {description}
-
-Generate complete, production-ready code following the patterns in this codebase.
-Include proper imports, docstrings, and error handling."""
-            
-            try:
-                response_payload = chat_engine.chat(prompt)
-                if isinstance(response_payload, tuple):
-                    response, _ = response_payload
-                else:
-                    response = response_payload
-                    
-                st.markdown(f"### {file_name}")
-                st.markdown(response)
-                
-                st.download_button(
-                    "📋 Download file",
-                    response,
-                    file_name=file_name.split("/")[-1] if "/" in file_name else file_name,
-                    mime="text/plain"
-                )
-            except Exception as e:
-                st.error(f"Error: {str(e)}")