Update app.py

app.py

@@ -10,7 +10,6 @@ from PIL import Image
 import fitz  # PyMuPDF
 import time
 import re
-import json
 
 # Set page configuration
 st.set_page_config(
@@ -25,7 +24,7 @@ def is_pdflatex_installed():
     return shutil.which("pdflatex") is not None
 
 # Function to convert LaTeX to PDF
-def latex_to_pdf(latex_code, output_dir=None):
+def latex_to_pdf(latex_code):
     # Check if pdflatex is installed
     if not is_pdflatex_installed():
         st.error("pdflatex not found. Debug info:")
@@ -35,12 +34,9 @@ def latex_to_pdf(latex_code, output_dir=None):
         return None, "", "Error: pdflatex is not installed or not in PATH."
     
     with tempfile.TemporaryDirectory() as temp_dir:
-        if output_dir is None:
-            output_dir = temp_dir
-            
         temp_path = Path(temp_dir)
         tex_file = temp_path / "document.tex"
-        pdf_file = Path(output_dir) / "document.pdf"
+        pdf_file = temp_path / "document.pdf"
         
         # Write LaTeX code to file
         with open(tex_file, "w") as f:
@@ -49,7 +45,7 @@ def latex_to_pdf(latex_code, output_dir=None):
         try:
             # Run pdflatex to compile the LaTeX file
             process = subprocess.run(
-                ["pdflatex", "-interaction=nonstopmode", "-output-directory", output_dir, str(tex_file)],
+                ["pdflatex", "-interaction=nonstopmode", "-output-directory", temp_dir, str(tex_file)],
                 capture_output=True,
                 text=True
             )
@@ -64,6 +60,11 @@ def latex_to_pdf(latex_code, output_dir=None):
         except Exception as e:
             return None, "", str(e)
 
+# Function to create download link for PDF
+def get_download_link(pdf_data, filename="document.pdf"):
+    b64_pdf = base64.b64encode(pdf_data).decode()
+    return f'<a href="data:application/pdf;base64,{b64_pdf}" download="{filename}" class="download-button">Download PDF</a>'
+
 # Convert PDF to image for preview
 def render_pdf_preview(pdf_data):
     if not pdf_data:
@@ -78,7 +79,7 @@ def render_pdf_preview(pdf_data):
         
         # Render pages as images
         images = []
-        for page_num in range(min(5, len(pdf_document))):  # Preview first 5 pages max
+        for page_num in range(min(3, len(pdf_document))):  # Preview first 3 pages max
             page = pdf_document.load_page(page_num)
             pix = page.get_pixmap(matrix=fitz.Matrix(2, 2))  # Zoom factor 2 for better resolution
             img_data = pix.tobytes("png")
@@ -91,11 +92,6 @@ def render_pdf_preview(pdf_data):
         st.error(f"Error rendering PDF preview: {str(e)}")
         return None
 
-# Function to create download link for PDF
-def get_download_link(pdf_data, filename="document.pdf"):
-    b64_pdf = base64.b64encode(pdf_data).decode()
-    return f'<a href="data:application/pdf;base64,{b64_pdf}" download="{filename}" class="download-button">Download PDF</a>'
-
 # Function to parse LaTeX errors
 def parse_latex_errors(output):
     errors = []
@@ -113,10 +109,20 @@ def parse_latex_errors(output):
     
     return errors, warnings
 
-# Function to extract document structure
+# Function to extract document structure - fixed to handle non-string input
 def extract_document_structure(latex_code):
     structure = []
     
+    # Ensure latex_code is a string
+    if latex_code is None:
+        return structure
+    
+    if not isinstance(latex_code, str):
+        try:
+            latex_code = str(latex_code)
+        except:
+            return structure
+    
     # Find sections, subsections, etc.
     section_pattern = r'\\(section|subsection|subsubsection|chapter|part)\{([^}]+)\}'
     matches = re.finditer(section_pattern, latex_code)
@@ -238,57 +244,6 @@ Your conclusion here.
 \end{document}
 """
 
-# Ace Editor component for syntax highlighting
-def create_ace_editor():
-    ace_editor_html = """
-    <div id="editor" style="height: 600px; width: 100%; border-radius: 4px;"></div>
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/ace/1.24.1/ace.js"></script>
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/ace/1.24.1/ext-language_tools.js"></script>
-    <script>
-        // Create the editor
-        var editor = ace.edit("editor");
-        editor.setTheme("ace/theme/tomorrow_night_eighties");
-        editor.session.setMode("ace/mode/latex");
-        editor.setValue(JSON.parse('{latex_content}'));
-        editor.clearSelection();
-        
-        // Enable advanced features
-        editor.setOptions({
-            enableBasicAutocompletion: true,
-            enableSnippets: true,
-            enableLiveAutocompletion: true,
-            showPrintMargin: false,
-            fontSize: 14,
-            fontFamily: "'Fira Code', 'Courier New', monospace",
-            scrollPastEnd: 0.5
-        });
-        
-        // Function to update Streamlit with the content
-        function updateStreamlit() {
-            const content = editor.getValue();
-            if (window.Streamlit) {
-                window.Streamlit.setComponentValue(content);
-            }
-        }
-        
-        // Add change event listener
-        editor.session.on('change', function() {
-            updateStreamlit();
-        });
-        
-        // Initialize Streamlit component communication
-        function onStreamlitLoad() {
-            updateStreamlit();
-        }
-        
-        if (window.Streamlit) {
-            window.Streamlit.setComponentReady();
-            window.Streamlit.onCustomComponentReady(onStreamlitLoad);
-        }
-    </script>
-    """
-    return ace_editor_html
-
 # Enhanced VSCode-like advanced styling
 st.markdown("""
 <style>
@@ -312,6 +267,18 @@ st.markdown("""
         background-color: #1177bb;
     }
     
+    /* VS Code-like editor */
+    .stTextArea textarea {
+        font-family: 'Consolas', 'Monaco', 'Courier New', monospace !important;
+        font-size: 14px !important;
+        line-height: 1.5 !important;
+        background-color: #1e1e1e !important;
+        color: #d4d4d4 !important;
+        padding: 10px !important;
+        border-radius: 4px !important;
+        border: 1px solid #252526 !important;
+    }
+    
     /* VS Code-like editor container */
     .editor-container {
         background-color: #1e1e1e;
@@ -363,27 +330,33 @@ st.markdown("""
         font-size: 12px;
     }
     
-    /* Activity bar */
-    .activity-bar {
-        width: 50px;
-        background-color: #333333;
+    /* Toolbar */
+    .toolbar {
+        background-color: #2d2d2d;
+        padding: 5px;
         display: flex;
-        flex-direction: column;
+        gap: 5px;
         align-items: center;
-        padding-top: 10px;
     }
     
-    .activity-icon {
-        width: 30px;
-        height: 30px;
-        margin-bottom: 20px;
-        opacity: 0.7;
+    .toolbar-button {
+        background-color: transparent;
+        border: none;
+        color: #cccccc;
+        padding: 5px;
         cursor: pointer;
-        text-align: center;
+        border-radius: 2px;
+    }
+    
+    .toolbar-button:hover {
+        background-color: #3c3c3c;
     }
     
-    .activity-icon:hover {
-        opacity: 1;
+    .toolbar-separator {
+        width: 1px;
+        height: 20px;
+        background-color: #555;
+        margin: 0 5px;
     }
     
     /* Terminal/console styling */
@@ -472,35 +445,6 @@ st.markdown("""
         gap: 5px;
     }
     
-    /* Toolbar */
-    .toolbar {
-        background-color: #2d2d2d;
-        padding: 5px;
-        display: flex;
-        gap: 5px;
-        align-items: center;
-    }
-    
-    .toolbar-button {
-        background-color: transparent;
-        border: none;
-        color: #cccccc;
-        padding: 5px;
-        cursor: pointer;
-        border-radius: 2px;
-    }
-    
-    .toolbar-button:hover {
-        background-color: #3c3c3c;
-    }
-    
-    .toolbar-separator {
-        width: 1px;
-        height: 20px;
-        background-color: #555;
-        margin: 0 5px;
-    }
-    
     /* Messages */
     .stInfo {
         background-color: #063b49;
@@ -545,39 +489,7 @@ st.markdown("""
         color: white;
     }
     
-    /* Custom file explorer */
-    .file-explorer {
-        background-color: #252526;
-        border: 1px solid #2d2d2d;
-        border-radius: 4px;
-        padding: 0;
-        margin-top: 10px;
-    }
-    
-    .file-header {
-        background-color: #2d2d2d;
-        padding: 5px 10px;
-        font-weight: bold;
-        font-size: 13px;
-    }
-    
-    .file-item {
-        padding: 5px 10px;
-        display: flex;
-        align-items: center;
-        cursor: pointer;
-    }
-    
-    .file-item:hover {
-        background-color: #2a2d2e;
-    }
-    
-    .file-icon {
-        margin-right: 5px;
-        opacity: 0.7;
-    }
-    
-    /* Enhanced styles for problems panel */
+    /* Problems panel */
     .problems-panel {
         background-color: #252526;
         border: 1px solid #2d2d2d;
@@ -644,81 +556,58 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
 
-# Create editor with custom HTML component
-def create_editor_component(key, height=600):
-    # Get the current value from session state or use default
-    content = st.session_state.get(key, "")
-    
-    # Create Ace editor with proper LaTeX content escaping
-    escaped_content = json.dumps(content)
-    editor_html = create_ace_editor().replace('{latex_content}', escaped_content)
-    
-    # Create component
-    component = st.components.v1.html(editor_html, height=height, scrolling=False)
-    
-    # Return the latest value
-    return component if component is not None else content
+# VS Code-style editor tabs
+def render_editor_tabs(active_tab="document.tex"):
+    tab_html = f"""
+    <div class="tab-bar">
+        <div class="tab active">
+            <span class="tab-icon">📄</span> {active_tab}
+        </div>
+    </div>
+    """
+    st.markdown(tab_html, unsafe_allow_html=True)
 
 # Build a VS Code-like toolbar
 def render_toolbar():
     toolbar_html = """
     <div class="toolbar">
-        <button class="toolbar-button" title="Bold" onclick="insertTextAtCursor('\\\\textbf{}')">
+        <button class="toolbar-button" title="Bold">
             <strong>B</strong>
         </button>
-        <button class="toolbar-button" title="Italic" onclick="insertTextAtCursor('\\\\textit{}')">
+        <button class="toolbar-button" title="Italic">
             <em>I</em>
         </button>
-        <button class="toolbar-button" title="Math" onclick="insertTextAtCursor('$  $')">
+        <button class="toolbar-button" title="Math">
             ∑
         </button>
         <div class="toolbar-separator"></div>
-        <button class="toolbar-button" title="Section" onclick="insertTextAtCursor('\\\\section{}')">
+        <button class="toolbar-button" title="Section">
             §
         </button>
-        <button class="toolbar-button" title="Subsection" onclick="insertTextAtCursor('\\\\subsection{}')">
+        <button class="toolbar-button" title="Subsection">
             §§
         </button>
         <div class="toolbar-separator"></div>
-        <button class="toolbar-button" title="Itemize" onclick="insertTextAtCursor('\\\\begin{itemize}\\n\\t\\\\item \\n\\\\end{itemize}')">
+        <button class="toolbar-button" title="Itemize">
             •
         </button>
-        <button class="toolbar-button" title="Enumerate" onclick="insertTextAtCursor('\\\\begin{enumerate}\\n\\t\\\\item \\n\\\\end{enumerate}')">
+        <button class="toolbar-button" title="Enumerate">
             1.
         </button>
         <div class="toolbar-separator"></div>
-        <button class="toolbar-button" title="Equation" onclick="insertTextAtCursor('\\\\begin{equation}\\n  \\n\\\\end{equation}')">
+        <button class="toolbar-button" title="Equation">
             =
         </button>
-        <button class="toolbar-button" title="Figure" onclick="insertTextAtCursor('\\\\begin{figure}\\n\\t\\\\centering\\n\\t\\\\includegraphics[width=0.8\\\\textwidth]{image}\\n\\t\\\\caption{Caption}\\n\\t\\\\label{fig:label}\\n\\\\end{figure}')">
+        <button class="toolbar-button" title="Figure">
             🖼
         </button>
-        <button class="toolbar-button" title="Table" onclick="insertTextAtCursor('\\\\begin{table}\\n\\t\\\\centering\\n\\t\\\\begin{tabular}{ccc}\\n\\t\\tA & B & C \\\\\\\\\\n\\t\\t1 & 2 & 3 \\\\\\\\\\n\\t\\\\end{tabular}\\n\\t\\\\caption{Caption}\\n\\t\\\\label{tab:label}\\n\\\\end{table}')">
+        <button class="toolbar-button" title="Table">
             ⊞
         </button>
     </div>
-    
-    <script>
-    function insertTextAtCursor(text) {
-        var editor = ace.edit("editor");
-        editor.insert(text);
-        editor.focus();
-    }
-    </script>
     """
     st.markdown(toolbar_html, unsafe_allow_html=True)
 
-# VS Code-style editor tabs
-def render_editor_tabs(active_tab="document.tex"):
-    tab_html = f"""
-    <div class="tab-bar">
-        <div class="tab active">
-            <span class="tab-icon">📄</span> {active_tab}
-        </div>
-    </div>
-    """
-    st.markdown(tab_html, unsafe_allow_html=True)
-
 # VS Code-style status bar
 def render_status_bar():
     status_html = """
@@ -752,7 +641,7 @@ def render_document_outline(structure):
         indent = "&nbsp;" * (level * 4)
         
         st.markdown(
-            f'<div class="outline-item" title="Go to {item["type"]}: {title}">'
+            f'<div class="outline-item">'
             f'<span>{type_icon}</span>'
             f'<span class="outline-item-text">{indent}{title}</span>'
             f'</div>',
@@ -795,6 +684,13 @@ def render_problems(errors, warnings):
 
 # Render document information panel
 def render_document_info(latex_code):
+    # Ensure latex_code is a string
+    if not isinstance(latex_code, str):
+        try:
+            latex_code = str(latex_code)
+        except:
+            latex_code = ""
+    
     # Calculate basic document stats
     word_count = len(re.findall(r'\b\w+\b', latex_code))
     char_count = len(latex_code)
@@ -872,10 +768,15 @@ def main():
             # Toolbar
             render_toolbar()
             
-            # Initialize Ace editor
-            latex_code = create_editor_component("ace_editor", height=500)
-            if latex_code is not None:
-                st.session_state.latex_code = latex_code
+            # Simple text area with VS Code styling
+            latex_code = st.text_area(
+                "",
+                value=st.session_state.latex_code,
+                height=500,
+                key="latex_editor",
+                label_visibility="collapsed"
+            )
+            st.session_state.latex_code = latex_code
             
             # Status bar
             render_status_bar()
@@ -1014,8 +915,11 @@ def main():
                 st.info("Click 'Compile' to generate PDF output")
         
         with output_tabs[1]:
-            # Document structure/outline view
-            structure = extract_document_structure(st.session_state.latex_code)
+            # Document structure/outline view (with type check)
+            if isinstance(st.session_state.latex_code, str):
+                structure = extract_document_structure(st.session_state.latex_code)
+            else:
+                structure = []
             render_document_outline(structure)
         
         with output_tabs[2]: